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 write_data_to_file().

Browse Source
This commit is contained in:
Steve Howell
2016-08-11 07:08:13 -07:00
committed by Tim Abbott
parent f29b32bbb2
commit df3aa39be3
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
zerver/lib/export.py
View File

@@ -124,6 +124,11 @@ def sanity_check_output(data):
if table not in data: if table not in data:
logging.warn('??? NO DATA EXPORTED FOR TABLE %s!!!' % (table,)) logging.warn('??? NO DATA EXPORTED FOR TABLE %s!!!' % (table,))
def write_data_to_file(output_file, data):
# type: (Path, Any) -> None
with open(output_file, "w") as f:
f.write(ujson.dumps(data, indent=4))
def make_raw(query): def make_raw(query):
# type: (Any) -> List[Record] # type: (Any) -> List[Record]
''' '''
@@ -330,8 +335,7 @@ def export_usermessages_batch(input_path, output_path):
def write_message_export(message_filename, output): def write_message_export(message_filename, output):
# type: (Path, MessageOutput) -> None # type: (Path, MessageOutput) -> None
with open(message_filename, "w") as f: write_data_to_file(output_file=message_filename, data=output)
f.write(ujson.dumps(output, indent=4))
logging.info("Dumped to %s" % (message_filename,)) logging.info("Dumped to %s" % (message_filename,))
def export_partial_message_files(realm, response, chunk_size=1000, output_dir=None): def export_partial_message_files(realm, response, chunk_size=1000, output_dir=None):
@@ -593,8 +597,7 @@ def do_export_realm(realm, output_dir, threads):
logging.info("Exporting core realm data") logging.info("Exporting core realm data")
export_with_admin_auth(realm, response) export_with_admin_auth(realm, response)
export_file = os.path.join(output_dir, "realm.json") export_file = os.path.join(output_dir, "realm.json")
with open(export_file, "w") as f: write_data_to_file(output_file=export_file, data=response)
f.write(ujson.dumps(response, indent=4))
sanity_check_output(response) sanity_check_output(response)
@@ -634,8 +637,7 @@ def do_export_user(user_profile, output_dir):
export_single_user(user_profile, response) export_single_user(user_profile, response)
export_file = os.path.join(output_dir, "user.json") export_file = os.path.join(output_dir, "user.json")
with open(export_file, "w") as f: write_data_to_file(output_file=export_file, data=response)
f.write(ujson.dumps(response, indent=4))
logging.info("Exporting messages") logging.info("Exporting messages")
export_messages_single_user(user_profile, output_dir=output_dir) export_messages_single_user(user_profile, output_dir=output_dir)