XhdZddlmZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z mZddlmZmZdd lmZd d lmZd ZerddlZGd deej2Z ddZddZGddeZdefdefgZy)zOA Websocket Handler for emitting Jupyter server events. .. versionadded:: 2.0 ) annotationsN)datetime) TYPE_CHECKINGAnyOptionalcast) ensure_async)web websocket) authorizedws_authenticated)JupyterHandler) APIHandlereventscZeZdZdZeZdZefdZ ddZ dZ dZ xZ S)SubscribeWebsocketz+Websocket handler for subscribing to eventscK|j}t|jj||ddd{}|st j dy7w)z{Handles authorization when attempting to subscribe to events emitted by Jupyter Server's eventbus. executerNi) current_userr authorizer is_authorizedr HTTPError)selfuserr s [/var/www/html/myenv/lib/python3.12/site-packages/jupyter_server/services/events/handlers.pypre_getzSubscribeWebsocket.pre_get#sX   ' OO ) )$i J  --$ $ s7AAAcKt|jd{t| |i|}| |d{yy7!7w)zGet an event socket.N)r rsuperget)rargskwargsres __class__s rr zSubscribeWebsocket.get0sL4<<>***gk4*6* ?II  + sAAAAAAcnKtdd|i|}|jtj|yw)zWrite an event message. schema_idN)dict write_messagejsondumps)rloggerr&datacapsules revent_listenerz!SubscribeWebsocket.event_listener8s133d3 4::g./s35cP|jj|jy)zrRoutes events that are emitted by Jupyter Server's EventBus to a WebSocket client in the browser. listenerN) event_logger add_listenerr/rs ropenzSubscribeWebsocket.open?s! &&0C0C&DcP|jj|jy)zHandle a socket close.r1N)r3remove_listenerr/r5s ron_closezSubscribeWebsocket.on_closeEs ))43F3F)Gr7)r,z!jupyter_events.logger.EventLoggerr&strr-dict[str, Any]returnNone) __name__ __module__ __qualname____doc__ AUTH_RESOURCE auth_resourcerr r r/r6r: __classcell__)r$s@rrrsW6!M %070DG0O]0 0E Hr7rcThd}|D]}||vsd|d}t|tt|jd}|j|}ttt|jd}|j|k7r!d|d|jd|d }t|y ) zoValidates for required fields in the JSON request body and verifies that a registered schema/version exists>r-versionr&z Missing `z` in the JSON request body.r&rGzUnregistered version: u≠z for ``N) Exceptionrr;r rG)r-registry required_keyskeymessager&schemarGs rvalidate_modelrOJs 5M% d?!#&ABGG$ $%S$((;/0I\\) $F$sDHHY/01G ~~ *7+S8J&QZP[[\]  !r7c d|vrtj|dd}|Sd} |S#t$r}tjdd|d}~wwxYw)z+Parses timestamp from the JSON request body timestampz%Y-%m-%dT%H:%M:%S%zZNzFailed to parse timestamp from JSON request body, an ISO format datetime string with UTC offset is expected, for example, 2022-05-26T13:50:00+05:00Z)rstrptimerIr r)r-rQes r get_timestamprU_sn  $  ))${*;=STI I  mm  7   s&& A AA cBeZdZdZeZejedZ y) EventHandlerzREST api handler for eventsc K|j}|tjdd t||jj |jj tt|jdtd|jdt||jd|jy#t$r%}tjdt||d}~wwxYww) zEmit an event.NrRzNo JSON data providedr&r<r-)r&r-timestamp_override) get_json_bodyr rrOr3schemasemitrr;r rU set_statusfinishrI)rpayloadrTs rpostzEventHandler.postvs$$& ?--%<= = 4 7D$5$5$=$= >    " "sGKK $<=*GKK,?@#0#9 #  OOC KKM 4--SV,! 3 4s))C<BC C< C9 C44C99C<N) r?r@rArBrCrDr authenticatedr rar'r7rrWrWqs*%!M44r7rWz /api/eventsz/api/events/subscribe)r-r<rJz-jupyter_events.schema_registry.SchemaRegistryr=r>)r-r<r=zOptional[datetime])rB __future__rr*rtypingrrrrjupyter_core.utilsr tornador r jupyter_server.auth.decoratorr r jupyter_server.base.handlersr base.handlersrrCjupyter_events.loggerjupyter_eventsWebSocketHandlerrrOrUrWdefault_handlersr'r7rrns # 55+"F7'  ,H ,H^! !$Q! !*$4:4:\"12r7