o ŸÁŠiç2ã@szdZddlZddlmZddlZddlmZmZmZddl m Z ddl m Z ddl m Z e e¡ZGdd „d ƒZeƒZdS) zm Bot Configuration Service for MCube Voice Bot Handles dynamic bot configuration based on DID number mapping éN)ÚError)ÚOptionalÚDictÚAny)Údatetime)ÚConfig)ÚLogc @sìeZdZdZdd„Zddefdd„Zded eefd d „Z defd d „Z ded eefdd„Z deded ee ee ffdd„Zded e ee ffdd„Zdeded ee ee ffdd„Zded eefdd„Zd e eeffdd„ZdS)ÚBotConfigurationServicezž Service for managing dynamic bot configurations based on DID number mapping. Maps DID numbers to specific bot configurations stored in database. cCsHtjdtjtjtjddœ|_tjtjtjtjtjddœ|_t  d¡dS)NÚmainvoicebot_masterT)ÚhostÚdatabaseÚuserÚpasswordÚcharsetÚ autocommitu*🤖 Bot Configuration Service initialized) rÚ DATABASE_HOSTÚ DATABASE_USERÚDATABASE_PASSWORDÚDATABASE_CHARSETÚmaster_db_configÚ DATABASE_NAMEÚcluster_db_configrÚinfo)Úself©rúG/var/www/html/live_calls/homebook/services/bot_configuration_service.pyÚ__init__sú ú z BotConfigurationService.__init__ÚclusterÚ database_typec Cs^z|dkr|jn|j}tjjdi|¤Ž}|WSty.}z t d|›d|›¡‚d}~ww)zÏ Get database connection for master or cluster database. Args: database_type: 'master' or 'cluster' Returns: Database connection Úmasteru⌠Error connecting to z database: Nr)rrÚmysqlÚ connectorÚconnectrrÚerror)rrÚconfigÚ connectionÚerrrÚget_connection,s €þz&BotConfigurationService.get_connectionÚ did_numberÚreturnc Cs<d}d}z‹z]| d¡}|jdd}d}| ||f¡| ¡}|rI|d}|d}t d|›d |›d |›¡||fWW|rA| ¡|rH| ¡SSt d |›¡WW|rY| ¡|ra| ¡d Sd Sty}z!t  d |›d|›¡WYd}~W|r| ¡|r‰| ¡d Sd Sd}~ww|r–| ¡|r| ¡ww)zÖ Get bot_id from DID number mapping in master database. Args: did_number: DID number to lookup Returns: bot_id if found, None otherwise NrT©Ú dictionaryz‚ SELECT bot_id, business_id FROM did_numbers WHERE did_no = %s AND status = '1' Úbot_idÚ business_idu🔠Found bot mapping - DID: z , Bot ID: z, Business ID: õ%âš ï¸ No bot mapping found for DID: )NNu⌠Error looking up DID ú: © r'ÚcursorÚexecuteÚfetchonerrÚcloseÚwarningrr#) rr(r%r1ÚqueryÚresultr,r-r&rrrÚget_bot_id_by_did>sP      ÿ÷ ÿú  ÿ€ú ÿz)BotConfigurationService.get_bot_id_by_didcƒó2ddl}‡‡fdd„}| ¡}| d|¡IdHS)z5Async version of get_bot_id_by_did using thread pool.rNcó ˆ ˆ¡S©N)r8r©r(rrrÚ_sync_get_bot_idnó zIBotConfigurationService.get_bot_id_by_did_async.._sync_get_bot_id©ÚasyncioÚget_event_loopÚrun_in_executor)rr(r@r=Úlooprr<rÚget_bot_id_by_did_asyncjs €z/BotConfigurationService.get_bot_id_by_did_asyncr-c Cs*d}d}z‚zT| d¡}|jdd}d}| ||f¡| ¡}|r@|d}t d|›d|›¡|WW|r8| ¡|r?| ¡SSt d |›¡WW|rP| ¡|rX| ¡dSdSty†}z!t  d |›d |›¡WYd}~W|rx| ¡|r€| ¡dSdSd}~ww|r| ¡|r”| ¡ww) zâ Get business name from business table in master database. Args: business_id: Business ID to lookup Returns: Business name if found, None otherwise NrTr*z… SELECT business_name FROM businesses WHERE business_id = %s AND status = 'active' Ú business_nameu(🢠Found business name - Business ID: z, Name: u*âš ï¸ No business found for Business ID: u⌠Error looking up business r/r0)rr-r%r1r6r7rEr&rrrÚget_business_name_by_idtsN     ÿ÷ ÿú  ÿ€ú ÿz/BotConfigurationService.get_business_name_by_idr,c CsPd}d}z•zg| d¡}|jdd}|›d}d|›d}| ||f¡| ¡}|rS|d|d |d d œ}t d |d ›d |›¡|WW|rK| ¡|rR| ¡SSt d|›¡WW|rc| ¡|rk| ¡dSdSty™} z!t  d|›d| ›¡WYd} ~ W|r‹| ¡|r“| ¡dSdSd} ~ ww|r | ¡|r§| ¡ww)zä Get bot configuration from cluster database. Args: business_id: Business ID bot_id: Bot ID Returns: Bot configuration dictionary or None NrTr*Ú_botszA SELECT bot_id, bot_name, agent_id FROM `z>` WHERE bot_id = %s AND is_active = 1 r,Úbot_nameÚagent_id)r,rHrIu&🤖 Loaded bot configuration - Name: z, ID: u5âš ï¸ No active bot configuration found for Bot ID: u0⌠Error fetching bot configuration for Bot ID r/r0) rr-r,r%r1Ú table_namer6r7Ú bot_configr&rrrÚget_bot_configurationŸsZ    þý  ÿ÷ ÿú  ÿ€ú ÿz-BotConfigurationService.get_bot_configurationc ÃsÂz¹t ¡}| |¡IdH}|r|ddus|ddur/t d|›d¡ddddtjdœWS|\}}ddl}g}| |  ||¡¡}|  d|f¡| |  |¡¡} |  d | f¡|j d d „|Dƒd d iŽIdH} t | dtƒss| dnd} t | dtƒs€| dnd} | s™t d|›d¡dd||| p•tjdœWSt ¡| ¡} t d| d›d¡| |  dd¡||| p·tjdœWStyà}zt d|›d|›¡ddddtjdœWYd}~Sd}~ww)a\ Get complete bot configuration by DID number - OPTIMIZED for separate databases. All database operations run in thread pool to avoid blocking event loop. Args: did_number: DID number to lookup Returns: Dictionary containing bot configuration and system message Nrér.z, using default configurationzDefault Assistant)rKrHr-r,Ú company_namerKrNcSsg|]\}}|‘qSrr)Ú.0Ú_ÚtaskrrrÚ ÿszHBotConfigurationService.get_bot_configuration_by_did..Úreturn_exceptionsTu.âš ï¸ No bot configuration found for Bot ID: z, using defaultu âš¡ Bot configuration loaded in z.2fz secondsÚnameÚ Assistantu,⌠Error getting bot configuration for DID r/)rÚnowrDrr5rÚ COMPANY_NAMEr@Ú create_taskÚ_get_bot_config_asyncÚappendÚ_get_company_name_asyncÚgatherÚ isinstanceÚ ExceptionÚ total_secondsrÚgetr#)rr(Ú start_timer7r,r-r@ÚtasksÚbot_taskÚ company_taskÚresultsrKrNÚ total_timer&rrrÚget_bot_configuration_by_didÓs`€ û û  ûû€þz4BotConfigurationService.get_bot_configuration_by_didcƒs<ddl}ddl}‡‡‡fdd„}| ¡}| d|¡IdHS)z-Async version of bot configuration retrieval.rNcs ˆ ˆˆ¡Sr;)rLr©r,r-rrrÚ_sync_get_bot_config,s zKBotConfigurationService._get_bot_config_async.._sync_get_bot_config)r@Úmysql.connectorrArB)rr-r,r@r rirCrrhrrY's €z-BotConfigurationService._get_bot_config_asynccƒr9)z(Async version of company name retrieval.rNcr:r;)rFr©r-rrrÚ_sync_get_company_name7r>zOBotConfigurationService._get_company_name_async.._sync_get_company_namer?)rr-r@rlrCrrkrr[3s €z/BotConfigurationService._get_company_name_asyncc Csdddœ}z%| d¡}| ¡}| d¡| ¡| ¡| ¡d|d<t d¡WntyD}zt d|›¡WYd}~nd}~wwz&| d ¡}| ¡}| d¡| ¡| ¡| ¡d|d <t d ¡W|Sty†}zt d |›¡WYd}~|Sd}~ww) z– Test connections to both master and cluster databases. Returns: Dictionary with connection test results F)rrrzSELECT 1Tu)✅ Master database connection successfulu'⌠Master database connection failed: Nru*✅ Cluster database connection successfulu(⌠Cluster database connection failed: ) r'r1r2r3r4rrr^r#)rrer%r1r&rrrÚtest_database_connections>s:   €ÿ   ý€ýz1BotConfigurationService.test_database_connectionsN)r)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚstrr'rÚintr8rDrFrrrLrgrYr[Úboolrmrrrrr s, "+4"T  r )rqrjr rÚloggingÚtypingrrrrr$rÚservices.log_utilsrÚ getLoggerrnÚloggerr Úbot_configuration_servicerrrrÚs      X