-e5vUdZddlmZddlZejeZddlZddlZ ddl m Z m Z m Z ddlmZddlmZmZddlmZmZddlmZdd lmZmZmZd d lmZd d lmZm Z m!Z!d d l"m#Z#d dl$m%Z%d dl&m'Z'ddl(m)Z)ddl*m+Z+e rddl,m-Z-dZ.Gdde+eZ/e'GddZ0dZ1de2d<dS)zA Provide a web socket handler for the Bokeh Server application. ) annotationsN) TYPE_CHECKINGAnycast)urlparse)locksweb)WebSocketClosedErrorWebSocketHandler)settings)check_token_signatureget_session_idget_token_payload)Protocol) MessageError ProtocolErrorValidationError)Message)Receiver) dataclass)ProtocolHandler)AuthRequestHandler)ServerConnection) WSHandlerceZdZUdZded<d.fd ZdZd/d Zej d.d Z d0dZ d1dZ d2dZ d3dZd4dZd5dZ d6d7fd# Zd.d$Zd8d&Zd9d(Zd:d+Zd;d,Zd;d-ZxZS)d|_d|_d|_|d|_d|_t j|_d|_| dd|_ | dd|_ tj |g|Ri|dS)Napplication_contextcompression_level mem_level)receiverhandlerrr# latest_pongrLock write_lock_tokenpop_compression_level _mem_levelsuper__init__)self tornado_appargskw __class__s 5lib/python3.11/site-packages/bokeh/server/views/ws.pyr1zWSHandler.__init__Hs  #%&;#<  *,, "$&&)rB parsed_origin origin_host allowed_hostsalloweds r7 check_originzWSHandler.check_origin]s +***** (( #*0022 (:  % ' ' > : < <==M!/+}==  4 IIEk;  G G G5r8ctd|j}|jdkr#|t d|#|t dt jtj tj j  }t|}d|vr#|t d||dkr#|t d t||jj|jj s9t'|}td |t d  |jj|j|jdS#t0$r&}td |Yd}~dSd}~wwxYw)zR Initialize a connection to a client. Returns: None zWebSocket connection openedbokehz!Subprotocol header is not 'bokeh'Nz'No token received in subprotocol header)tzsession_expiryz$Session expiry has not been providedzToken is expired.)signed secret_keyz*Token for session %r had invalid signaturezInvalid token signaturez"Failed to fully open connection %r)rJinfor,selected_subprotocolclosercalendartimegmdtdatetimenowtimezoneutc timetuplerr rF sign_sessionsrVrrKio_loop add_callback _async_open Exceptiondebug)r2tokenr^payload session_ides r7openzWSHandler.open|s .///   $ / / JJLLL CDD D ] JJLLL IJJ JobkoooAAKKMMNN#E** 7 * * JJLLL FGG G G,- - - JJLLL 344 4&u.2.>.L262B2MOOO ;(..J IIBJ O O O 9:: : ?   $ 1 1$2BDK P P P P P ? ? ? II:A > > > > > > > > >  ?s *F99 G)G$$G) subprotocols list[str] str | Nonectdtd|t|dksdS|d|_|dS)NzSubprotocol header receivedz Supplied subprotocol headers: %rrrr)rJrgtracelenr,)r2rms r7select_subprotocolzWSHandler.select_subprotocolsW /000 4lCCC<  A%%4"1o Ar8dict[str, Any] | NonecL|jdSd|ji}|j |j|d<|S)Nr%r&)r.r/)r2optionss r7get_compression_optionsz!WSHandler.get_compression_optionss6  " *4&(?@ ? &#'?GK r8rhcK t|}|j||j|d{V|j|}t }t ||_t d|t|_ t d||j |||j||_tdnC#t $r6}td|||d}~wwxYw|jjd}||d{VdS)a Perform the specific steps needed to open a connection to a Bokeh session Specifically, this method coordinates: * Getting a session for a session ID (creating a new one if needed) * Creating a protocol receiver and handler * Opening a new ServerConnection and sending it an ACK Args: session_id (str) : A session ID to for a session to connect to If no session exists with the given ID, a new session is made Returns: None NzReceiver created for %rzProtocolHandler created for %rzServerConnection createdz/Could not create new server session, reason: %sACK)rr#create_session_if_neededrequest get_sessionrrr'rJrgrr(rFnew_connectionrrWrrKrYprotocolcreate send_message)r2rhrjsessionr~rkmsgs r7rezWSHandler._async_opensr& '..J*CCJPTP\^cdd d d d d d d d.:::FFGzzH$X..DM II/ : : :*,,DL II6 A A A".==hdNfhoppDO HH/ 0 0 0 0    IIG K K K JJLLLG  o&--e44$$$$$$$$$tsCA?r Message[Any]cK ttj|||d{Vn*#t$rt dYnwxYwdS)z Send a Bokeh Server protocol message to the connected client. Args: message (Message) : a message to send Nz/Failed sending message as connection was closed)rsentrsendr rJwarningr2rs r7rzWSHandler.send_messages K!-"'..w777,,t$$ $ $ $ $ $ $ $ $# K K K KKI J J J J J KtsAA$A-,A-FTbytes | str | dict[str, Any]binarylockedc$K|rb|jd{V5t||d{VddddS#1swxYwYdSt||d{VdS)zj Override parent write_message with a version that acquires a write lock before writing. N)r+acquirer0 write_message)r2rrrr6s r7rzWSHandler.write_message"s  9..00000000 = =gg++GV<<<<<<<<< = = = = = = = = = = = = = = = = = =''''88 8 8 8 8 8 8 8 8 8s)AA"Actd|j|j|j?|jj|j|jdSdS)z2 Clean up when the connection is closed. z/WebSocket connection closed: code=%s, reason=%rN) rJrW close_code close_reasonrrnotify_connection_lostrF client_lost)r2s r7on_closezWSHandler.on_close.sg BDOUYUfggg ? & O # : : < < <   ( ( 9 9 9 9 9 ' &r8Message[Any] | NonecK |j|d{V}|S#tttf$r-}|t |Yd}~dSd}~wwxYwr:)r'consumerrr_protocol_errorr?)r2rrrks r7rzWSHandler._receive7s  M11(;;;;;;;;GNm_=     Q ( ( (44444 s!&A*"A%%A* Any | NonecK |j||jd{V}|S#ttt f$r-}|t|Yd}~dSd}~wwxYwr:)r(handlerrrrrr?)r2rrrks r7rzWSHandler._handle@s ,,WdoFFFFFFFFDKm_=     Q ( ( (44444 s',A0"A++A0rrcKt|tr:|ttt|d{Vn|d|dS)Nzexpected a Message not ) isinstancerrrrr)r2rs r7rzWSHandler._scheduleIsr dG $ $ E##Dt$<$<== = = = = = = = =  !C4!C!C D D Dtr8chtd||d|dS)Nz3Bokeh Server internal error: %s, closing connectioni'rJrKrYrs r7rzWSHandler._internal_errorQ1 GQQQ 5'"""""r8chtd||d|dS)Nz3Bokeh Server protocol error: %s, closing connectioni'rrs r7rzWSHandler._protocol_errorUrr8)r r!)r>r?r r@)rmrnr ro)r rt)rhr?r r!)rrr r!)rrr r!)rrr r!)FT)rrrr@rr@r r!)rrr r)rrr r)rrr r!)rr?r r!)__name__ __module__ __qualname____doc____annotations__r1r=rPr authenticatedrlrsrwrerrrrrrrrrr __classcell__)r6s@r7rrAs('''333333$   > &?&?&?&?P****X%%%%NRRRR"26 9 9 9 9 9 9 9::::############r8rc$eZdZUded<ded<dS)MessageTestPortzlist[Message[Any]]rrN)rrrrr;r8r7rr`s*      r8rzMessageTestPort | Noner)3r __future__rlogging getLoggerrrJrZr]r\typingrrr urllib.parsertornadorr tornado.websocketr r bokeh.settingsr bokeh.util.tokenr rrr~rprotocol.exceptionsrrrprotocol.messagerprotocol.receiverrutil.dataclassesrprotocol_handlerrauth_request_handlerrrr__all__rrrrr;r8r7rs #"""""g!!++++++++++!!!!!!DDDDDDDD$#####UUUUUUUUUU! OOOOOOOOOO''''''))))))))))))......444444.------  V#V#V#V#V#"$4V#V#V#~ !!!!!!! !.2111111r8