mirror of
				https://github.com/zulip/zulip.git
				synced 2025-11-03 21:43:21 +00:00 
			
		
		
		
	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>
		
			
				
	
	
		
			62 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			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
 | 
						|
        ),
 | 
						|
    ]
 |