hUdZddlZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z ddlmZmZd d lmZmZmZmZmZmZmZmZd d lmZmZd Zeeed <dZ GddZ!dS)z%WebSocket protocol versions 13 and 8.N)partial)AnyFinalOptionalUnion) BaseProtocol)ClientConnectionResetError) ZLibBackendZLibCompressor)MASK_LENMSG_SIZEPACK_CLOSE_CODE PACK_LEN1 PACK_LEN2 PACK_LEN3 PACK_RANDBITSwebsocket_mask)WS_DEFLATE_TRAILING WSMsgType DEFAULT_LIMITiceZdZdZdeejddddedej de de d ejd e d e d d fdZ dde de d ee d d fdZd e d efdZdde dee efd d fdZd S)WebSocketWritera3WebSocket writer. The writer is responsible for sending messages to the client. It is created by the protocol when a connection is established. The writer should avoid implementing any application logic and should only be concerned with the low-level details of the WebSocket protocol. Fr)use_masklimitrandomcompress notakeoverprotocol transportrrrrr returnNc||_||_||_t|jd|_||_||_d|_||_ d|_ d|_ dS)zInitialize a WebSocket writer. FrN) r!r"rr getrandbitsget_random_bitsrr _closing_limit _output_size _compressobj)selfr!r"rrrrr s a/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/aiohttp/_websocket/writer.py__init__zWebSocketWriter.__init__,s`! "  &v'92>>  $  !%messageopcodecbK|jr|tjzstdd}|s|jr|dkrd}|r||}n-|js||j|_|j}||d{V||jr tj n tj z t}t|}|j}|rdnd}d|z|z} |dkrt!| ||z} d} n3|d krt#| d|z|} d } nt%| d |z|} d } |jrtd|rvt+|} t/|}t1| ||j| | z|z|xjt6z c_n]|t8kr5|j| |j|n|j| |z|xj| |zz c_|j|jkr4d|_|jjr#|j d{VdSdSdS) z:FFFJJz3>:FFFJ > $ $ & & R,-PQQ Q  3 !5!5!7!788D((G 4 ) ) ) N $!8 9 9 9    )    ( " " N  ( ( ( N  ) ) ) ) N '!1 2 2 2 Z*44  t{ * * !D }$ 4m1133333333333 + * 4 4r/cFttj| tS)N)levelwbitsmax_sync_chunk_size)r r Z_BEST_SPEEDWEBSOCKET_MAX_SYNC_CHUNK_SIZE)r,rs r-r;z"WebSocketWriter._make_compress_objs'*) =    r/r/codecKt|tr|d} |t ||zt jd{Vd|_dS#d|_wxYw)z     ! ! !U5#:5F !QU ! ! ! ! ! !r/r)"r_rar functoolsrtypingrrrr base_protocolr client_exceptionsr compression_utilsr r helpersrrrrrrrrmodelsrrrrd__annotations__rTrrfr/r-rosh+++ ............((((((::::::;;;;;;;;                    32222222! uSz!!!!)O!O!O!O!O!O!O!O!O!O!r/