paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-16 20:02:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

rocketchat: Emoji can have an ObjectID as an _id.

Browse Source
This commit is contained in:
Alex Vandiver
2025-11-03 16:18:32 +00:00
committed by Tim Abbott
parent 6c1f3c09ba
commit 7a640cf91b
1 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
zerver/data_import/rocketchat.py
View File

@@ -3,6 +3,7 @@ import os
import random import random
import secrets import secrets
import uuid import uuid
from collections import defaultdict
from typing import Any from typing import Any
import bson import bson
@@ -288,21 +289,18 @@ def build_custom_emoji(
emoji_records: list[ZerverFieldsT] = [] emoji_records: list[ZerverFieldsT] = []
# Map emoji file_id to emoji file data # Map emoji file_id to emoji file data
emoji_file_data = {} emoji_file_data = defaultdict(list)
object_id_to_filename = {}
for emoji_file in custom_emoji_data["file"]: for emoji_file in custom_emoji_data["file"]:
emoji_file_data[str(emoji_file["_id"])] = {"filename": emoji_file["filename"], "chunks": []} if isinstance(emoji_file["_id"], bson.objectid.ObjectId): # nocoverage
object_id_to_filename[str(emoji_file["_id"])] = emoji_file["filename"]
for emoji_chunk in custom_emoji_data["chunk"]: for emoji_chunk in custom_emoji_data["chunk"]:
emoji_file_data[emoji_chunk["files_id"]]["chunks"].append(emoji_chunk["data"]) file_id = str(emoji_chunk["files_id"])
emoji_file_data[object_id_to_filename.get(file_id, file_id)].append(emoji_chunk["data"])
# Build custom emoji
for rc_emoji in custom_emoji_data["emoji"]: for rc_emoji in custom_emoji_data["emoji"]:
# Subject to change with changes in database emoji_filename = f"{rc_emoji['name']}.{rc_emoji['extension']}"
emoji_file_id = f"{rc_emoji['name']}.{rc_emoji['extension']}" emoji_data = b"".join(emoji_file_data[emoji_filename])
emoji_file_info = emoji_file_data[emoji_file_id]
emoji_filename = emoji_file_info["filename"]
emoji_data = b"".join(emoji_file_info["chunks"])
target_sub_path = RealmEmoji.PATH_ID_TEMPLATE.format( target_sub_path = RealmEmoji.PATH_ID_TEMPLATE.format(
realm_id=realm_id, realm_id=realm_id,