From d0e9d6913d12c0afaecbf601685219ca341ca197 Mon Sep 17 00:00:00 2001 From: Michal Kunc Date: Wed, 24 Nov 2021 14:25:43 +0100 Subject: [PATCH] Add mongo transaction --- db-importer/db_importer/__main__.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/db-importer/db_importer/__main__.py b/db-importer/db_importer/__main__.py index fc3a978..c3b64ea 100644 --- a/db-importer/db_importer/__main__.py +++ b/db-importer/db_importer/__main__.py @@ -28,11 +28,15 @@ def main(): if obj_type not in collections: collections[obj_type] = [] collections[obj_type].append(obj) - print("Inserting all data") - mongo[DB][COLLECTION].insert_many(items) - for collection in collections: - print(f"Inserting collection: {collection}") - mongo[DB][collection].insert_many(collections[collection]) + # print("Inserting all data") + # mongo[DB][COLLECTION].insert_many(items) + with mongo.start_session() as s: + s.start_transaction() + s[DB].drop_database() + for collection in collections: + print(f"Inserting collection: {collection}") + mongo[DB][collection].insert_many(collections[collection]) + s.commit_transaction() if __name__ == "__main__": main() \ No newline at end of file