Coverage for ivatar / ivataraccount / migrations / 0008_userpreference.py: 100%

1# pylint: disable=invalid-name,missing-docstring 

2# Generated by Django 2.0.6 on 2018-07-04 12:32 

3 

4from django.conf import settings 

5from django.db import migrations, models 

6import django.db.models.deletion 

7 

8 

9def add_preference_to_user(apps, schema_editor): # pylint: disable=unused-argument 

10 """ 

11 Make sure all users have preferences set up 

12 """ 

13 from django.contrib.auth.models import User 

14 

15 UserPreference = apps.get_model( 

16 "ivataraccount", "UserPreference" 

17 ) # pylint: disable=invalid-name 

18 for user in User.objects.filter(userpreference=None): 

19 pref = UserPreference.objects.create(user_id=user.pk) # pragma: no cover 

20 pref.save() # pragma: no cover 

21 

22 

23class Migration(migrations.Migration): # pylint: disable=missing-docstring 

24 

25 dependencies = [ 

26 ("auth", "0009_alter_user_last_name_max_length"), 

27 ("ivataraccount", "0007_auto_20180627_0624"), 

28 ] 

29 

30 operations = [ 

31 migrations.CreateModel( 

32 name="UserPreference", 

33 fields=[ 

34 ( 

35 "theme", 

36 models.CharField( 

37 choices=[ 

38 ("default", "Default theme"), 

39 ("clime", "Climes theme"), 

40 ], 

41 default="default", 

42 max_length=10, 

43 ), 

44 ), 

45 ( 

46 "user", 

47 models.OneToOneField( 

48 on_delete=django.db.models.deletion.CASCADE, 

49 primary_key=True, 

50 serialize=False, 

51 to=settings.AUTH_USER_MODEL, 

52 ), 

53 ), 

54 ], 

55 ), 

56 migrations.RunPython(add_preference_to_user), 

57 ]