mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 05:53:43 +00:00
69730a78cc
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>
77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
# Generated by Django 1.11.4 on 2017-08-30 00:26
|
|
import ujson
|
|
from django.db import connection, migrations
|
|
from django.db.backends.postgresql.schema import DatabaseSchemaEditor
|
|
from django.db.migrations.state import StateApps
|
|
from psycopg2.sql import SQL
|
|
|
|
|
|
def convert_muted_topics(apps: StateApps, schema_editor: DatabaseSchemaEditor) -> None:
|
|
stream_query = SQL('''
|
|
SELECT
|
|
zerver_stream.name,
|
|
zerver_stream.realm_id,
|
|
zerver_stream.id,
|
|
zerver_recipient.id
|
|
FROM
|
|
zerver_stream
|
|
INNER JOIN zerver_recipient ON (
|
|
zerver_recipient.type_id = zerver_stream.id AND
|
|
zerver_recipient.type = 2
|
|
)
|
|
''')
|
|
|
|
stream_dict = {}
|
|
|
|
with connection.cursor() as cursor:
|
|
cursor.execute(stream_query)
|
|
rows = cursor.fetchall()
|
|
for (stream_name, realm_id, stream_id, recipient_id) in rows:
|
|
stream_name = stream_name.lower()
|
|
stream_dict[(stream_name, realm_id)] = (stream_id, recipient_id)
|
|
|
|
UserProfile = apps.get_model("zerver", "UserProfile")
|
|
MutedTopic = apps.get_model("zerver", "MutedTopic")
|
|
|
|
new_objs = []
|
|
|
|
user_query = UserProfile.objects.values(
|
|
'id',
|
|
'realm_id',
|
|
'muted_topics',
|
|
)
|
|
|
|
for row in user_query:
|
|
user_profile_id = row['id']
|
|
realm_id = row['realm_id']
|
|
muted_topics = row['muted_topics']
|
|
|
|
tups = ujson.loads(muted_topics)
|
|
for (stream_name, topic_name) in tups:
|
|
stream_name = stream_name.lower()
|
|
val = stream_dict.get((stream_name, realm_id))
|
|
if val is not None:
|
|
stream_id, recipient_id = val
|
|
muted_topic = MutedTopic(
|
|
user_profile_id=user_profile_id,
|
|
stream_id=stream_id,
|
|
recipient_id=recipient_id,
|
|
topic_name=topic_name,
|
|
)
|
|
new_objs.append(muted_topic)
|
|
|
|
with connection.cursor() as cursor:
|
|
cursor.execute('DELETE from zerver_mutedtopic')
|
|
|
|
MutedTopic.objects.bulk_create(new_objs)
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('zerver', '0101_muted_topic'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(convert_muted_topics, elidable=True),
|
|
]
|