# Generated by Django 2.2.13 on 2020-06-20 15:22

from django.db import migrations, models
from django.db.backends.postgresql.schema import BaseDatabaseSchemaEditor
from django.db.migrations.state import StateApps

COLOR_SCHEME_AUTOMATIC = 1
COLOR_SCHEME_NIGHT = 2

# Set color_scheme to night mode, if night_mode is True.
def set_color_scheme_to_night_mode(
    apps: StateApps, schema_editor: BaseDatabaseSchemaEditor
) -> None:
    UserProfile = apps.get_model("zerver", "UserProfile")
    UserProfile.objects.filter(night_mode=True).update(color_scheme=COLOR_SCHEME_NIGHT)


class Migration(migrations.Migration):
    atomic = False

    dependencies = [
        ("zerver", "0289_tighten_attachment_size"),
    ]

    operations = [
        migrations.AddField(
            model_name="userprofile",
            name="color_scheme",
            field=models.PositiveSmallIntegerField(default=COLOR_SCHEME_AUTOMATIC),
        ),
        migrations.RunPython(
            set_color_scheme_to_night_mode, reverse_code=migrations.RunPython.noop, elidable=True
        ),
        migrations.RemoveField(
            model_name="userprofile",
            name="night_mode",
        ),
    ]
