diff --git a/db-importer/db_importer/__main__.py b/db-importer/db_importer/__main__.py index 3e4cd69..919162e 100644 --- a/db-importer/db_importer/__main__.py +++ b/db-importer/db_importer/__main__.py @@ -32,7 +32,7 @@ def main(): # mongo[DB][COLLECTION].insert_many(items) with mongo.start_session() as s: s.start_transaction() - mongo[DB].drop_database() + mongo.drop_database(DB) for collection in collections: print(f"Inserting collection: {collection}") mongo[DB][collection].insert_many(collections[collection])