Compare commits

...
2 Commits
+8 -2
View File
@@ -29,17 +29,23 @@ NO_MSRP_PRIORITY = 6
NEW_LISTING_PRIORITY = 6 NEW_LISTING_PRIORITY = 6
def get_collection(collection_name: str) -> AsyncIOMotorCollection: def get_collection(
collection_name: str, mongo_client: AsyncIOMotorClient = None
) -> AsyncIOMotorCollection:
"""Return the named collection from the shared client's database. """Return the named collection from the shared client's database.
Args: Args:
collection_name: Name of the collection to return. collection_name: Name of the collection to return.
mongo_client: optional client for projects with already configured client
Returns: Returns:
The requested collection. The requested collection.
""" """
return get_mongo_client()[MONGO_DATABASE_NAME][collection_name] if mongo_client is None:
mongo_client = get_mongo_client()
return mongo_client[MONGO_DATABASE_NAME][collection_name]
def resolve_alerts_collection( def resolve_alerts_collection(