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: if obj_type not in collections:
collections[obj_type] = [] collections[obj_type] = []
collections[obj_type].append(obj) collections[obj_type].append(obj)
print("Inserting all data") # print("Inserting all data")
mongo[DB][COLLECTION].insert_many(items) # mongo[DB][COLLECTION].insert_many(items)
for collection in collections: with mongo.start_session() as s:
print(f"Inserting collection: {collection}") s.start_transaction()
mongo[DB][collection].insert_many(collections[collection]) 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__": if __name__ == "__main__":
main() main()