2016-09-28 00:03:13 +02:00
|
|
|
from django.db import migrations
|
2020-04-27 07:19:08 +02:00
|
|
|
from django.db.backends.postgresql.schema import DatabaseSchemaEditor
|
2016-09-28 00:03:13 +02:00
|
|
|
from django.db.migrations.state import StateApps
|
|
|
|
|
|
2020-01-14 21:59:46 +01:00
|
|
|
|
2017-10-26 11:36:30 +02:00
|
|
|
def add_domain_to_realm_alias_if_needed(apps: StateApps, schema_editor: DatabaseSchemaEditor) -> None:
|
2016-09-28 00:03:13 +02:00
|
|
|
Realm = apps.get_model('zerver', 'Realm')
|
|
|
|
|
RealmAlias = apps.get_model('zerver', 'RealmAlias')
|
|
|
|
|
|
|
|
|
|
for realm in Realm.objects.all():
|
|
|
|
|
# if realm.domain already exists in RealmAlias, assume it is correct
|
|
|
|
|
if not RealmAlias.objects.filter(domain=realm.domain).exists():
|
|
|
|
|
RealmAlias.objects.create(realm=realm, domain=realm.domain)
|
|
|
|
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
|
|
|
|
|
|
dependencies = [
|
|
|
|
|
('zerver', '0032_verify_all_medium_avatar_images'),
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
operations = [
|
python: Use trailing commas consistently.
Automatically generated by the following script, based on the output
of lint with flake8-comma:
import re
import sys
last_filename = None
last_row = None
lines = []
for msg in sys.stdin:
m = re.match(
r"\x1b\[35mflake8 \|\x1b\[0m \x1b\[1;31m(.+):(\d+):(\d+): (\w+)", msg
)
if m:
filename, row_str, col_str, err = m.groups()
row, col = int(row_str), int(col_str)
if filename == last_filename:
assert last_row != row
else:
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
with open(filename) as f:
lines = f.readlines()
last_filename = filename
last_row = row
line = lines[row - 1]
if err in ["C812", "C815"]:
lines[row - 1] = line[: col - 1] + "," + line[col - 1 :]
elif err in ["C819"]:
assert line[col - 2] == ","
lines[row - 1] = line[: col - 2] + line[col - 1 :].lstrip(" ")
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
Signed-off-by: Anders Kaseorg <anders@zulipchat.com>
2020-04-10 05:23:40 +02:00
|
|
|
migrations.RunPython(add_domain_to_realm_alias_if_needed, elidable=True),
|
2016-09-28 00:03:13 +02:00
|
|
|
]
|
