o i,@sdZddlZddlmZddlmZmZmZddlmZddl m Z m Z ddl m Z ddlmZd d lmZd ZGd d d ee jZdeeefddfddZdeeefdeefddZGdddeZdefdefgZdS)zOA Websocket Handler for emitting Jupyter server events. .. versionadded:: 2.0 N)datetime)AnyDictOptional) EventLogger)web websocket) authorized)JupyterHandler) APIHandlereventscsVeZdZdZeZddZfddZdede de d d fd d Z d dZ ddZ ZS)SubscribeWebsocketz+Websocket handler for subscribing to eventscCsD|j}|dur|jdtd|j||dds tddS)zHandles authentication/authorization when attempting to subscribe to events emitted by Jupyter Server's eventbus. Nz*Couldn't authenticate WebSocket connectioniexecuter ) current_userlogwarningr HTTPError authorizer is_authorized)selfuserrb/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/jupyter_server/services/events/handlers.pypre_gets   zSubscribeWebsocket.pre_getcs*|tj|i|}|IdHdS)zGet an event socket.N)rsuperget)rargskwargsres __class__rrr+szSubscribeWebsocket.getlogger schema_iddatareturnNcs(tdd|i|}|t|dS)zWrite an event message.r#Nr)dict write_messagejsondumps)rr"r#r$capsulerrrevent_listener1sz!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_listenerr+rrrropen6szSubscribeWebsocket.opencCr,)zHandle a socket close.r-N)r/remove_listenerr+r1rrron_close<szSubscribeWebsocket.on_close)__name__ __module__ __qualname____doc__ AUTH_RESOURCE auth_resourcerrrstrr&r+r2r4 __classcell__rrr rrs rr$r%cCs2hd}|D]}||vrtdd|dqdS)z6Validates for required fields in the JSON request body>r$versionr#z Missing `z` in the JSON request body.N)rr)r$ required_keyskeyrrrvalidate_modelAs rAc 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%zZNr>zFailed 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)rstrptime Exceptionrr)r$rBerrr get_timestampIs  rFc@s&eZdZdZeZejeddZ dS) EventHandlerzREST api handler for eventsc s|}|durtddz!t||jj|d|dt|d|d| WdSt yE}z tdt ||d}~ww) zEmit an event.Nr>zNo JSON data providedr#r$)r#r$timestamp_overridei) get_json_bodyrrrAr/emitrrF set_statusfinishrDr;)rpayloadrErrrpost`s"  zEventHandler.postN) r5r6r7r8r9r:r authenticatedr rOrrrrrG[s rGz /api/eventsz/api/events/subscribe)r8r(rtypingrrrjupyter_eventsrtornadorrjupyter_server.authr jupyter_server.base.handlersr base.handlersr r9WebSocketHandlerrr;rArFrGdefault_handlersrrrrs(      -