CCfT'ddlmZddlZddlmZddlZddlZddlZddlm Z m Z m Z m Z m Z mZmZmZddlmZmZmZmZGddeZGdd Zd d Zy) ) annotationsN)Enum)AnyCallableDictListLiteralTupleUnionNoReturn)WebSocketStateMessage WebSocketWebSocketDisconnectceZdZdZdZdZy) ReceiveTypetextbytesjsonN)__name__ __module__ __qualname__TEXTBYTESJSONT/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/websocket_rooms/room.pyrrs D E DrrceZdZUeej j ejj ejj fZ de d<de d<de d<de d<de d <d e d <d Z dd Z ddZ ddZddZddZdZddZd d!dZd"dZej j f d#dZ d$ d%dZ d$ d%dZd&dZy)'RoomzList[WebSocket] _websocketszCDict[Literal['before', 'after'], Callable[[Room, WebSocket], None]] _on_connectzNDict[Literal['text', 'bytes', 'json'], Callable[[Room, WebSocket, Any], None]] _on_receive_on_disconnectz asyncio.Queue_to_pushz asyncio.Task_publisher_taskc|g|_i|_i|_i|_t j |_d|_yN)r!r#r$r"asyncioQueuer%r&selfs r__init__z Room.__init__%s7   #rcK|js(tj|j|_|jj d}|rt |||d{|jd{|jj||jj d}|rt |||d{|j|d{y77r7"7 w)a Accepts a websocket connection and, adds it to the room clients list, and runs the _run_client_lifecycle function to manage the connection. Parameters ---------- websocket : WebSocket The websocket to connect Returns ------- NoReturn The function runs for the entire client lifetime. beforeNafter) r&r) create_task _publisherr"getawait_if_awaitableacceptr!append_run_client_lifecycle)r, websocketr/r0s rconnectz Room.connect-s###*#6#6t7H#ID !!%%h/ $VD)%<= = =     *  $$W- $U4%;< < <((333 >  =3sIA'C:)C2*C:C4AC:C6C:,C8-C:4C:6C:8C:cpK|jr$|jj|d{yy7w)z Pushes the message object into the message queue. Parameters ---------- message_object : Tuple[Any, Literal["text", "bytes", "json"]] the message object to be pushed N)r&r%put)r,message_objects r_pushz Room._pushHs1   --##N3 3 3 3s +646cFK|j|dfd{y7w)z Pushes the json data as a broadcast to all the room clients Parameters ---------- message : dict the message to be pushed rNr=r,messages r push_jsonzRoom.push_jsonTjj'6*+++ !!cFK|j|dfd{y7w)z Pushes the str data as a broadcast to all the room clients Parameters ---------- message: str the message to be pushed rNr?r@s r push_textzRoom.push_text_rCrDcFK|j|dfd{y7w)z Pushes the bytes data as a broadcast to all the room clients Parameters ---------- message: bytes the message to be pushed rNr?r@s r push_byteszRoom.push_bytesjsjj'7+,,,rDcK |jjd{\}}g}t|jdkDrg|jj } |j d||d{|j|t|jdkDrg||_778#t $rYwxYww)z A publisher that runs on the message queue and pushes the message to all the clients. This courtine is ran in the background in the _publisher_task task. Nrsend_)r%r3lenr!pop__getattribute__rr6)r,rA message_typeliving_connectionsr8s rr2zRoom._publisherus ,0MM,=,=,?&? #Wl24 d&&'!+ ,,002 L)44u\N5KLWUUU#)))4 d&&'!+ 2D &? V*sKCB2;CB6;B4<B6)C* C4B66 C?CCCcK |jtjk7r td|j d{}t j ||j|d}d}tjj|jvr|tjj}tjj|jjvr)|jtjj}ntj|jjvr`tj|j!d}|jtjj}ntj"j|jvr|tj"j}tj"j|jjvr(|jtj"j}nptjj|jjvr?#// 0@0@0F0FG %%++w||~=";#3#3#9#9:D"''--1A1A1F1F1HH#// 0@0@0F0FG$))//43C3C3H3H3JJ#zz$/#// 0@0@0F0FGD#D)T:H**84&= $=2'" 6++i+5 5 5 6sMK= # =s43CCCCBCCCCCcK|jj}|D]}|j|d{y7w)z9 Close the room and all its connections. N)r!copyr_)r, websocketsr8s rrcz Room.closes@%%**, # )I++i( ( ( ) (s4A>Ac6dvr tddfd }|S)a  The decorator to specify the callbacks that will be run when a message is received from client websocket. Parameters ---------- mode : Literal["text", "bytes", "json"] The type of the message that callback will be ran on. Raises ------ RuntimeError('The "mode" argument should be "text", "bytes" or "json".') Returns ------- Callable[[Room, WebSocket, Any], None] The function that the decorator receives. )rrrz8The "mode" argument should be "text", "bytes" or "json".c&|j<|Sr()r#r`moder,s rinnerzRoom.on_receive..inner%)D  T "Kr)r`&Callable[[Room, WebSocket, Any], None]rTr,rkrls`` r on_receivezRoom.on_receives**00J   rc6dvr tddfd }|S)a The decorator to specify the callbacks that will run on websockets connection. Parameters ---------- mode : Literal["before", "after"] The execution time of the callback - before / after connecting the websocket. Raises ------ RuntimeError('The "mode" argument should be "before" or "after".') Returns ------- Callable[[Room, WebSocket], None] The function that the decorator receives. r/r02The "mode" argument should be "before" or "after".c&|j<|Sr()r"rjs rrlzRoom.on_connect..innerrmrr`!Callable[[Room, WebSocket], None]rorps`` r on_connectzRoom.on_connect&( * *ST T  rc6dvr tddfd }|S)a The decorator to specify the callbacks that will run on websockets disconnect. Paramaters ---------- mode : Literal["before", "after"] The execution time of the callback - before / after disconnecting the websocket. Raises ------ RuntimeError('The "mode" argument should be "before" or "after".') Returns ------- Callable[[Room, WebSocket], None] The function that the decorator receives. rsrtc&|j<|Sr()r$rjs rrlz!Room.on_disconnect..inner s(,D   %Krrvrorps`` r on_disconnectzRoom.on_disconnect ryrc|Sr(rr+s r__call__z Room.__call__&s rN)r8rreturnr )r<z,Tuple[Any, Literal['text', 'bytes', 'json']])rAdictrNone)rAstrrr)rArrr)F)r8rrQboolrr)rr)rkzRoom.RECEIVE_TYPESrrn)r0)rkzLiteral['before', 'after']rrw)rr )rrrr rrrYrr RECEIVE_TYPES__annotations__r-r9r=rBrFrHr2r7r_rcrqrxr|r~rrrr r s 0 0 6 6 8I8I8O8OOM! TT__WW!!$46 4 , , -2"*6X=$)*5)9)9)?)?& /B29. *<29. *:rr cPKtj|r |d{yy7wr()r]r^)ras rr4r4*s#8$%s &$&)rar) __future__rr)enumrr]rrVtypingrrrrr r r r starlette.websocketsr rrrrr r4rrrrsM"    YX$ MM`r