:vjL@dZddlmZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl m Z ddlmZddlmZddlmZmZmZmZdd lmZmZd d lmZd d lmZd d lmZd dl m!Z!e dZ"ee"edk\rddl#m$Z$nddl%m$Z$d Z&Gdde'Z(Gdde)Z*Gdde)Z+e jXde(Gdde)Z-GddeZ.y)z? Emit structured, discrete events when various actions happen. ) annotationsN)datetimetimezone)version)ValidationError)parse)DictInstanceSetdefault)ConfigLoggingConfigurable) SchemaTypeSchemaRegistry)Handlers)JUPYTER_EVENTS_CORE_VALIDATORzpython-json-loggerz3.1.0) JsonFormatterceZdZdZy)SchemaNotRegisteredzzA warning to raise when an event is given to the logger but its schema has not be registered with the EventLogger N__name__ __module__ __qualname____doc__f/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_events/logger.pyrr%rrceZdZdZy) ModifierErrorzRAn exception to raise when a modifier does not show the proper signature. Nrrrrr"r"+r rr"ceZdZdZy)CoreMetadataErrorz:An exception raised when event core metadata is not valid.Nrrrrr$r$1sDrr$onceceZdZdZy) ListenerErrorzRAn exception to raise when a listener does not show the proper signature. Nrrrrr'r':r rr'ceZdZUdZedddj dZeedZ e id Z e id Z e id Z eZd ed <ddZedddZd fd Z d! d"fd Zd#dZd$dZd$dZdd d%dZdd d%dZddd d&dZdd d'dZdd d(dZxZS)) EventLoggera An Event logger for emitting structured events. Event schemas must be registered with the EventLogger using the `register_schema` or `register_schema_file` methods. Every schema will be validated against Jupyter Event's metaschema. NTzA list of logging.Handler instances to send events to. When set to None (the default), all events are discarded. ) default_value allow_nonehelp)configzbThe SchemaRegistry for caching validated schemas and their jsonschema validators. )r,z0A mapping of schemas to their list of modifiers.z9A mapping of schemas to the listeners of modified events.z?A mapping of schemas to the listeners of unmodified/raw events.zset[asyncio.Task[t.Any]]_active_listenersc\Ktj|jddid{S7w)z#Gather all of the active listeners.return_exceptionsTN)asynciogatherr.selfs rgather_listenerszEventLogger.gather_listenersds&^^T%;%;TtTTTTs #,*,schemasctSNrr3s r_default_schemaszEventLogger._default_schemashs rcft||i|tdztt |z}t j ||_d|j_|jjt j|jr#|jD]}|j|yy)zInitialize the logger..FN) super__init__rstridlogging getLogger_logger propagatesetLevelINFOhandlersregister_handler)r4argskwargslog_namehandler __class__s rr=zEventLogger.__init__ls $)&)c>C4M1((2 !&  gll+ ====%%g.) rc|j|}|jdgdfd }||d<td|i}t||ddy)zLoad EventLogger traits from a Config object, patching the handlers trait in the Config object to avoid deepcopy errors. rFcSr8r)rFsr get_handlersz.EventLogger._load_config..get_handlerssOrr)N) section_namestraits)returnzlist[logging.Handler])_find_my_configpopr r< _load_config) r4cfgrPrQmy_cfgrOeventlogger_cfgrFrLs @rrUzEventLogger._load_config~s[%%c**0**Z*D *z!-!89 _DNrc2|jj|}|j}||jvrt |j|<||j vrt |j |<||j vrt |j |<yy)zRegister this schema with the schema registry. Get this registered schema using the EventLogger.schema.get() method. N)r6registerr? _modifiersset_modified_listeners_unmodified_listeners)r4schema event_schemakeys rregister_event_schemaz!EventLogger.register_event_schemas ||,,V4 oo doo %#&5DOOC d.. .,/ED $ $S ) d00 0.1eD & &s + 1rcdfd }t|}|j|jj||jvrjj |yy)zRegister a new logging handler to the Event Logger. All outgoing messages will be formatted as a JSON string. cjj|d}d|jvr|d=tj|fi|S)aEPython's logger always emits the "message" field with the value as "null" unless it's present in the schema/data. Message happens to be a common field for event logs, so special case it here and only emit it if "message" is found the in the schema's property list. __schema__message)r6get propertiesjsondumps)recordrIr_r4s r_handle_message_fieldz;EventLogger.register_handler.._handle_message_fieldsH\\%%f\&:;F 1 119%::f// /r)json_serializerN)rkt.AnyrIrnrRr>)r setFormatterrB addHandlerrFappend)r4rKrl formatters` rrGzEventLogger.register_handlers\ 0"1  Y' ( $-- ' MM  ) (rc|jj|||jvr|jj|yy)z>Remove a logging handler from the logger and list of handlers.N)rB removeHandlerrFremove)r4rKs rremove_handlerzEventLogger.remove_handlers7 ""7+ dmm # MM  ) $r) schema_idc0t|s d}t||rE|jj|t }|j |||j|<y|jD](}|||k(s |j|j |*y)aGAdd a modifier (callable) to a registered event. Parameters ---------- modifier: Callable A callable function/method that executes when the named event occurs. This method enforces a string signature for modifiers: (schema_id: str, data: dict) -> dict: z`modifier` must be a callableN)callable TypeErrorr[rgr\add)r4rwmodifiermsg modifiersid_s r add_modifierzEventLogger.add_modifiers"!1CC.  ++Isu=I MM( #)2DOOI & ??C C9$4$((2#rc|r|j|j|y|jjD]>}|j|j||j|j|@y)aRemove a modifier from an event or all events. Parameters ---------- schema_id: str If given, remove this modifier only for a specific event type. modifier: Callable[[str, dict], dict] The modifier to remove. N)r[discardr6 schema_ids)r4rwr|s rremove_modifierzEventLogger.remove_modifiersb"  OOI & . .x 8"\\44  *228< *228<5r)modifiedrwct|s d}t||r|rE|jj|t }|j |||j|<y|j j|t }|j |||j |<y|jjD]I}|||k(s |r|j|j |,|j |j |Ky)aAdd a listener (callable) to a registered event. Parameters ---------- modified: bool If True (default), listens to the data after it has been mutated/modified by the list of modifiers. schema_id: str $id of the schema listener: Callable A callable function/method that executes when the named event occurs. z`listener` must be a callableN) ryrzr]rgr\r{r^r6r)r4rrwlistenerr} listenersrs r add_listenerzEventLogger.add_listeners&!1CC.  !4488CEJ  h'6?((32266y#%HI MM( #4=D & &y 1 <<**C C9$4,,S155h?..s377A +rc0|r=|j|j||j|j|y|jjD]>}|j|j||j|j|@y)a Remove a listener from an event or all events. Parameters ---------- schema_id: str If given, remove this modifier only for a specific event type. listener: Callable[[EventLogger, str, dict], dict] The modifier to remove. N)r]rr^r6r)r4rwrs rremove_listenerzEventLogger.remove_listener,s"   $ $Y / 7 7 A  & &y 1 9 9( C"\\44 ((3;;HE**95==hG5r)timestamp_overridec js7jjsjjsyjvr t j dtdyjj}tj|}j|jD] }||}j|jrjj|jj||$tjtj n|}|j#dz|j$t&d} t)j*||j1|j2j5|d fd } jD]L} t7j8| | } j:j=| | j?| NjD]R} t7j8| | } j:j=| d fd } | j?| T|S#t,$r } t.| d} ~ wwxYw) a Record given event with schema has occurred. Parameters ---------- schema_id: str $id of the schema data: dict The event to record timestamp_override: datetime, optional Optionally override the event timestamp. By default it is set to the current timestamp. Returns ------- dict The recorded event data Nz has not been registered yet. If this was not intentional, please register the schema using the `register_event_schema` method.) stacklevel)rwdata)tzZ) __timestamp__re__schema_version____metadata_version__c& |j}|r.j j d|j||jj |y#tj$rjj |YywxYw)Nz#Event listener %s failed for %s: %s)exc_info) exceptionr1CancelledErrorr.rlogerrorget_name)taskerrrwr4s r_listener_task_donez-EventLogger.emit.._listener_task_dones nn&9MMO   " " * *4 0)) &&..t4 sA.BB)loggerrwrc|j}|rjj|jj |yr8)rrrr.r)rrr4s rrz-EventLogger.emit.._listener_task_dones8nn&HHNN3'&&..t4r)rzasyncio.Task[t.Any]rRNone) rFr]rgr^r6warningswarnrcopydeepcopyr[r?validate_eventrnowrutc isoformatrEVENTS_METADATA_VERSIONrvalidaterr$updaterBinfor1 create_taskr.r{add_done_callback) r4rwrrr_ modified_datar| timestampcapsulerrrrs `` remitzEventLogger.emitGsO, ,,00;..229= DLL ( MM+22$  !!), d+  2H$y}MM3  % %fii 0 LL ' ' 4 8 ##I}=.@-GHLLHLL )M_ '002S8#"(..$;    - ) 2 27 ; }% '" 1&00;H&&'&D  " " & &t ,  " "#6 7<229=H&&xtyW['\]D  " " & &t , 5  " "#6 7> w -# , -s,I88 J JJ )rRz list[t.Any])rRr)rHrnrIrnrRr)NN)rVr rPlist[str] | NonerQrrRr)r_rrRr)rKzlogging.HandlerrRr)rw str | Noner|z5t.Callable[[str, dict[str, t.Any]], dict[str, t.Any]]rRr)rboolrwrrQt.Callable[[EventLogger, str, dict[str, t.Any]], t.Coroutine[t.Any, t.Any, None]]rRr)rwrrrrRr)rwr>rzdict[str, t.Any]rzdatetime | NonerRzdict[str, t.Any] | None)rrrrrtagrFr rr6r r[r]r^r r.__annotations__r5r r9r=rUrbrGrvrrrrr __classcell__)rLs@rr)r)@s  cc  GbQRJr(cd  R36%/7U Y  /*+/#' O O(O! O  O,4 *4*!% 3 3H 3  3J!%==H =  =: $ *B*B *B d *B  *B^!%HHd H  H8`d}}'7}M\} }rr))/r __future__rr1rrir@typingtrrrimportlib.metadatar jsonschemarpackaging.versionr traitletsr r r r traitlets.configr rr_rschema_registryrrQr validatorsr version_infopythonjsonlogger.jsonrpythonjsonlogger.jsonloggerrWarningr Exceptionr"r$ simplefilterr'r)rrrrs# '&&#228+5+, %.(39' I E E f12I D%Dr