MhdZddlmZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z mZddlmZmZdd lmZd d lmZd ZerddlZGd deejZddZddZGddeZdefdefgZdS)zOA Websocket Handler for emitting Jupyter server events. .. versionadded:: 2.0 ) annotationsN)datetime) TYPE_CHECKINGAnyOptionalcast) ensure_async)web websocket) authorizedws_authenticated)JupyterHandler) APIHandlereventscPeZdZdZeZdZefdZdd Z d Z dZ xZ S)SubscribeWebsocketz+Websocket handler for subscribing to eventscK|j}t|j||ddd{V}|st jddS)z{Handles authorization when attempting to subscribe to events emitted by Jupyter Server's eventbus. executerNi) current_userr authorizer is_authorizedr HTTPError)selfuserr s g/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/jupyter_server/services/events/handlers.pypre_getzSubscribeWebsocket.pre_get#sz  ' O ) )$i J J         %-$$ $ % %cKt|d{Vtj|i|}| |d{VdSdS)zGet an event socket.N)r rsuperget)rargskwargsres __class__s rr!zSubscribeWebsocket.get0sl4<<>>*********eggk4*6** ?IIIIIIIII ?rlogger!jupyter_events.logger.EventLogger schema_idstrdatadict[str, Any]returnNonectKtdd|i|}|tj|dS)zWrite an event message.r(N)dict write_messagejsondumps)rr&r(r*capsules revent_listenerz!SubscribeWebsocket.event_listener8sD333d33 4:g../////rcF|j|jdS)zrRoutes events that are emitted by Jupyter Server's EventBus to a WebSocket client in the browser. listenerN) event_logger add_listenerr5rs ropenzSubscribeWebsocket.open?s' &&0C&DDDDDrcF|j|jdS)zHandle a socket close.r7N)r9remove_listenerr5r;s ron_closezSubscribeWebsocket.on_closeEs% ))43F)GGGGGr)r&r'r(r)r*r+r,r-) __name__ __module__ __qualname____doc__ AUTH_RESOURCE auth_resourcerr r!r5r<r? __classcell__)r%s@rrrs65!M % % %0000EEE HHHHHHHrrr*r+registry-jupyter_events.schema_registry.SchemaRegistryr,r-chd}|D]}||vrd|d}t|tt|d}||}ttt|d}|j|kr d|d|jd|d }t|d S) 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(rJzUnregistered version: u≠z for ``N) Exceptionrr)r!rJ)r*rG required_keyskeymessager(schemarJs rvalidate_modelrQJs 544M%% d??B#BBBGG$$ $ S$((;//00I\\) $ $F$sDHHY//0011G ~  ]7]]]]QZ]]]   ! rOptional[datetime]c d|vrtj|dd}nd}n(#t$r}tjdd|d}~wwxYw|S)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)rstrptimerLr r)r*rTes r get_timestamprX_sz  $   )${*;=STTIII m  7     s"% A AA cFeZdZdZeZejedZ dS) EventHandlerzREST api handler for eventsc 4K|}|tjdd t||jj|jtt| dtd| dt|| d| dS#t$r(}tjdt||d}~wwxYw) zEmit an event.NrUzNo JSON data providedr(r+r*)r(r*timestamp_override) get_json_bodyr rrQr9schemasemitrr)r!rX set_statusfinishrL)rpayloadrWs rpostzEventHandler.postvs $$&& ?-%<== = 4 7D$5$= > > >   " "sGKK $<$<==*GKK,?,?@@#0#9#9 #    OOC KKMMMMM 4 4 4-SVV,,! 3 4sB4C%% D/#DDN) r@rArBrCrDrEr authenticatedr rdr/rrrZrZqsF%%!M44Z444rrZz /api/eventsz/api/events/subscribe)r*r+rGrHr,r-)r*r+r,rR)rC __future__rr2rtypingrrrrjupyter_core.utilsr tornador r jupyter_server.auth.decoratorr r jupyter_server.base.handlersr base.handlersrrDjupyter_events.loggerjupyter_eventsWebSocketHandlerrrQrXrZdefault_handlersr/rrrqs #""""" 555555555555++++++""""""""FFFFFFFF777777'''''' !    ,H,H,H,H,H ,H,H,H^!!!!*$44444:444:\"12r