Ki dUdZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z ddl Z dedejfd Z dd ed ezd ejdzd edzdeddf dZej( dd edzd ejdzd edzdefdZej,ej.ej0ej2ej4ddZeed dzedzfed<Gddej<Z dd ejded dzded dzddfdZ d ejddfdZ!y)zLogging utilities for FastMCP.N)AnyLiteralcast)Console) RichHandler)overridenamereturnc|jdrtj|Stjd|S)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 s e/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/utilities/logging.py get_loggerrs9 z"  d++   HTF"3 44levelDEBUGINFOWARNINGERRORCRITICALloggerenable_rich_tracebacks rich_kwargsc tjjsy|tjj}|t j d}t j d}d|_|j|td dtdi|}|j||jdd dl }d dl}tddd|d t||gd }|j|td i|} | j|| jd |j ddD]} |j#| |j%||j%| y)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)stderrc|jduSNexc_inforecords rz#configure_logging..DsV__%<rr)r show_path show_levelrich_tracebackstracebacks_max_framestracebacks_suppressc|jduSr"r#r%s rr'z#configure_logging..\svd/Jr)rsettings log_enabledrrr Formatter propagatesetLevelrr setFormatter addFiltermcppydanticupdatehandlers removeHandler addHandler) rrrr formatterhandlerr7r8traceback_kwargstraceback_handlerhdlrs rconfigure_loggingrBs`    ' '%!(!1!1!H!H ~""9-!!-0IF OOEt$ G # <= $'1!" 'h7 K(#7&67""9- JK"#T"# g '(rc +K|r_tjj}ttd|j }t d|||d| dt d|||d|ydy#t d|||d|wxYww)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/)rr0 log_levelrrupperrB)rrrroriginal_levellog_level_literals rtemporary_log_levelrHfs*  ))33! C D KKM   ##9      $'=      $'=   sA A=A)A=)A::A=)rrrrrN _level_to_noceZdZUeeefdzed<eeefdzed< d deddzdeddzffd Ze de jde fdZ xZS) _ClampedLogFilterN min_level max_levelrcd|_d|_tj|x}r|t |f|_tj|x}r|t |f|_t |yr")rLrMrIgetstrsuper__init__)selfrLrM min_level_no max_level_no __class__s rrRz_ClampedLogFilter.__init__sj'++I6 6< 6*C N;DN'++I6 6< 6*C N;DN rr&r c|jr-|j\}}|j|kDr||_||_y|jr-|j\}}|j|kr||_||_yy)NT)rMlevelno levelnamerL)rSr&rUmax_level_namerTmin_level_names rfilterz_ClampedLogFilter.filtersk >>+/>> (L.~~ ,!-#1  >>+/>> (L.~~ ,!-#1 rNN)__name__ __module__ __qualname__tupleintrP__annotations__rrRrr LogRecordboolr\ __classcell__)rVs@rrKrKsS#X%%S#X%%  JK JK  "W..4rrKrLrMcVt||jt||y)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)rLrM)r\N)_unclamp_loggerr6rK)rrLrMs r _clamp_loggerris%6" - YWXrcp|jddD]$}t|ts|j|&y)z/Remove all clamped log filters from the logger.N)filters isinstancerK removeFilter)rr\s rrhrhs3..#( f/ 0    '(r)rNNr])"__doc__ contextlibrtypingrrr rich.consoler rich.loggingrtyping_extensionsrrrPLoggerrrbrerBcontextmanagerrHrrrrrrIdictrcFilterrKrirhr/rrrxs$%% $& 5S 5W^^ 5 MS$(*.F) B Cc IF) NNT !F)!4KF) F)  F)R %)*.0 :0 NNT !0!4K0 00l]] LL ]]    d ;