michal/pf2db
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add mongo transaction

Browse Source
This commit is contained in:
Michal Kunc 2021-11-24 14:25:43 +01:00
parent c03ebbf853
commit d0e9d6913d
Signed by: michal
GPG Key ID: 4CA5FB6559E0BDF8
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
db-importer/db_importer/__main__.py
View File

@ -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()