export: Use run_parallel for rewriting UserMessage data.

This commit is contained in:
Alex Vandiver
2024-01-19 21:33:37 +00:00
committed by Tim Abbott
parent a815261abf
commit b3b9d2c3cc
3 changed files with 46 additions and 107 deletions

View File

@@ -53,7 +53,6 @@ from zerver.lib.export import (
Record,
do_export_realm,
do_export_user,
export_usermessages_batch,
get_consented_user_ids,
)
from zerver.lib.import_realm import do_import_realm, get_incoming_message_ids
@@ -416,7 +415,7 @@ class RealmImportExportTest(ExportFile):
do_export_realm(
realm=realm,
output_dir=output_dir,
processes=0,
processes=1,
export_type=export_type,
exportable_user_ids=exportable_user_ids,
)
@@ -427,13 +426,6 @@ class RealmImportExportTest(ExportFile):
realm.uuid = uuid.uuid4()
realm.save()
export_usermessages_batch(
input_path=os.path.join(output_dir, "messages-000001.json.partial"),
output_path=os.path.join(output_dir, "messages-000001.json"),
export_full_with_consent=export_type == RealmExport.EXPORT_FULL_WITH_CONSENT,
consented_user_ids=exportable_user_ids,
)
def export_realm_and_create_auditlog(
self,
original_realm: Realm,