o i @sjUdZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z ddl Z dedejfd d Z  dd ed eBdejdBdedBdeddf ddZej  d d edBdejdBdedBdefddZejejejejejddZeed dBedBfed<GdddejZ  d dejded dBded dBddfddZ dejddfddZ!dS)!zLogging utilities for FastMCP.N)AnyLiteralcast)Console) RichHandler)overridenamereturncCs(|dr tj|dStjd|dS)zGet a logger nested under FastMCP namespace. Args: name: the name of the logger, which will be prefixed with 'FastMCP.' Returns: a configured logger instance zfastmcp.r) startswithlogging getLoggerr r_/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/utilities/logging.py get_loggers rINFOlevelDEBUGrWARNINGERRORCRITICALloggerenable_rich_tracebacks rich_kwargsc Ks tjjsdS|durtjj}|durtd}td}d|_||t ddt ddi|}| || dd d dl }d dl}t dddd|d t||gd }||t di|} | || d d |jddD]} || qr|||| dS)z Configure logging for FastMCP. Args: logger: the logger to configure level: the log level to use rich_kwargs: the parameters to use for creating RichHandler Nfastmcpz %(message)sFconsoleT)stderrcSs |jduSNexc_inforecordrrrD z#configure_logging..r)r show_path show_levelrich_tracebackstracebacks_max_framestracebacks_suppresscSs |jduSrrr!rrrr#\r$r)rsettings log_enabledrr r Formatter propagatesetLevelrr setFormatter addFiltermcppydanticupdatehandlers removeHandler addHandler) rrrr formatterhandlerr2r3traceback_kwargstraceback_handlerhdlrrrrconfigure_loggingsB        r=c ks||r9tjj}ttd|}td|||d|zdVWtd|||d|dStd|||d|wdVdS)a Context manager to temporarily set log level and restore it afterwards. Args: level: The temporary log level to set (e.g., "DEBUG", "INFO") logger: Optional logger to configure (defaults to FastMCP logger) enable_rich_tracebacks: Whether to enable rich tracebacks **rich_kwargs: Additional parameters for RichHandler Usage: with temporary_log_level("DEBUG"): # Code that runs with DEBUG logging pass # Original log level is restored here r)rrrNr)rr+ log_levelrrupperr=)rrrroriginal_levellog_level_literalrrrtemporary_log_levelfs< rB)rrrrrN _level_to_nocs~eZdZUeeefdBed<eeefdBed<  d deddBdeddBffdd Ze de j de fd d Z ZS) _ClampedLogFilterN min_level max_levelrcsRd|_d|_t|}r|t|f|_t|}r"|t|f|_tdSr)rErFrCgetstrsuper__init__)selfrErF min_level_no max_level_no __class__rrrJsz_ClampedLogFilter.__init__r"r cCsX|jr|j\}}|j|kr||_||_dS|jr*|j\}}|j|kr*||_||_dSdS)NT)rFlevelno levelnamerE)rKr"rMmax_level_namerLmin_level_namerrrfilters    z_ClampedLogFilter.filterNN)__name__ __module__ __qualname__tupleintrH__annotations__rrJrr LogRecordboolrT __classcell__rrrNrrDs rDrErFcCs"t|d|jt||dddS)aClamp the logger to a minimum and maximum level. If min_level is provided, messages logged at a lower level than `min_level` will have their level increased to `min_level`. If max_level is provided, messages logged at a higher level than `max_level` will have their level decreased to `max_level`. Args: min_level: The lower bound of the clamp max_level: The upper bound of the clamp )r)rErF)rTN)_unclamp_loggerr1rD)rrErFrrr _clamp_loggers r`cCs,|jddD] }t|tr||qdS)z/Remove all clamped log filters from the logger.N)filters isinstancerD removeFilter)rrTrrrr_s   r_)rNNrU)"__doc__ contextlibr typingrrr rich.consoler rich.loggingrtyping_extensionsrrrHLoggerrrZr]r=contextmanagerrBrrrrrrCdictr[FilterrDr`r_rrrrst     I 6 +