A>qjd8ddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z ddlmZmZmZmZddlZddlZddlmZddlmZdd lmZdd lmZdd lmZdd lm Z m!Z!m"Z"dd l#m$Z$m%Z%m&Z&ddl'm(Z(ddlm)Z)m*Z*m+Z+m,Z,ddl-m.Z.ddl/m0Z0m1Z1ddl-m2Z2gdZ3Gdde0Z4GddejjjlZ7 d ddddddddddde ddddddddejpd d!dZ9y)") annotationsN) AwaitableMappingSequence) TracebackType)AnyCallable CoroutineSelf) basic_auth)ServerExtensionFactory) enable_server_permessage_deflate) CloseCode)validate_subprotocols)SERVERRequestResponse) CONNECTINGOPENEvent)ServerProtocol) LoggerLikeOrigin StatusLike Subprotocol)get_socket_name) Connection broadcast) race_events)r serveServerConnectionServerr ceZdZdZddddd d fdZd dZddef d d Zdfd ZxZ S)r#a :mod:`trio` implementation of a WebSocket server connection. :class:`ServerConnection` provides :meth:`recv` and :meth:`send` methods for receiving and sending messages. It supports asynchronous iteration to receive messages:: async for message in websocket: await process(message) The iterator exits normally when the connection is closed with close code 1000 (OK) or 1001 (going away) or without a close code. It raises a :exc:`~websockets.exceptions.ConnectionClosedError` when the connection is closed with any other code. The ``ping_interval``, ``ping_timeout``, ``close_timeout``, and ``max_queue`` arguments have the same meaning as in :func:`serve`. Args: nursery: Trio nursery. stream: Trio stream connected to a WebSocket client. protocol: Sans-I/O connection. server: Server that manages this connection.   ping_interval ping_timeout close_timeout max_queuec |t ||||||||||_tj|_|||y)Nr))super__init__servertrior request_rcvd) selfnurserystreamprotocolr1r*r+r,r- __class__s U/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/websockets/trio/server.pyr0zServerConnection.__init__BsO     '%'   (,    c:|jj||S)a Create a plain text HTTP response. ``process_request`` and ``process_response`` may call this method to return an HTTP response instead of performing the WebSocket opening handshake. You can modify the response before returning it, for example by changing HTTP headers. Args: status: HTTP status code. text: HTTP response body; it will be encoded to UTF-8. Returns: HTTP response to send to the client. )r7reject)r4statustexts r9respondzServerConnection.respond^s&}}##FD11r:NcHKt|j|jd{|jd}|. |||j}t |t r |d{}|+|jj|j|_nt |t"sJ||_|||j j$d<d}|9 |||j|j }t |t r |d{}|t |t"sJ||_|j j&tjj(k(rO|j*j,r9|jjtjj.d|_|j0t2urU|j5t2 4d{|jj7|j dddd{yyy77#t $rm}||j_|jjdd|jjtjjd}Yd}~d}~wwxYw7#t $rm}||j_|jjdd|jjtjjd}Yd}~d}~wwxYw787#1d{7swYyxYww) z1 Perform the opening handshake. Nzprocess_request failedTexc_infozLFailed to open a WebSocket connection. See server log for more information. r$zprocess_response failedzServer is shutting down. )expected_state)r!r3 stream_closedrequest isinstancer Exceptionr7 handshake_excloggererrorr<http HTTPStatusINTERNAL_SERVER_ERRORacceptresponserheaders status_codeSWITCHING_PROTOCOLSr1closingSERVICE_UNAVAILABLEstater send_context send_response)r4process_requestprocess_response server_headerrOexcs r9 handshakezServerConnection.handshakess,$++T-?-?@@@ << #H* .tT\\BH!(I6)1> $ 4 4T\\ B !(H555 ( (2? %%h/H+ /dllDMMRH!(I6)1>#!(H555 (  ))T__-P-PPKK'' $ 4 4OO770! zzZ',,J,G??MM// >???( $ A$2 25DMM/KK%%&>%N#}}33==E H 4$2 25DMM/KK%%&?$%O#}}33==E H :?????s$L"H L"(H%H&H*A%L"3JJ JCL"LL"&L 9 L"L L"H JA"J=L"JL" J LA"K?9L"?LL" L" LL LL"c|j4t|tsJ||_|jj yt ||y)z. Process one incoming event. N)rErFrr3setr/ process_event)r4eventr8s r9r_zServerConnection.process_eventsG << eW- -- DL    ! ! # G !% (r:)r5z trio.Nurseryr6trio.abc.Streamr7rr1r$r* float | Noner+rbr,rbr-*int | None | tuple[int | None, int | None]returnNone)r=rr>strrdr)rXZCallable[[ServerConnection, Request], Awaitable[Response | None] | Response | None] | NonerYdCallable[[ServerConnection, Request, Response], Awaitable[Response | None] | Response | None] | NonerZ str | Nonerdre)r`rrdre) __name__ __module__ __qualname____doc__r0r?rr\r_ __classcell__)r8s@r9r#r#&sD')%'&(@B// /! /  /$/#/$/>/ /82:  $*!Y? Y? Y? "!Y?" #Y?v ) )r:r#ceZdZdZ d d dZed dZejf ddZ de jdf ddZ dd Z dd Zy)r$a WebSocket server returned by :func:`serve`. Args: listeners: List of Trio listeners accepting new connections. handler: Handler for one connection. It receives a Trio stream. logger: Logger for this server. It defaults to ``logging.getLogger("websockets.server")``. See the :doc:`logging guide <../../topics/logging>` for details. Nc||_||_|tjd}||_t |_tj|_ d|_ y)Nzwebsockets.serverF) listenershandlerlogging getLoggerrIr^all_connectionsr2rhandlers_waiterrS)r4rqrrrIs r9r0zServer.__init__sR # >&&':;F 7:e $zz| r:cx|jDchc] }|jjtur|"c}Scc}w)a Set of active connections. This property contains all connections that completed the opening handshake successfully and didn't start the closing handshake yet. It can be useful in combination with :func:`~broadcast`. )rur7rUr)r4 connections r9 connectionszServer.connectionss>#22 ""((D0    s%7c Ktj4d{|_tj4d{|_|jj t j tj|j|j|jd{|jD]1}|jjdt|j3|j|dddd{dddd{|jj!y77776#1d{7swYFxYw7=#1d{7swYMxYww)N)handler_nurseryzserver listening on %s)r2 open_nurseryr{ serve_nurserystart functoolspartialserve_listenersrrrqrIinforsocketstartedrvr^)r4 task_statuslisteners r9 serve_foreverzServer.serve_forevers9$$& * *$*>((* * *d.@((..%%,, (,(<(< !%HKK$$0(8 ##D)! * * * **   "+ * * * * * * * * * *sE.D9E.ED<EA'E(D>)AE> E E E E.E E.<E>EEE E  E EE.E+E" E+'E.TcK|jjd|jjj d|_|rU|j D]F}|jjtus |jj|j||H|jjd{|jjdy7 w)a Close the server. * Close the TCP listeners. * When ``close_connections`` is :obj:`True`, which is the default, close existing connections. Specifically: * Reject opening WebSocket connections with an HTTP 503 (service unavailable) error. This happens when the server accepted the TCP connection but didn't complete the opening handshake before closing. * Close open WebSocket connections with close code 1001 (going away). ``code`` and ``reason`` can be customized, for example to use code 1012 (service restart). * Wait until all connection handlers have returned. :meth:`aclose` is idempotent. zserver closingTNz server closed)rIrr} cancel_scopecancelrSrur7rUrr{ start_soonacloservwait)r4close_connectionscodereasonrxs r9rz Server.aclose1s2 )* ''..0  "22 U &&,,4((33J4E4EtVT U ""''))) ) *sA4C 7AC =C>!C cK|SwN)r4s r9 __aenter__zServer.__aenter__]s  sc@K|jd{y7wr)r)r4exc_type exc_value tracebacks r9 __aexit__zServer.__aexit__`s kkms r)rqzlist[trio.SocketListener]rrz6Callable[[trio.abc.Stream], Coroutine[Any, Any, None]]rILoggerLike | Nonerdre)rdzset[ServerConnection])rtrio.TaskStatus[Server]rdre)rboolrzCloseCode | intrrfrdre)rdr )rztype[BaseException] | NonerzBaseException | NonerzTracebackType | Nonerdre)rjrkrlrmr0propertyryr2TASK_STATUS_IGNOREDrr GOING_AWAYrrrrr:r9r$r$s  %) ,H"   *    "04/G/G#,# #H#' ) 4 4 ****** **  **X,((   r:r$deflater'r&ir()hostbacklogrqsslorigins extensions subprotocolsselect_subprotocol compressionrXrYrZ open_timeoutr*r+r,max_sizer-rIcreate_connectionrc K t| dk(r tn| td| t|.| tdt j |||d{}n'| td| td| tdd   fd }t ||j| d{y7k7w) a9 Create a WebSocket server listening on ``port``. Whenever a client connects, the server creates a :class:`ServerConnection`, performs the opening handshake, and delegates to the ``handler`` coroutine. The handler receives the :class:`ServerConnection` instance, which you can use to send and receive messages. Once the handler completes, either normally or with an exception, the server performs the closing handshake and closes the connection. When using :func:`serve` with :meth:`nursery.start `, you get back a :class:`Server` object. Treat it as an asynchronous context manager to ensure that the server will be closed gracefully:: from websockets.trio.server import serve async def handler(websocket): ... # set this event to exit the server stop = trio.Event() with trio.open_nursery() as nursery: server = await nursery.start(serve, handler, port) async with server: await stop.wait() Alternatively, to stop the server gracefully, call its :meth:`~Server.aclose` method:: with trio.open_nursery() as nursery: server = await nursery.start(serve, handler, port) try: await stop.wait() finally: await server.aclose() Args: handler: Connection handler. It receives the WebSocket connection, which is a :class:`ServerConnection`, in argument. port: TCP port the server listens on. See :func:`~trio.open_tcp_listeners` for details. host: Network interfaces the server binds to. See :func:`~trio.open_tcp_listeners` for details. backlog: Listen backlog. See :func:`~trio.open_tcp_listeners` for details. listeners: Preexisting TCP listeners. ``listeners`` replaces ``port``, ``host``, and ``backlog``. See :func:`trio.serve_listeners` for details. ssl: Configuration for enabling TLS on the connection. origins: Acceptable values of the ``Origin`` header, for defending against Cross-Site WebSocket Hijacking attacks. Values can be :class:`str` to test for an exact match or regular expressions compiled by :func:`re.compile` to test against a pattern. Include :obj:`None` in the list if the lack of an origin is acceptable. extensions: List of supported extensions, in order in which they should be negotiated and run. subprotocols: List of supported subprotocols, in order of decreasing preference. select_subprotocol: Callback for selecting a subprotocol among those supported by the client and the server. It receives a :class:`ServerConnection` (not a :class:`~websockets.server.ServerProtocol`!) instance and a list of subprotocols offered by the client. Other than the first argument, it has the same behavior as the :meth:`ServerProtocol.select_subprotocol ` method. compression: The "permessage-deflate" extension is enabled by default. Set ``compression`` to :obj:`None` to disable it. See the :doc:`compression guide <../../topics/compression>` for details. process_request: Intercept the request during the opening handshake. Return an HTTP response to force the response or :obj:`None` to continue normally. When you force an HTTP 101 Continue response, the handshake is successful. Else, the connection is aborted. ``process_request`` may be a function or a coroutine. process_response: Intercept the response during the opening handshake. Return an HTTP response to force the response or :obj:`None` to continue normally. When you force an HTTP 101 Continue response, the handshake is successful. Else, the connection is aborted. ``process_response`` may be a function or a coroutine. server_header: Value of the ``Server`` response header. It defaults to ``"Python/x.y.z websockets/X.Y"``. Setting it to :obj:`None` removes the header. open_timeout: Timeout for opening connections in seconds. :obj:`None` disables the timeout. ping_interval: Interval between keepalive pings in seconds. :obj:`None` disables keepalive. ping_timeout: Timeout for keepalive pings in seconds. :obj:`None` disables timeouts. close_timeout: Timeout for closing connections in seconds. :obj:`None` disables the timeout. max_size: Maximum size of incoming messages in bytes. :obj:`None` disables the limit. You may pass a ``(max_message_size, max_fragment_size)`` tuple to set different limits for messages and fragments when you expect long messages sent in short fragments. max_queue: High-water mark of the buffer where frames are received. It defaults to 16 frames. The low-water mark defaults to ``max_queue // 4``. You may pass a ``(high, low)`` tuple to set the high-water and low-water marks. If you want to disable flow control entirely, you may set it to ``None``, although that's a bad idea. logger: Logger for this server. It defaults to ``logging.getLogger("websockets.server")``. See the :doc:`logging guide <../../topics/logging>` for details. create_connection: Factory for the :class:`ServerConnection` managing the connection. Set it to a wrapper or a subclass to customize connection handling. task_status: For compatibility with :meth:`nursery.start `. Nrzunsupported compression: z/port is required when listeners is not provided)rrz#port is incompatible with listenersz#host is incompatible with listenersz&backlog is incompatible with listenersc Ktj4d{} tjntj 5Ntj|dd}t |tjsJ |j d{d} d fd }t | }||| jd{dddjjtur*jd{ dddd{yjjj! d{j#d{jj/ dddd{y77$#tj$rYddddddd{7ywxYw7 #1swY xYw7777#t$$rHj&j)ddj#t*j,d{7YwxYw#jj/wxYw#t$$r!tj0|d{7YwxYw7#1d{7swYyxYww) a Handle the lifecycle of a WebSocket connection. Since this coroutine doesn't have a caller that can handle exceptions, it attempts to log relevant ones. It guarantees that the TCP connection is closed before exiting. NT) server_sidehttps_compatiblec>J|jusJ|Sr)r7)r7rrxrs r9protocol_select_subprotocolzBserve..stream_handler..protocol_select_subprotocolUs2 $6#AA#A#+z/B/B#BB#B#5j,#OOr:)rrrrrrIr)zconnection handler failedrA)r7rrzSequence[Subprotocol]rdzSubprotocol | None)r2r| CancelScope fail_after SSLStreamrF do_handshakeBrokenResourceErrorrr\r7rUr close_streamruaddstart_keepaliverrGrIrJrINTERNAL_ERRORdiscardaclose_forcefully)r6r5rr7rxr,rrrrrIr-rrrr*r+rXrYrr1rZrrs @r9stream_handlerzserve..stream_handler)s$$&[ 5[ 5'Z 5$+$$&6C "&"(,-1 " *&$..AAA#"("5"5"777 0*5P&4P*?P0P . '#-%1+F!)%  H"3 &3%1&3"+ "J%..'(%CJ&&,,D8$11333U[ 5[ 5[ 5X&&**:6**,?!*--- %++---**22:>m[ 5[ 5[ 5(8#77#"'C[ 5[ 5[ 5*#ZCCL4S[ 5^. . !F%%++,GRV+W$++I,D,DEEEF **22:> 5,,V444 5q[ 5[ 5[ 5[ 5swKF;KJ.+I?8G6GF>GAG6'G3(G6,7I?#H$I?) K4H5K:+I?& H 1H2H 6I H  II?)J.* K5J,6K>GG0G6I? K(G+)K/G00G66H ; I?KH  I AIIIIIII<<I??!J) J#!J)&J.(J))J.,K.K4J7 5K<K)r)r6rardre)rr ValueErrorr#r2open_tcp_listenersr$r)rrportrrrqrrrrrrrXrYrZrr*r+r,rr-rIrrrr1s` ````` ``````````` @r9r"r"isNl+i5jA  4[MBCC , <NO O11$T7SS  BC C  BC C  EF Fe5e5e5PI~v 6F   ;  777eTd8s%A!C%5C!6A%C%C#C%#C%r)0rrz-Callable[[ServerConnection], Awaitable[None]]r int | Nonerzstr | bytes | Nonerrrqz list[trio.SocketListener] | Nonerzssl_module.SSLContext | Nonerz0Sequence[Origin | re.Pattern[str] | None] | Nonerz'Sequence[ServerExtensionFactory] | NonerzSequence[Subprotocol] | NonerzNCallable[[ServerConnection, Sequence[Subprotocol]], Subprotocol | None] | NonerrirXrgrYrhrZrirrbr*rbr+rbr,rbrrcr-rcrIrrztype[ServerConnection] | Nonerrrdre): __future__rrrKrsrer ssl_modulecollections.abcrrrtypesrtypingrr r r r2trio.abcasyncio.serverr extensions.baserextensions.permessage_deflaterframesrrPrhttp11rrrr7rrrr1rrrrrutilsrrxrr r!__all__r#abc AsyncResourcer$rr"rr:r9rs"  8811 '4L+....#@@#- t)zt)nITXX # #I\i8 $26(,@D:>15 '  &!#"$!#"$;@<> $7;+/+C+Cei8 :i8 i8  i8  i80i8 &i8>i88i8/i8 i8()i8. /i8< =i8HIi8LMi8N Oi8PQi8R Si8V9Wi8X:Yi8\ ]i8`5ai8d)ei8f gi8r: