Files
zulip/zerver/migrations/0097_reactions_emoji_code.py
Anders Kaseorg df001db1a9 black: Reformat with Black 23.
Black 23 enforces some slightly more specific rules about empty line
counts and redundant parenthesis removal, but the result is still
compatible with Black 22.

(This does not actually upgrade our Python environment to Black 23
yet.)

Signed-off-by: Anders Kaseorg <anders@zulip.com>
2023-02-02 10:40:13 -08:00

62 lines
2.2 KiB
Python

# Generated by Django 1.11.2 on 2017-06-18 21:26
import os
import orjson
from django.db import migrations, models
from django.db.backends.postgresql.schema import BaseDatabaseSchemaEditor
from django.db.migrations.state import StateApps
def populate_new_fields(apps: StateApps, schema_editor: BaseDatabaseSchemaEditor) -> None:
# Open the JSON file which contains the data to be used for migration.
MIGRATION_DATA_PATH = os.path.join(
os.path.dirname(os.path.dirname(__file__)), "management", "data"
)
path_to_unified_reactions = os.path.join(MIGRATION_DATA_PATH, "unified_reactions.json")
with open(path_to_unified_reactions, "rb") as f:
unified_reactions = orjson.loads(f.read())
Reaction = apps.get_model("zerver", "Reaction")
for reaction in Reaction.objects.all():
reaction.emoji_code = unified_reactions.get(reaction.emoji_name)
if reaction.emoji_code is None:
# If it's not present in the unified_reactions map, it's a realm emoji.
reaction.emoji_code = reaction.emoji_name
if reaction.emoji_name == "zulip":
# `:zulip:` emoji is a zulip special custom emoji.
reaction.reaction_type = "zulip_extra_emoji"
else:
reaction.reaction_type = "realm_emoji"
reaction.save()
class Migration(migrations.Migration):
dependencies = [
("zerver", "0096_add_password_required"),
]
operations = [
migrations.AddField(
model_name="reaction",
name="emoji_code",
field=models.TextField(default="unset"),
preserve_default=False,
),
migrations.AddField(
model_name="reaction",
name="reaction_type",
field=models.CharField(
choices=[
("unicode_emoji", "Unicode emoji"),
("realm_emoji", "Custom emoji"),
("zulip_extra_emoji", "Zulip extra emoji"),
],
default="unicode_emoji",
max_length=30,
),
),
migrations.RunPython(
populate_new_fields, reverse_code=migrations.RunPython.noop, elidable=True
),
]