Mh>dZddlmZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl m Z ddlmZddlmZddlmZmZmZmZdd lmZmZd d lmZd d lmZd d lmZd dl m!Z!e dZ"ee"edkrddl#m$Z$nddl%m$Z$d Z&Gdde'Z(Gdde)Z*Gdde)Z+e j,de(Gdde)Z-GddeZ.dS)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) JsonFormatterceZdZdZdS)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__U/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/jupyter_events/logger.pyrr%rrceZdZdZdS) ModifierErrorzRAn exception to raise when a modifier does not show the proper signature. Nrrrrr"r"+r rr"ceZdZdZdS)CoreMetadataErrorz:An exception raised when event core metadata is not valid.Nrrrrr$r$1sDDDDrr$onceceZdZdZdS) ListenerErrorzRAn exception to raise when a listener does not show the proper signature. Nrrrrr'r':r rr'ceZdZUdZeddddZeedZ e id Z e id Z e id Z eZd ed <dfd Z d?d@fd ZdAd!ZdBd$ZdBd%Zdd&dCd+Zdd&dCd,Zddd-dDd2Zdd&dEd3Zdd4dFd;ZxZS)G 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_listenersreturn list[t.Any]c>Ktj|jddid{VS)z#Gather all of the active listeners.return_exceptionsTN)asynciogatherr.selfs rgather_listenerszEventLogger.gather_listenersds0^T%;TtTTTTTTTTTrschemasrctSNrr5s r_default_schemaszEventLogger._default_schemashsrargst.AnykwargsNonecjtj|i|tdztt |z}t j||_d|j_|j t j |j r|j D]}| |dSdS)zInitialize the logger..FN) super__init__rstridlogging getLogger_logger propagatesetLevelINFOhandlersregister_handler)r6r<r>log_namehandler __class__s rrCzEventLogger.__init__ls $)&)))c>C4MM1(22 !&  gl+++ = /= / /%%g.... / / / /rcfgr section_nameslist[str] | Nonetraitsc||}|dgdfd }||d<td|i}t|dddS) zLoad EventLogger traits from a Config object, patching the handlers trait in the Config object to avoid deepcopy errors. rLr/list[logging.Handler]cSr:r)rLsr get_handlersz.EventLogger._load_config..get_handlerssOrr)N)rRrT)r/rV)_find_my_configpopr rB _load_config) r6rQrRrTmy_cfgrXeventlogger_cfgrLrPs @rr[zEventLogger._load_config~s%%c***0**Z*D*D      *z!-!899 _DNNNNNrschemarc|j|}|j}||jvrt |j|<||jvrt |j|<||jvrt |j|<dSdS)zRegister this schema with the schema registry. Get this registered schema using the EventLogger.schema.get() method. N)r8registerrE _modifiersset_modified_listeners_unmodified_listeners)r6r^ event_schemakeys rregister_event_schemaz!EventLogger.register_event_schemas |,,V44 o do % %#&55DOC d. . .,/EED $S ) d0 0 0.1eeD &s + + + 1 0rrOlogging.Handlercd fd }t|}||j||jvrj|dSdS) zRegister a new logging handler to the Event Logger. All outgoing messages will be formatted as a JSON string. recordr=r>r/rDcj|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)r8get propertiesjsondumps)rjr>r^r6s r_handle_message_fieldz;EventLogger.register_handler.._handle_message_fieldsM\%%f\&:;;F 1119%:f//// /r)json_serializerN)rjr=r>r=r/rD)r setFormatterrH addHandlerrLappend)r6rOrr formatters` rrMzEventLogger.register_handlers 0 0 0 0 0 0"1    Y''' ((( $- ' ' M  ) ) ) ) ) ( 'rc|j|||jvr|j|dSdS)z>Remove a logging handler from the logger and list of handlers.N)rH removeHandlerrLremove)r6rOs rremove_handlerzEventLogger.remove_handlersI ""7+++ dm # # M  ) ) ) ) ) $ #r) schema_idr| str | Nonemodifier5t.Callable[[str, dict[str, t.Any]], dict[str, t.Any]]c>t|sd}t||rH|j|t }||||j|<dS|jD]*}|||kr |j||+dS)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 TypeErrorrarnrbadd)r6r|r~msg modifiersid_s r add_modifierzEventLogger.add_modifiers"!! !1CC..   ++Isuu==I MM( # # #)2DOI & F? 3 3C C9$4$4$((222 3 3rc|r"|j||dS|jjD]B}|j|||j||CdS)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)radiscardr8 schema_ids)r6r|r~s rremove_modifierzEventLogger.remove_modifiers"  = OI & . .x 8 8 8 8 8"\4 = =  *228<<< *228<<<< = =r)modifiedr|rboollistenerQt.Callable[[EventLogger, str, dict[str, t.Any]], t.Coroutine[t.Any, t.Any, None]]c"t|sd}t||r|rH|j|t }||||j|<dS|j|t }||||j|<dS|jjD]M}|||krC|r!|j||-|j||NdS)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) rrrcrnrbrrdr8r)r6rr|rr listenersrs r add_listenerzEventLogger.add_listeners0&!! !1CC..    !488CEEJJ  h'''6?(3266y#%%HHI MM( # # #4=D &y 1 F<* B BC C9$4$4B,S155h????.s377AAA  B Brc,|rB|j|||j||dS|jjD]B}|j|||j||CdS)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)rcrrdr8r)r6r|rs rremove_listenerzEventLogger.remove_listener,s"  H  $Y / 7 7 A A A  &y 1 9 9( C C C C C"\4 H H (3;;HEEE*95==hGGGG H Hr)timestamp_overriderDdatadict[str, t.Any]rdatetime | Nonedict[str, t.Any] | Nonecjs6j|sj|sdS|jvr!t j|dtddSj|}tj |}j |j D]}|||}j|j rj ||j |||tjtjn|}|dz||jt&d} t)j|n#t,$r } t.| d} ~ wwxYw||j|dfd } j|D]Q} t7j| ||} j| | | Rj|D]W} t7j| ||} j| dfd } | | X|S)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)r|r)tzZ) __timestamp__rl__schema_version____metadata_version__taskasyncio.Task[t.Any]r/r?c|}|rj|j|dSr: exceptionlogerrorr.rrerrr6s r_listener_task_donez-EventLogger.emit.._listener_task_donesM..""C $s###  " * *4 0 0 0 0 0r)loggerr|rc|}|rj|j|dSr:rrs rrz-EventLogger.emit.._listener_task_donesMnn&&(HNN3'''&..t44444r)rrr/r?) rLrcrnrdr8warningswarnrcopydeepcopyrarEvalidate_eventrnowrutc isoformatrEVENTS_METADATA_VERSIONrvalidaterr$updaterHinfor3 create_taskr.radd_done_callback) r6r|rrr^ modified_datar~ timestampcapsulerrrrs ` remitzEventLogger.emitGs,  ,00;; .229==  4 DL ( ( M222$     4!!),, d++  2 N NH$Hy}MMMMM  %fi 0 9 L ' ' 4 8 8 8 ##I}===.@-GHLHL ) ) ) )M_ '0022S8#"(.$;     - ) 27 ; ; ; ; - - -# , - }%%% '""" 1 1 1 1 1 10; 8 8H&'&D  " & &t , , ,  " "#6 7 7 7 729= 8 8H&xxtyW['\'\'\]]D  " & &t , , , 5 5 5 5 5 5  " "#6 7 7 7 7s E E7*E22E7)r/r0)r/r)r<r=r>r=r/r?)NN)rQr rRrSrTrSr/r?)r^rr/r?)rOrhr/r?)r|r}r~rr/r?)rrr|r}rrr/r?)r|r}rrr/r?)r|rDrrrrr/r)rrrrrtagrLr rr8r rarcrdr r.__annotations__r7r r;rCr[rgrMr{rrrrr __classcell__)rPs@rr)r)@sx  cc h GbQRRRJ$r(cddd D R36#%%7777UUUU WY    //////*+/#' OOOOOOO,4444 ****4****!% 3 3 3 3 3 3J!%======: $ *B*B*B*B*B*B^!%HHHHHH8`dssssssssssrr))/r __future__rr3rrprFtypingtrrrimportlib.metadatar jsonschemarpackaging.versionr traitletsr r r r traitlets.configr rr^rschema_registryrrTr validatorsr version_infopythonjsonlogger.jsonrpythonjsonlogger.jsonloggerrWarningr Exceptionr"r$ simplefilterr'r)rrrrs#""""" ''''''''&&&&&&&&&&&&######22222222222288888888++++++555555w+,, 5%%..((3333333999999' I EEEEE EEE f1222I zzzzz%zzzzzr