snapshot of https://github.com/zulip/zulip.git @ 48c69911c7dd4e03b20552eb3bd77ad8151c1357 (2026-09-15)

zerver/migrations/0436_realmauthenticationmethods.py

@@ -0,0 +1,53 @@
+# Generated by Django 4.2 on 2023-04-13 23:45
+
+import django.db.models.deletion
+from django.db import migrations, models
+from django.db.backends.base.schema import BaseDatabaseSchemaEditor
+from django.db.migrations.state import StateApps
+
+
+def fill_RealmAuthenticationMethod_data(
+    apps: StateApps, schema_editor: BaseDatabaseSchemaEditor
+) -> None:
+    Realm = apps.get_model("zerver", "Realm")
+    RealmAuthenticationMethod = apps.get_model("zerver", "RealmAuthenticationMethod")
+    rows_to_create = []
+    for realm in Realm.objects.order_by("id"):
+        for key, value in realm.authentication_methods.iteritems():
+            if value:
+                rows_to_create.append(RealmAuthenticationMethod(name=key, realm_id=realm.id))
+
+    RealmAuthenticationMethod.objects.bulk_create(rows_to_create, batch_size=10000)
+
+
+class Migration(migrations.Migration):
+    atomic = False
+
+    dependencies = [
+        ("zerver", "0435_scheduledmessage_rendered_content"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="RealmAuthenticationMethod",
+            fields=[
+                (
+                    "id",
+                    models.AutoField(
+                        auto_created=True, primary_key=True, serialize=False, verbose_name="ID"
+                    ),
+                ),
+                ("name", models.CharField(max_length=80)),
+                (
+                    "realm",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE, to="zerver.realm"
+                    ),
+                ),
+            ],
+            options={
+                "unique_together": {("realm", "name")},
+            },
+        ),
+        migrations.RunPython(fill_RealmAuthenticationMethod_data, elidable=True),
+    ]