paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-03 21:43:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

export: Extract import_message_data().

Browse Source
This commit is contained in:
Steve Howell
2016-08-09 14:04:39 -07:00
committed by Tim Abbott
parent 5386ed280e
commit cecfaa7761
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
zerver/lib/export.py
View File

@@ -843,6 +843,20 @@ def do_import_realm(import_dir):
import_uploads(os.path.join(import_dir, "avatars"), avatar_bucket=True) import_uploads(os.path.join(import_dir, "avatars"), avatar_bucket=True)
import_uploads(os.path.join(import_dir, "uploads")) import_uploads(os.path.join(import_dir, "uploads"))
# Import zerver_message and zerver_usermessage
import_message_data(import_dir)
# Do attachments AFTER message data is loaded.
fix_datetime_fields(data, 'zerver_attachment', 'create_time')
re_map_foreign_keys(data, 'zerver_attachment', 'owner', related_table="user_profile")
convert_to_id_fields(data, 'zerver_attachment', 'realm')
# TODO: Handle the `messages` keys.
# fix_foreign_keys(data, 'zerver_attachment', 'messages')
bulk_import_model(data, Attachment, 'zerver_attachment')
def import_message_data(import_dir):
# type: (Path) -> None
dump_file_id = 1 dump_file_id = 1
while True: while True:
message_filename = os.path.join(import_dir, "messages-%06d.json" % (dump_file_id,)) message_filename = os.path.join(import_dir, "messages-%06d.json" % (dump_file_id,))
@@ -870,9 +884,3 @@ def do_import_realm(import_dir):
dump_file_id += 1 dump_file_id += 1
fix_datetime_fields(data, 'zerver_attachment', 'create_time')
re_map_foreign_keys(data, 'zerver_attachment', 'owner', related_table="user_profile")
convert_to_id_fields(data, 'zerver_attachment', 'realm')
# TODO: Handle the `messages` keys.
# fix_foreign_keys(data, 'zerver_attachment', 'messages')
bulk_import_model(data, Attachment, 'zerver_attachment')