ג[i,pdZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddl Z GddZdS) 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 eZdZdZddefdZdedeeeffdZde fdZ de fdZ de efd Z defd Zdeefd Zd ede efd ZdedefdZ d dedededeeeeffdZdedeeeffdZdeefdZdedeefdZ d!dedede edeeeeffdZdedeeeffdZdededeefdZdZdS)" 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_pathcd|||_d|_d|_d|_dS)z Initialize MongoDB service with configuration. Args: config_path: Path to YAML configuration file N) _load_configconfigclientdb_connection_error)selfr s */var/www/html/IGF-ODF-V3/services/mongo.py__init__zMongoService.__init__s5'' 44 -1 &*04returnc,t|d5}tj|}dddn #1swxYwY|didd}t j|}|r ||dd<n d|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 frrenv_uris rrzMongoService._load_config*s+s # # 'q^A&&F ' ' ' ' ' ' ' ' ' ' ' ' ' ' '**Y++// ;GG)G$$  Z'.F9 e $ $(ZF9 e $ s 266c |jdd}|jdd}|jddd}|jddd}t||||_|jjd|j||_d |_d S#t$r}t||_Yd }~d Sd }~wwxYw) z Establish connection to MongoDB. Returns: True if connection successful, False otherwise rrdatabaseconnection_timeoutiserver_selection_timeout)serverSelectionTimeoutMSconnectTimeoutMSpingNTF) rr"rradmincommandrr Exceptionstr)r mongo_uri database_namer)r*es rconnectzMongoService.connectCs  I.u5I K 2:>M!%Y!7!;!;z1MongoService.list_collections..s1cccvvzzR[]aGbGbccccr)rr"items)rcollections_configs rlist_collectionszMongoService.list_collectionss;"[__]B??cc);)A)A)C)CccccrrBc.|jdS|j|S)z Get a MongoDB collection by name. Args: name: Collection name Returns: Collection object or None if not connected N)r)rrBs rget_collectionzMongoService.get_collections 7 4wt}rcollection_namec ||}|dS|iS#t$rYdSwxYw)z Count total documents in a collection. Args: collection_name: Name of the collection Returns: Number of documents, or 0 if error Nr)rHcount_documentsr0)rrI collections rrKzMongoService.count_documentss] ,,_==J q--b11 1   11 s00 >>2rlimitskipc* ||}|gS|i||}t |}|S#t $r!}t d|d|gcYd}~Sd}~wwxYw)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 : )rHfindrOrNlistr0print)rrIrNrOrLcursor documentsr4s rsample_documentszMongoService.sample_documentss" ,,_==J   __R((--d3399%@@FV I     I?IIaII J J JIIIIII s#A'A A'' B1B B Bc ||}|iS|jd|}|dd|dd|dd|dd|ddd S#t$r||ddddd cYSwxYw) z Get statistics for a collection. Args: collection_name: Name of the collection Returns: Dictionary with collection statistics N collStatscountrsize avgObjSize storageSizenindexes)rZr[r\r]indexes)rHrr/r"r0rK)rrIrLstatss rget_collection_statsz!MongoService.get_collection_statss ,,_==J  GOOKAAE7A.. &!,,#ii a88$yy:: 99Z33     --o>>      sB"BB""%C  C c |sgS|j}d|DS#t$r}t d|gcYd}~Sd}~wwxYw)zx List all databases on the MongoDB server. Returns: List of database names cg|]}|dv| S))r.localr)rArs rrCz3MongoService.list_all_databases..s#SSS2B6R,RSBSSSrzError listing databases: N)r8rlist_database_namesr0rT)rdb_listr4s rlist_all_databaseszMongoService.list_all_databasess $$&&  k5577GTSSSS S    1a11 2 2 2IIIIII s=$= A%A A% A%r3c |sgS|j|}|S#t$r!}t d|d|gcYd}~Sd}~wwxYw)z List all collections in a specific database. Args: database_name: Name of the database Returns: List of collection names zError listing collections in rQN)r8rlist_collection_namesr0rT)rr3rr4s rlist_collections_in_databasez)MongoService.list_collections_in_databases $$&&  ]+B++-- -    F-FF1FF G G GIIIIII s9 9 A$AA$A$Nc Z |sgS|j|}||}|r)|i|}n|i}t |S#t $r$}t d|d|d|gcYd}~Sd}~wwxYw)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 .rQN)r8rrRrNrSr0rT)rr3rIrNrrLrUr4s rexport_collection_dataz#MongoService.export_collection_datas" $$&&  ]+BO,J -#,,22599#,,<<     K]KK_KKKK L L LIIIIII s#A<A#A<< B*B%B*%B*c |siS|j|}|d}|dd|dd|dd|dd|dddS#t$r!}t d |d |icYd }~Sd }~wwxYw) z Get statistics for a database. Args: database_name: Name of the database Returns: Dictionary with database statistics dbStatsr>rdataSizer]r_objects)r>rqr]r_rrzError getting stats for rQN)r8rr/r"r0rT)rr3rr`r4s rget_database_statszMongoService.get_database_stats@s $$&&  ]+BJJy))E %yy::!IIj!44$yy:: 99Y22 99Y22      A]AAaAA B B BIIIIII s#B&B B&& C0C C C field_namec  |j|}||S#t$r$}td|d|d|gcYd}~Sd}~wwxYw)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 rmrQN)rdistinctr0rT)rrIrtrLr4s rget_distinct_valuesz MongoService.get_distinct_values]s} 1J&&z22 2    [[[*[[XY[[ \ \ \IIIIII s!$ AA A Acf|j)|jd|_d|_dSdS)z+ Close MongoDB connection. N)rcloserr7s rryzMongoService.closeos> ;  K     DKDGGG  r)r )rMr)N)__name__ __module__ __qualname____doc__r1rrrrboolr5r8rr:r<rrFr rHintrKrWrarhrkrnrsrwryrerrr r s 5 5C 5 5 5 5S#X2Bd &hsm&&&&232222d$s)dddd 3 8J+?    ss*   d38n  @ C DcN    DDI*#$s)2 $ !!!!} ! d38n  !!!!FS#X:3CDQTI$rr )r}r#typingrrrrpymongorpymongo.databaserpymongo.collectionr r r rerrrs   ,,,,,,,,,,,,%%%%%%)))))) ``````````r