From 2a6ef2b9ebb4c4e7a4530bea39d401468a54b99f Mon Sep 17 00:00:00 2001 From: Tim Abbott Date: Wed, 24 Jun 2020 11:38:41 -0700 Subject: [PATCH] models: Add missing migrations from invite_as change. I neglected to `git add` after doing `renumber-migrations` for 1f8f2274445afc3b888ee3d83d342c1a7ba2aa56. --- ...0292_update_default_value_of_invited_as.py | 23 ++++++++ .../0293_update_invite_as_dict_values.py | 53 +++++++++++++++++++ 2 files changed, 76 insertions(+) create mode 100644 zerver/migrations/0292_update_default_value_of_invited_as.py create mode 100644 zerver/migrations/0293_update_invite_as_dict_values.py diff --git a/zerver/migrations/0292_update_default_value_of_invited_as.py b/zerver/migrations/0292_update_default_value_of_invited_as.py new file mode 100644 index 0000000000..f1d847afad --- /dev/null +++ b/zerver/migrations/0292_update_default_value_of_invited_as.py @@ -0,0 +1,23 @@ +# Generated by Django 2.2.13 on 2020-06-21 21:09 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ('zerver', '0291_realm_retention_days_not_null'), + ] + + operations = [ + migrations.AlterField( + model_name='multiuseinvite', + name='invited_as', + field=models.PositiveSmallIntegerField(default=400), + ), + migrations.AlterField( + model_name='preregistrationuser', + name='invited_as', + field=models.PositiveSmallIntegerField(default=400), + ), + ] diff --git a/zerver/migrations/0293_update_invite_as_dict_values.py b/zerver/migrations/0293_update_invite_as_dict_values.py new file mode 100644 index 0000000000..736ed8587a --- /dev/null +++ b/zerver/migrations/0293_update_invite_as_dict_values.py @@ -0,0 +1,53 @@ +# Generated by Django 2.2.13 on 2020-06-21 21:13 + +from django.db import migrations +from django.db.backends.postgresql.schema import DatabaseSchemaEditor +from django.db.migrations.state import StateApps + + +def update_invite_as_dict_values(apps: StateApps, schema_editor: DatabaseSchemaEditor) -> None: + PreregistrationUser = apps.get_model('zerver', 'PreregistrationUser') + MultiuseInvite = apps.get_model('zerver', 'MultiuseInvite') + + OLD_INVITE_AS_DICT = dict( + MEMBER = 1, + REALM_ADMIN = 2, + GUEST_USER = 3, + REALM_OWNER = 4, + ) + NEW_INVITE_AS_DICT = dict( + REALM_OWNER = 100, + REALM_ADMIN = 200, + MEMBER = 400, + GUEST_USER = 600, + ) + + PreregistrationUser.objects.filter(invited_as=OLD_INVITE_AS_DICT['REALM_OWNER']) \ + .update(invited_as=NEW_INVITE_AS_DICT['REALM_OWNER']) + PreregistrationUser.objects.filter(invited_as=OLD_INVITE_AS_DICT['REALM_ADMIN']) \ + .update(invited_as=NEW_INVITE_AS_DICT['REALM_ADMIN']) + PreregistrationUser.objects.filter(invited_as=OLD_INVITE_AS_DICT['MEMBER']) \ + .update(invited_as=NEW_INVITE_AS_DICT['MEMBER']) + PreregistrationUser.objects.filter(invited_as=OLD_INVITE_AS_DICT['GUEST_USER']) \ + .update(invited_as=NEW_INVITE_AS_DICT['GUEST_USER']) + + MultiuseInvite.objects.filter(invited_as=OLD_INVITE_AS_DICT['REALM_OWNER']) \ + .update(invited_as=NEW_INVITE_AS_DICT['REALM_OWNER']) + MultiuseInvite.objects.filter(invited_as=OLD_INVITE_AS_DICT['REALM_ADMIN']) \ + .update(invited_as=NEW_INVITE_AS_DICT['REALM_ADMIN']) + MultiuseInvite.objects.filter(invited_as=OLD_INVITE_AS_DICT['MEMBER']) \ + .update(invited_as=NEW_INVITE_AS_DICT['MEMBER']) + MultiuseInvite.objects.filter(invited_as=OLD_INVITE_AS_DICT['GUEST_USER']) \ + .update(invited_as=NEW_INVITE_AS_DICT['GUEST_USER']) + +class Migration(migrations.Migration): + + dependencies = [ + ('zerver', '0292_update_default_value_of_invited_as'), + ] + + operations = [ + migrations.RunPython(update_invite_as_dict_values, + reverse_code=migrations.RunPython.noop, + elidable=True), + ]