upstream-zulip/zerver/migrations/0323_show_starred_message_counts.py

# Generated by Django 3.1.8 on 2021-05-01 12:34

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


# Set starred_message_count for already existing users to True.
def set_starred_message_count_to_true(
    apps: StateApps, schema_editor: BaseDatabaseSchemaEditor
) -> None:
    UserProfile = apps.get_model("zerver", "UserProfile")
    UserProfile.objects.filter(starred_message_counts=False).update(starred_message_counts=True)


class Migration(migrations.Migration):
    atomic = False

    dependencies = [
        ("zerver", "0322_realm_create_audit_log_backfill"),
    ]

    operations = [
        migrations.AlterField(
            model_name="userprofile",
            name="starred_message_counts",
            field=models.BooleanField(default=True),
        ),
        migrations.RunPython(
            set_starred_message_count_to_true,
            reverse_code=migrations.RunPython.noop,
            elidable=True,
        ),
    ]