o ^i@sdZddlmZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z mZddlmZmZdd lmZd d lmZd ZerHddlZGd ddeejZdddZdddZGdddeZdefdefgZdS) zOA Websocket Handler for emitting Jupyter server events. .. versionadded:: 2.0 ) annotationsN)datetime) TYPE_CHECKINGAnyOptionalcast) ensure_async)web websocket) authorizedws_authenticated)JupyterHandler) APIHandlereventscsJeZdZdZeZddZefddZdddZ ddZ ddZ Z S)SubscribeWebsocketz+Websocket handler for subscribing to eventscs6|j}t|j||ddIdH}|stddS)z{Handles authorization when attempting to subscribe to events emitted by Jupyter Server's eventbus. executerNi) current_userr authorizer is_authorizedr HTTPError)selfuserr rb/home/ubuntu/hpml_nyu/venv/lib/python3.10/site-packages/jupyter_server/services/events/handlers.pypre_get#s  zSubscribeWebsocket.pre_getcs@t|IdHtj|i|}|dur|IdHdSdS)zGet an event socket.N)rrsuperget)rargskwargsres __class__rrr0s zSubscribeWebsocket.getlogger!jupyter_events.logger.EventLogger schema_idstrdatadict[str, Any]returnNonecs(tdd|i|}|t|dS)zWrite an event message.r%Nr)dict write_messagejsondumps)rr#r%r'capsulerrrevent_listener8sz!SubscribeWebsocket.event_listenercC|jj|jddS)zrRoutes events that are emitted by Jupyter Server's EventBus to a WebSocket client in the browser. listenerN) event_logger add_listenerr0rrrropen?szSubscribeWebsocket.opencCr1)zHandle a socket close.r2N)r4remove_listenerr0r6rrron_closeEszSubscribeWebsocket.on_close)r#r$r%r&r'r(r)r*) __name__ __module__ __qualname____doc__ AUTH_RESOURCE auth_resourcerr rr0r7r9 __classcell__rrr!rrs  rr'r(registry-jupyter_events.schema_registry.SchemaRegistryr)r*cCshd}|D]}||vrd|d}t|qtt|d}||}ttt|d}|j|krDd|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%rCzUnregistered version: u≠z for ``N) Exceptionrr&rrC)r'rA required_keyskeymessager%schemarCrrrvalidate_modelJs   rJOptional[datetime]c CsPzd|vrt|dd}W|Sd}W|Sty'}ztdd|d}~ww)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)rstrptimerEr r)r'rLerrr get_timestamp_s  rPc@s&eZdZdZeZejeddZ dS) EventHandlerzREST api handler for eventsc s|}|durtddz*t||jj|jjtt| dtd| dt |d| d| WdSt yN}z tdt||d}~ww) zEmit an event.NrMzNo JSON data providedr%r(r')r%r'timestamp_override) get_json_bodyr rrJr4schemasemitrr&rrP set_statusfinishrE)rpayloadrOrrrpostvs"  zEventHandler.postN) r:r;r<r=r>r?r authenticatedr rZrrrrrQqs rQz /api/eventsz/api/events/subscribe)r'r(rArBr)r*)r'r(r)rK)r= __future__rr-rtypingrrrrjupyter_core.utilsrtornador r jupyter_server.auth.decoratorr r jupyter_server.base.handlersr base.handlersrr>jupyter_events.loggerjupyter_eventsWebSocketHandlerrrJrPrQdefault_handlersrrrrs.       /