o Uh @sdUddlmZddlmZddlmZddlmZddlm Z e e Z dZ ee d<Gdd d ZdS) ) MongoClient) Collection)Database)settings) get_loggerN db_clientcseZdZUdZdZeed<fddZdefddZ de de fd d Z d e de fd d Zde dedefddZd e fddZZS)MongoDBSingletonN_clientcs:|jdurt||_tj}t||j_td|jS)NzConnected to MongoDB) _instancesuper__new__r MONGODB_URIrr loginfo)clsconnection_string __class__2/var/www/html/RetailAPI/app/database/main/mongo.pyr s   zMongoDBSingleton.__new__returncCstj}|j|S)N)r MONGODB_DBr )selfdb_namerrr get_main_dbs zMongoDBSingleton.get_main_dbcollection_namecCs&|}||vr||||S)z(Get a collection from the main database.)rlist_collection_namescreate_collection)rrdbrrrget_collections  zMongoDBSingleton.get_collection collectioncCs|d|iS)a Retrieves an existing document from a MongoDB collection. Args: collection (Collection): The MongoDB collection. identifier: The identifier of the document. Returns: dict: The existing document if found, else None. _id)find_one)rr identifierrrrget_existing_document#s z&MongoDBSingleton.get_existing_documentr#cCs2t|ttfs td||}|d|iduS)a Checks if data exists in a MongoDB collection. Args: collection_name (str): The name of the collection. identifier: The identifier of the document. Returns: bool: True if data exists, False otherwise. z(Identifier must be an integer or string.r!N) isinstanceintstr TypeErrorrr")rrr#r rrr is_data_in_db0s  zMongoDBSingleton.is_data_in_dbcsD|||}|r|d|id|idSd|i|}||dS)Nr!z$set)r$ update_one insert_one)rr r#dataexisting_document new_documentrrrinsert_or_update_documentBs   z*MongoDBSingleton.insert_or_update_document)__name__ __module__ __qualname__r r r__annotations__r rrr'rrdictr$r&boolr)r/ __classcell__rrrrr s    r)pymongorpymongo.collectionrpymongo.databaser app.configrapp.logrr0rrr3rrrrrs