from django.db import migrations, models


def migrate_break_glass_to_superuser_status(apps, schema_editor):
    User = apps.get_model("accounts", "User")
    User.objects.filter(is_superuser=True).exclude(status="inactive").update(
        status="superuser"
    )


class Migration(migrations.Migration):
    dependencies = [
        ("accounts", "0010_remove_is_staff_decouple_director_admin"),
    ]

    operations = [
        migrations.AlterField(
            model_name="user",
            name="status",
            field=models.CharField(
                choices=[
                    ("inactive", "Inactive"),
                    ("gm", "GM"),
                    ("gm_admin", "GM Admin"),
                    ("buh", "BUH"),
                    ("bu_admin", "BU Admin"),
                    ("director", "Director"),
                    ("director_admin", "Director Admin"),
                    ("account_admin", "Account Admin"),
                    ("superuser", "Superuser"),
                ],
                default="gm",
                max_length=30,
            ),
        ),
        migrations.RunPython(
            migrate_break_glass_to_superuser_status, migrations.RunPython.noop
        ),
    ]