,h tUddlmZddlmZddlmZddlmZddlm Z e e Z dZ ee d<Gdd Zy) ) MongoClient) Collection)Database)settings) get_loggerN db_clientc~eZdZUdZdZeed<fdZdefdZ de de fdZ de de fd Zde d edefd Zde fd ZxZS) MongoDBSingletonN_clientc|jSt| ||_tj}t ||j_tjd|jS)NzConnected to MongoDB) _instancesuper__new__r MONGODB_URIrr loginfo)clsconnection_string __class__s 2/var/www/html/RetailAPI/app/database/main/mongo.pyrzMongoDBSingleton.__new__sR == !GOC0CM ( 4 4 $/0A$BCMM ! HH+ ,}}returnc@tj}|j|S)N)r MONGODB_DBr )selfdb_names r get_main_dbzMongoDBSingleton.get_main_dbs%%||G$$rcollection_namecr|j}||jvr|j|||S)z(Get a collection from the main database.)rlist_collection_namescreate_collection)rrdbs rget_collectionzMongoDBSingleton.get_collections:     "":":"< <   1/""r collectionc(|jd|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$ identifiers rget_existing_documentz&MongoDBSingleton.get_existing_document#s""E:#677rr(ct|ttfs td|j |}|j 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 TypeErrorr#r')rrr(r$s r is_data_in_dbzMongoDBSingleton.is_data_in_db0sI*sCj1FG G((9 ""E:#67tCCrcK|j||}|r|jd|id|iyd|i|}|j|yw)Nr&z$set)r) update_one insert_one)rr$r(dataexisting_document new_documents rinsert_or_update_documentz*MongoDBSingleton.insert_or_update_documentBsR 66z:N   ! !5*"5~ F!:66L  ! !, /sAA)__name__ __module__ __qualname__r r r__annotations__rrrr-rr#dictr)r,boolr/r6 __classcell__)rs@rr r ssIG[%X%#c#j# 8 84 8DSDcDdD$0*0rr )pymongorpymongo.collectionrpymongo.databaser app.configrapp.logrr7rrr:r rrrDs2)% ;<0<0r