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

1# -*- coding: utf-8 -*- 

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

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

4 

5from django.conf import settings 

6from django.db import migrations, models 

7import django.db.models.deletion 

8 

9 

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

11 """ 

12 Make sure all users have preferences set up 

13 """ 

14 from django.contrib.auth.models import User 

15 

16 UserPreference = apps.get_model( 

17 "ivataraccount", "UserPreference" 

18 ) # pylint: disable=invalid-name 

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

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

21 pref.save() # pragma: no cover 

22 

23 

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

25 

26 dependencies = [ 

27 ("auth", "0009_alter_user_last_name_max_length"), 

28 ("ivataraccount", "0007_auto_20180627_0624"), 

29 ] 

30 

31 operations = [ 

32 migrations.CreateModel( 

33 name="UserPreference", 

34 fields=[ 

35 ( 

36 "theme", 

37 models.CharField( 

38 choices=[ 

39 ("default", "Default theme"), 

40 ("clime", "Climes theme"), 

41 ], 

42 default="default", 

43 max_length=10, 

44 ), 

45 ), 

46 ( 

47 "user", 

48 models.OneToOneField( 

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

50 primary_key=True, 

51 serialize=False, 

52 to=settings.AUTH_USER_MODEL, 

53 ), 

54 ), 

55 ], 

56 ), 

57 migrations.RunPython(add_preference_to_user), 

58 ]