paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-12 09:58:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

import script: Support import of reactions.

Browse Source
This commit is contained in:
Rhea Parekh
2018-03-25 17:00:04 +05:30
committed by Tim Abbott
parent 8a028142d8
commit 6f867fee40
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
zerver/lib/export.py
View File

@@ -23,7 +23,7 @@ from zerver.lib.bulk_create import bulk_create_users
from zerver.models import UserProfile, Realm, Client, Huddle, Stream, \ from zerver.models import UserProfile, Realm, Client, Huddle, Stream, \
UserMessage, Subscription, Message, RealmEmoji, RealmFilter, \ UserMessage, Subscription, Message, RealmEmoji, RealmFilter, \
RealmDomain, Recipient, DefaultStream, get_user_profile_by_id, \ RealmDomain, Recipient, DefaultStream, get_user_profile_by_id, \
UserPresence, UserActivity, UserActivityInterval, \ UserPresence, UserActivity, UserActivityInterval, Reaction, \
get_display_recipient, Attachment, get_system_bot, email_to_username get_display_recipient, Attachment, get_system_bot, email_to_username
from zerver.lib.parallel import run_parallel from zerver.lib.parallel import run_parallel
from typing import Any, Callable, Dict, List, Optional, Set, Tuple, \ from typing import Any, Callable, Dict, List, Optional, Set, Tuple, \
@@ -1224,6 +1224,7 @@ id_maps = {
'recipient': {}, 'recipient': {},
'subscription': {}, 'subscription': {},
'defaultstream': {}, 'defaultstream': {},
'reaction': {},
'realmemoji': {}, 'realmemoji': {},
'realmdomain': {}, 'realmdomain': {},
'realmfilter': {}, 'realmfilter': {},
@@ -1683,6 +1684,15 @@ def import_message_data(import_dir: Path) -> None:
update_model_ids(UserMessage, data, 'zerver_usermessage', 'usermessage') update_model_ids(UserMessage, data, 'zerver_usermessage', 'usermessage')
bulk_import_model(data, UserMessage, 'zerver_usermessage') bulk_import_model(data, UserMessage, 'zerver_usermessage')
# As the export of Reactions is not supported, Zulip exported
# data would not contain this field.
# However this is supported in slack importer script
if 'zerver_reaction' in data:
re_map_foreign_keys(data['zerver_reaction'], 'message', related_table="message")
re_map_foreign_keys(data['zerver_reaction'], 'user_profile', related_table="user_profile")
update_model_ids(Reaction, data, 'zerver_reaction', 'reaction')
bulk_import_model(data, Reaction, 'zerver_reaction')
dump_file_id += 1 dump_file_id += 1
def import_attachments(data: TableData) -> None: def import_attachments(data: TableData) -> None: