mirror of https://github.com/zulip/zulip.git
54 lines
2.1 KiB
Python
54 lines
2.1 KiB
Python
|
# 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),
|
||
|
|
]
|