Skip to content

Commit 0e97128

Browse files
committed
fix collection handling
1 parent d08aafe commit 0e97128

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

.evergreen/scaffold_atlas.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,13 +42,17 @@ def upload_data(db: Database, filename: Path) -> None:
4242
db.name,
4343
collection_name,
4444
)
45-
db[collection_name].delete_many({})
45+
collections = [c["name"] for c in db.list_collections()]
46+
if collection_name in collections:
47+
logger.debug("Clearing existing collection", collection_name)
48+
db[collection_name].delete_many({})
49+
4650
if not isinstance(loaded_collection, list):
4751
loaded_collection = [loaded_collection]
4852
if loaded_collection:
4953
result: InsertManyResult = db[collection_name].insert_many(loaded_collection)
5054
logger.debug("Uploaded results for %s: %s", filename.name, result.inserted_ids)
51-
else:
55+
elif collection_name not in collections:
5256
logger.debug("Empty collection named %s created", collection_name)
5357
db.create_collection(collection_name)
5458

0 commit comments

Comments
 (0)