renamed argument for clarity

This commit is contained in:
Brian Bjarke Jensen
2024-03-31 12:20:03 +02:00
parent eb4185c129
commit bb95e78132
+3 -3
View File
@@ -6,16 +6,16 @@ from pymongo.collection import Collection
def count_documents(
collection: Collection,
has_annotation: bool = False,
only_with_annotation: bool = False,
) -> int:
"""
Get the total number of documents
in database that matches the filters.
"""
assert isinstance(collection, Collection)
assert isinstance(has_annotation, bool)
assert isinstance(only_with_annotation, bool)
# build query
query = {}
if has_annotation:
if only_with_annotation:
query['annotation'] = {'$ne': None}
return collection.count_documents(filter=query)