ivatar/ivatar/ivataraccount/migrations/0008_userpreference.py

# pylint: disable=invalid-name,missing-docstring
# Generated by Django 2.0.6 on 2018-07-04 12:32

from django.conf import settings
from django.db import migrations, models
import django.db.models.deletion


def add_preference_to_user(apps, schema_editor):  # pylint: disable=unused-argument
    """
    Make sure all users have preferences set up
    """
    from django.contrib.auth.models import User

    UserPreference = apps.get_model(
        "ivataraccount", "UserPreference"
    )  # pylint: disable=invalid-name
    for user in User.objects.filter(userpreference=None):
        pref = UserPreference.objects.create(user_id=user.pk)  # pragma: no cover
        pref.save()  # pragma: no cover


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

    dependencies = [
        ("auth", "0009_alter_user_last_name_max_length"),
        ("ivataraccount", "0007_auto_20180627_0624"),
    ]

    operations = [
        migrations.CreateModel(
            name="UserPreference",
            fields=[
                (
                    "theme",
                    models.CharField(
                        choices=[
                            ("default", "Default theme"),
                            ("clime", "Climes theme"),
                        ],
                        default="default",
                        max_length=10,
                    ),
                ),
                (
                    "user",
                    models.OneToOneField(
                        on_delete=django.db.models.deletion.CASCADE,
                        primary_key=True,
                        serialize=False,
                        to=settings.AUTH_USER_MODEL,
                    ),
                ),
            ],
        ),
        migrations.RunPython(add_preference_to_user),
    ]