added is_production and overrides
This commit is contained in:
1 parent
1ae0636201
commit
79d3fb72f9
4 files changed
+52
-40
No files matched your search
@@ -7,23 +7,20 @@ from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorCollection
|
||||
from pymongo import UpdateOne
|
||||
from pymongo.errors import PyMongoError
|
||||
|
||||
from monitor_utils.config import get_settings
|
||||
from monitor_utils.models import Product
|
||||
from monitor_utils.constants import STATUS_PENDING
|
||||
from monitor_utils.constants import STATUS_PENDING, MONGO_DATABASE_NAME, MONGO_URL
|
||||
|
||||
LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
||||
def get_collection(collection_name: str) -> AsyncIOMotorCollection:
|
||||
settings = get_settings()
|
||||
return get_mongo_client()[settings.mongo_database_name][collection_name]
|
||||
return get_mongo_client()[MONGO_DATABASE_NAME][collection_name]
|
||||
|
||||
|
||||
@cache
|
||||
def get_mongo_client() -> AsyncIOMotorClient:
|
||||
settings = get_settings()
|
||||
return AsyncIOMotorClient(
|
||||
settings.mongo_url,
|
||||
MONGO_URL,
|
||||
maxPoolSize=200,
|
||||
minPoolSize=10,
|
||||
retryWrites=True,
|
||||
|
||||
Reference in new issue
Block a user