o ג[i,@sbdZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddl Z GdddZdS) a MongoDB Service Layer This module provides all MongoDB database access for the IGF Admin application. All database operations must go through this service to maintain separation of concerns. Key principles: - Single source of truth: MongoDB database - No hardcoded collection names or database names - All configuration comes from app_config.yaml - Read/write operations are explicit and controlled N)ListDictAnyOptional) MongoClient)Database) Collectionc @seZdZdZd0defddZdedeeeffddZde fd d Z de fd d Z de efd dZ defddZdeefddZdede efddZdedefddZ  d1dedededeeeeffddZdedeeeffdd Zdeefd!d"Zd#edeefd$d%Z &d2d#edede edeeeeffd'd(Zd#edeeeffd)d*Zded+edeefd,d-Zd.d/Zd&S)3 MongoServicez MongoDB service for managing database connections and operations. This service loads configuration from app_config.yaml and provides safe, controlled access to MongoDB collections. config/app_config.yaml config_pathcCs"|||_d|_d|_d|_dS)z Initialize MongoDB service with configuration. Args: config_path: Path to YAML configuration file N) _load_configconfigclientdb_connection_error)selfr r*/var/www/html/IGF-ODF-V3/services/mongo.py__init__s  zMongoService.__init__returncCsvt|d }t|}Wdn1swY|didd}t|}|r3||dd<|Sd|dd<|S)z Load configuration from YAML file. Args: config_path: Path to configuration file Returns: Configuration dictionary rNmongodburi_env MONGO_URIurizpmongodb://AdminUser:WnE0TjGIafVJutreZL7cTIoWNWU52YyxLVgwCUqIGCGg3YUT@18.184.249.241:27017/?authMechanism=DEFAULT)openyaml safe_loadgetosgetenv)rr fr renv_urirrrr *s    zMongoService._load_configc Csz;|jdd}|jdd}|jddd}|jddd}t|||d|_|jjd|j||_d |_Wd StyS}z t ||_WYd }~d Sd }~ww) z Establish connection to MongoDB. Returns: True if connection successful, False otherwise rrdatabaseconnection_timeoutiserver_selection_timeout)serverSelectionTimeoutMSconnectTimeoutMSpingNTF) r rrradmincommandrr Exceptionstr)r mongo_uri database_namer$r%errrconnectCs&  zMongoService.connectcCs:|jdus |jdur dSz |jjdWdSYdS)z Check if MongoDB connection is active. Returns: True if connected, False otherwise NFr(T)rrr)r*rrrr is_connecteddszMongoService.is_connectedcCs|jS)z Get the last connection error message. Returns: Error message or None if no error )rr1rrrget_connection_errortsz!MongoService.get_connection_errorcCs|jddS)zz Get the configured database name. Returns: Database name from configuration rr#)r r1rrrget_database_name}szMongoService.get_database_namecCs |jdi}dd|DS)z Get list of enabled collections from configuration. Returns: List of collection names that are enabled collectionscSs g|] \}}|ddr|qS)enabledT)r).0namer rrr s z1MongoService.list_collections..)r ritems)rcollections_configrrrlist_collectionsszMongoService.list_collectionsr8cCs|jdurdS|j|S)z Get a MongoDB collection by name. Args: name: Collection name Returns: Collection object or None if not connected N)r)rr8rrrget_collections  zMongoService.get_collectioncollection_namecCs:z||}|dur WdS|iWStyYdSw)z Count total documents in a collection. Args: collection_name: Name of the collection Returns: Number of documents, or 0 if error Nr)r=count_documentsr+)rr> collectionrrrr?s   zMongoService.count_documents2rlimitskipc Cszz||}|dur gWS|i||}t|}|WSty<}ztd|d|gWYd}~Sd}~ww)a< Retrieve sample documents from a collection. Args: collection_name: Name of the collection limit: Maximum number of documents to return skip: Number of documents to skip Returns: List of documents as dictionaries NzError sampling documents from : )r=findrCrBlistr+print)rr>rBrCr@cursor documentsr/rrrsample_documentss zMongoService.sample_documentsc Csz0||}|dur iWS|jd|}|dd|dd|dd|dd|ddd WStyD||ddddd YSw) z Get statistics for a collection. Args: collection_name: Name of the collection Returns: Dictionary with collection statistics N collStatscountrsize avgObjSize storageSizenindexes)rLrMrNrOindexes)r=rr*rr+r?)rr>r@statsrrrget_collection_statss&        z!MongoService.get_collection_statsc Cs`z|sgWS|j}dd|DWSty/}ztd|gWYd}~Sd}~ww)zx List all databases on the MongoDB server. Returns: List of database names cSsg|]}|dvr|qS))r)localr r)r7rrrrr9sz3MongoService.list_all_databases..zError listing databases: N)r2rlist_database_namesr+rG)rdb_listr/rrrlist_all_databasess zMongoService.list_all_databasesr.c Cs`z|sgWS|j|}|WSty/}ztd|d|gWYd}~Sd}~ww)z List all collections in a specific database. Args: database_name: Name of the database Returns: List of collection names zError listing collections in rDN)r2rlist_collection_namesr+rG)rr.rr/rrrlist_collections_in_databases   z)MongoService.list_collections_in_databaseNc Csz%|sgWS|j|}||}|r|i|}n|i}t|WStyF}ztd|d|d|gWYd}~Sd}~ww)a) Export all documents from a collection. Args: database_name: Name of the database collection_name: Name of the collection limit: Optional limit on number of documents Returns: List of documents zError exporting .rDN)r2rrErBrFr+rG)rr.r>rBrr@rHr/rrrexport_collection_datas   z#MongoService.export_collection_datac Csz.|siWS|j|}|d}|dd|dd|dd|dd|dddWStyL}ztd |d |iWYd }~Sd }~ww) z Get statistics for a database. Args: database_name: Name of the database Returns: Dictionary with database statistics dbStatsr5rdataSizerOrQobjects)r5r]rOrQr^zError getting stats for rDN)r2rr*rr+rG)rr.rrRr/rrrget_database_stats@s         zMongoService.get_database_stats field_namec CsZz |j|}||WSty,}ztd|d|d|gWYd}~Sd}~ww)z Get distinct values for a field in a collection. Args: collection_name: Name of the collection field_name: Name of the field Returns: List of distinct values z#Error getting distinct values from rZrDN)rdistinctr+rG)rr>r`r@r/rrrget_distinct_values]s  z MongoService.get_distinct_valuescCs(|jdur|jd|_d|_dSdS)z+ Close MongoDB connection. N)rcloserr1rrrrcos   zMongoService.close)r )rAr)N)__name__ __module__ __qualname____doc__r,rrrr boolr0r2rr3r4rr<rr=intr?rJrSrWrYr[r_rbrcrrrrr sL !     " # r )rgrtypingrrrrpymongorpymongo.databaserpymongo.collectionrrr rrrrs