mirror of
https://github.com/zulip/zulip.git
synced 2025-11-03 05:23:35 +00:00
b94a24ffe3
The original implementation of this migration had a highly unfortunate bug that would result in it deleting all reactions to realm emoji on the server; we missed this in review, so essentially all historical realm emoji reactions on chat.zulip.org were lost :(. We both correct the problem, and also add logging of the deleted rows that would help should anything be deleted erroneously.
53 lines
2.1 KiB
Python
53 lines
2.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import ujson
|
|
|
|
from collections import defaultdict
|
|
from django.conf import settings
|
|
from django.db import migrations, models
|
|
from django.db.backends.postgresql_psycopg2.schema import DatabaseSchemaEditor
|
|
from django.db.migrations.state import StateApps
|
|
from typing import Any, Dict
|
|
|
|
def realm_emoji_name_to_id(apps: StateApps, schema_editor: DatabaseSchemaEditor) -> None:
|
|
Reaction = apps.get_model('zerver', 'Reaction')
|
|
RealmEmoji = apps.get_model('zerver', 'RealmEmoji')
|
|
realm_emoji_by_realm_id = defaultdict(dict) # type: Dict[int, Dict[str, Any]]
|
|
for realm_emoji in RealmEmoji.objects.all():
|
|
realm_emoji_by_realm_id[realm_emoji.realm_id][realm_emoji.name] = {
|
|
'id': str(realm_emoji.id),
|
|
'name': realm_emoji.name,
|
|
'deactivated': realm_emoji.deactivated,
|
|
}
|
|
for reaction in Reaction.objects.filter(reaction_type='realm_emoji'):
|
|
realm_id = reaction.user_profile.realm_id
|
|
emoji_name = reaction.emoji_name
|
|
realm_emoji = realm_emoji_by_realm_id.get(realm_id, {}).get(emoji_name)
|
|
if realm_emoji is None:
|
|
# Realm emoji used in this reaction has been deleted so this
|
|
# reaction should also be deleted. We don't need to reverse
|
|
# this step in migration reversal code.
|
|
print("Reaction for (%s, %s) refers to deleted custom emoji %s; deleting" %
|
|
(emoji_name, reaction.message_id, reaction.user_profile_id))
|
|
reaction.delete()
|
|
else:
|
|
reaction.emoji_code = realm_emoji["id"]
|
|
reaction.save()
|
|
|
|
def reversal(apps: StateApps, schema_editor: DatabaseSchemaEditor) -> None:
|
|
Reaction = apps.get_model('zerver', 'Reaction')
|
|
for reaction in Reaction.objects.filter(reaction_type='realm_emoji'):
|
|
reaction.emoji_code = reaction.emoji_name
|
|
reaction.save()
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('zerver', '0144_remove_realm_create_generic_bot_by_admins_only'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(realm_emoji_name_to_id,
|
|
reverse_code=reversal),
|
|
]
|