2BfQddlZddlZddlmZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZmZddlmZddlmZddlmZd d lmZmZGd d ZGd deZGddeZy)N)time)urlsplit)ConnectionType WSConnection) AcceptConnectionRejectConnectionCloseConnectionMessageRequestPingPong TextMessage BytesMessage)PerMessageDeflate) CloseReason)LocalProtocolError)ConnectionErrorConnectionClosedcLeZdZ d dZdZdZdZd dZd dZdZ d Z d Z y)AioBaseNcbd|_||_||_||_||_d|_g|_d|_d|_d|_ |tjk(|_ tj|_d|_d|_d|_t'j(|_d|_d|_y)NTrF) subprotocolconnection_type receive_bytes ping_intervalmax_message_size pong_received input_bufferincoming_messageincoming_message_len connectedrSERVER is_serverrNO_STATUS_RCVD close_reason close_messagersockwsockasyncioEventeventwstask)selfrrrrs V/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/simple_websocket/aiows.py__init__zAioBase.__init__s .** 0! $$%!)^-B-BB'66!  ]]_  cKt|j|_|jd{|js t t j|j|_ y7CwN) rrr- handshaker"rr* create_task_taskr.r/s r0connectzAioBase.connect1sUt334nn~~!# #'' 5 s.A6A4AA6c Kywr4r8s r0r5zAioBase.handshake9s  scbK|js t|j|jt |t r&|j jt|}n.|j jtt|}|jj|yw)zSend data over the WebSocket connection. :param data: The data to send. If ``data`` is of type ``bytes``, then a binary message is sent. Else, the message is sent in text format. )dataN) r"rr&r' isinstancebytesr-sendr rstrr)write)r/r=out_datas r0r@z AioBase.send=sv~~"4#4#4d6H6HI I dE "ww||G$67Hww||KSY$?@H "sB-B/cK|jrw|jsk tj|jj |d{|jj|jr |jsk |jjdS7S#tj $rYywxYw#t$rYnwxYw|js t|j|jyw)a}Receive data over the WebSocket connection. :param timeout: Amount of time to wait for the data, in seconds. Set to ``None`` (the default) to wait indefinitely. Set to 0 to read without blocking. The data received is returned, as ``bytes`` or ``str``, depending on the type of the incoming message. timeoutNr) r"rr*wait_forr,wait TimeoutErrorclearpop IndexErrorrr&r')r/rFs r0receivezAioBase.receiveLsnnT%6%6 &&tzz'8'JJJ JJ    nnT%6%6  $$((+ + K''      ~~"4#4#4d6H6HI Is^C92B$B"B$2C9B=!C9"B$$B:7C99B::C9= C C9C  0C9cDK|js t|j|j|jj t |xstj|} |jj|d|_y#t$r Yd|_ywxYww)a@Close the WebSocket connection. :param reason: A numeric status code indicating the reason of the closure, as defined by the WebSocket specification. The default is 1000 (normal closure). :param message: A text message to be sent to the other side. FN) r"rr&r'r-r@r rNORMAL_CLOSUREr)rBBrokenPipeError)r/reasonmessagerCs r0closez AioBase.closecs~~"4#4#4d6H6HI I77<<  0k00'!;<  JJ  X &   s*A$B 'B B  B B BB cyr4r;)r/requests r0choose_subprotocolzAioBase.choose_subprotocolusr2c\Kd}|jrt|jz}|jr d}|rt}d}||kDrHd} tj|j j |j||z d{}|r|js*|jtjdd{nd|_ |jj|jj!t#t%|||jz}|j j |jd{}t'|dk(r t) |jj1||j3d{|_|jr|jjy7B#tj$rd}YXwxYw7*7#t(t*f$r$d|_|j,j/YqwxYw7w)Nr2TFrEzPing/Pong timeoutrQrRr)rrr"r*rGr(readrrIrrSrPOLICY_VIOLATIONr)rBr-r@r maxlenOSErrorConnectionResetErrorr,set receive_data_handle_events)r/ next_pingin_datanow timed_outs r0r7z AioBase._task{s   !3!33Inn &C $I 3$) -,3,<,< $ 0B0B C(1C-9'9G !#11"&**'2'C'C(;#-#==="-2* ((df)=>$'Y$7$:L:L$L  $(JJOOD4F4F$GGGw<1$!)O% GG  )#'#6#6#88DNGnnH 7'9 '33-(,I- =H12 !&    9s3H,G4AGGG2G4G/G4 H, AG4'H,)(G4G2G4./H,H*H,5H,GG,(G4+G,,G42G440H'$H,&H''H,c Kd}d}|jjD]} t|trT|j ||_||jj t|j tgz }nt|tr|jr,||jj |jz }|j|_ |j|_d|_|j"j%d}n t|t&r.||jj |jz }nt|t(r d|_nt|t,t.fr|xj0t3|j4z c_|j6rn|j0|j6kDrU||jj tt8j:dz }|j"j%d}n|j<|j4|_nt|t,r~t|j<t>s6t?|j<|j4zjA|_n|xj<|j4jAz c_nat|j<t>s(t?|j<|j4z|_n|xj<|j4z c_|jBs#t|j<tDtFfr&|jHjK|j<nrt|t,r4|jHjK|j<jMn.|jHjKtG|j<d|_d|_|j"j%n |r|jPjS||S#tN$r"d}|j"j%d}YQwxYww)NTr2)r extensionsFzMessage is too bigr)*r-eventsr>r rVrr@rrr r$responsecoder&rQr'r"r,r_r r rrrr!r\r=rrMESSAGE_TOO_BIGr bytearrayencodemessage_finishedrAr?rappenddecoderr)rB)r/ keep_goingrCr,s r0razAioBase._handle_eventssa WW^^%G #EF #eW-'+'>'>u'ED$ -=$($4$4$5$7#8.:!;;H 7~~ DGGLL1A$BB(- D%).D&%*DNJJNN$!&Jt, U^^-= >>Ht,)-D& \'BC--UZZ@-,, 558M8MM DGGLL'779M2O%PP (%* ,,4 16 -#E;7)$*?*?K4=!%!6!6!C K K M5OD1!11UZZ5F5F5HH1)$*?*?K4= $ 5 5 B5DD1!11UZZ?1 11 !$"7"7#uF))001F1FG#E;7))00 1188:<))00t7L7L1MN,0D)01D-JJNN$GG #P  JJ  X & & #  "  #sC#Q6HQ&Q6)DQQ6C"Q&"Q6'Q3/Q62Q33Q6NNNr4NN) __name__ __module__ __qualname__r1r9r5r@rMrSrVr7rar;r2r0rrs8;?6:06  #J.$ )VMr2rcVeZdZdZ dfd Ze ddZfdZdZdZ xZ S) AioServerzThis class implements a WebSocket server. Instead of creating an instance of this class directly, use the ``accept()`` class method to create individual instances of the server, each bound to a client request. ct|tj|||||_i|_|xsg|_t|j tr|j g|_d|_ y)Nrrrrunknown) superr1rr#rUheaders subprotocolsr>rAmode)r/rUrrrr __class__s r0r1zAioServer.__init__sk )>)>'4'4*:  <  (.B d'' -!%!2!2 3D  r2c K|r|s|r td|r|s|r td|r,ddlm} | j|d|d|d|d{S||||d |||| } | j d{| S7/7w) aAccept a WebSocket connection from a client. :param aiohttp: The request object from aiohttp. If this argument is provided, ``asgi``, ``sock`` and ``headers`` must not be set. :param asgi: A (scope, receive, send) tuple from an ASGI request. If this argument is provided, ``aiohttp``, ``sock`` and ``headers`` must not be set. :param sock: A connected socket to use. If this argument is provided, ``aiohttp`` and ``asgi`` must not be set. The ``headers`` argument must be set with the incoming request headers. :param headers: A dictionary with the incoming request headers, when ``sock`` is used. :param subprotocols: A list of supported subprotocols, or ``None`` (the default) to disable subprotocol negotiation. :param receive_bytes: The size of the receive buffer, in bytes. The default is 4096. :param ping_interval: Send ping packets to clients at the requested interval in seconds. Set to ``None`` (the default) to disable ping/pong logic. Enable to prevent disconnections when the line is idle for a certain amount of time, or to detect unresponsive clients and disconnect them. A recommended interval is 25 seconds. :param max_message_size: The maximum size allowed for a message, in bytes, or ``None`` for no limit. The default is ``None``. z1aiohttp argument cannot be used with asgi or sockz1asgi argument cannot be used with aiohttp or sockr) WebSocketASGIrrN)aiohttpsockr~)rrrr) ValueErrorasgiraccept_accept) clsrrrr~rrrrrr-s r0rzAioServer.accept s@ $% % W$% %  +&--d1gtAwQ;G.II IWdwG*-,"24jjl I s$A BA>(B7B8BBcK|jdr+|jd}|jd|_d|_nv|jdr\|jdjj dj }|jdj|_d|_n t dtj|d{\|_ |_ t|1d{y7(7w)Nrr~customrsocketzInvalid request)r) rUr~r transportget_extra_infoduprr*open_connectionr(r)r}r9)r/rrs r0rzAioServer._accept<s << <<'D<< 2DL DI \\) $<< *44CC#% << 2::DL!DI./ /'.'>'>D'I!I DJgo"Js$C C: C6"C:0C81C:8C:cKd}|jjD]\}}||d|djz } |dz }|jj ||j d{|_y7 w)NsGET / HTTP/1.1 z: z s )r~itemsrmr-r`rar")r/rcheadervalues r0r5zAioServer.handshakeMs}'!\\//1 9MFE &E7$/668 8G 97 W%#22444sA1B3A>4 BcJ|jD]}||jvs|cSy)aChoose a subprotocol to use for the WebSocket connection. The default implementation selects the first protocol requested by the client that is accepted by the server. Subclasses can override this method to implement a different subprotocol negotiation algorithm. :param request: A ``Request`` object. The method should return the subprotocol to use, or ``None`` if no subprotocol is chosen. Nr)r/rUrs r0rVzAioServer.choose_subprotocolUs2#// #Kd///"" #r2rr)NNNNNrsNN) rurvrw__doc__r1 classmethodrrr5rV __classcell__rs@r0ryrysD BF6: FJJN&*//b "5r2ryc`eZdZdZ dfd Ze ddZfdZdZd fd Z xZ S) AioClientzThis class implements a WebSocket client. Instead of creating an instance of this class directly, use the ``connect()`` class method to create an instance that is connected to a server. ct |tj|||||_||_t |}|jdv|_|j|_ |jxs|jrdnd|_ |j|_ |jr"|xjd|jzz c_ |xsg|_t|jt r|jg|_g|_t|t$r6|j'D]"\} } |j"j)| | f$yt|t*r||_yy)Nr{)httpswssiP?)r}r1rCLIENTurl ssl_contextrscheme is_securehostnamehostportpathqueryrr>rAextra_headeersdictrrolist) r/rrr~rrrr parsed_urlkeyrrs r0r1zAioClient.__init__ns/ )>)>'4'4*:  <&c] #**.>>'' OOFt~~2 OO    IIz/// /I(.B d'' -!%!2!2 3D   gt $%mmo 9 U##**C<8 9  &")D 'r2c `K||||||||} | jd{| S7w)aReturns a WebSocket client connection. :param url: The connection URL. Both ``ws://`` and ``wss://`` URLs are accepted. :param subprotocols: The name of the subprotocol to use, or a list of subprotocol names in order of preference. Set to ``None`` (the default) to not use a subprotocol. :param headers: A dictionary or list of tuples with additional HTTP headers to send with the connection request. Note that custom headers are not supported by the WebSocket protocol, so the use of this parameter is not recommended. :param receive_bytes: The size of the receive buffer, in bytes. The default is 4096. :param ping_interval: Send ping packets to the server at the requested interval in seconds. Set to ``None`` (the default) to disable ping/pong logic. Enable to prevent disconnections when the line is idle for a certain amount of time, or to detect an unresponsive server and disconnect. A recommended interval is 25 seconds. In general it is preferred to enable ping/pong on the server, and let the client respond with pong (which it does regardless of this setting). :param max_message_size: The maximum size allowed for a message, in bytes, or ``None`` for no limit. The default is ``None``. :param ssl_context: An ``SSLContext`` instance, if a default SSL context isn't sufficient. )rr~rrrrN)_connect) rrrr~rrrr thread_class event_classr-s r0r9zAioClient.connects?F<,M"2 Mkkm  s #.,.cnK|jr?|j3tjtjj |_t j|j|j|jd{\|_ |_ t|5d{y7(7w)N)purpose)ssl)rrrcreate_default_contextPurpose SERVER_AUTHr*rrrr(r)r}r9)r/rs r0rzAioClient._connects >>'#&#=#=KK33$5 '.'>'> IItyyd&6&6(8"8 DJgo"8s$BB5B1 "B5+B3,B53B5cK|jjt|j|j|j |j }|jj| |jj|jd{}|jj| t|jj} t!|t"rt%|j&t!|t(s t%d|j*|_d|_y7#t$rYnwxYwحw)N)rtargetr extra_headersTi)r-r@r rrrrr)rBr(rYrr`nextrh StopIterationr>rr status_coderrr")r/rCrcr,s r0r5zAioClient.handshakes77<<TYYtyy595F5F6:6I6I!KL " JJOOD,>,>??G GG  ) TWW^^-. e- .!%"3"34 4E#34!#& & ,,@!   s7BED5E8#D7AE7 EEEEcxKt|||d{|jjy7w)NrX)r}rSr))r/rQrRrs r0rSzAioClient.closes2gm67m;;;  rs[ !0   1.05[[|nnbllr2