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