mirror of
https://github.com/zulip/zulip.git
synced 2025-10-23 04:52:12 +00:00
69ae4931c3
This replaces django.db.backends.postgresql_psycopg2, which has been an alias to django.db.backends.postgresql since Django 1.9.
50 lines
2.0 KiB
Python
50 lines
2.0 KiB
Python
from collections import defaultdict
|
|
from typing import Any, Dict
|
|
|
|
from django.db import migrations
|
|
from django.db.backends.postgresql.schema import DatabaseSchemaEditor
|
|
from django.db.migrations.state import StateApps
|
|
|
|
|
|
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: Dict[int, Dict[str, Any]] = defaultdict(dict)
|
|
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),
|
|
]
|