o 96d4@s&ddlZddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZddlZddlmZddlmZddlmZddlmZddlmZdd lmZdd lmZmZmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*m+Z+m,Z,m-Z-ddl.m/Z/GdddZ0GdddeZ1dS)N) AnyDictListLiteralOptionalSequenceTupleUnioncast)unquote)Headers)ConnectionClosed)ServerPerMessageDeflateFactory) HTTPResponse)WebSocketServerProtocol) Subprotocol) ASGISendEventWebSocketAcceptEventWebSocketCloseEventWebSocketConnectEventWebSocketDisconnectEventWebSocketReceiveEventWebSocketScopeWebSocketSendEvent)Config)TRACE_LOG_LEVEL)get_local_addrget_path_with_query_stringget_remote_addris_ssl) ServerStatec@sBeZdZdZdeddfddZdeddfddZdefd d ZdS) ServerFwsreturnNcCdSNselfr"r&r&C:\Users\jesus\OneDrive\Desktop\erpjis_fastapi\backend\jisbackend\Lib\site-packages\uvicorn/protocols/websockets/websockets_impl.pyregister1zServer.registercCr$r%r&r'r&r&r) unregister4r+zServer.unregistercCs|j Sr%)closingr(r&r&r) is_serving7szServer.is_serving) __name__ __module__ __qualname__r-rr*r,boolr/r&r&r&r)r!.s r!c s(eZdZUeeeefed< d*dedede ee fde e j ffdd Zd e jd dffd d Zd e ed dffdd Zd+ddZde jd dfddZdeded e efddZdede eed e efddZd+ddZdeded e fdd Zd+d!d"Zd,d%d&Z d e!d'fd(d)Z"Z#S)-WebSocketProtocol extra_headersNconfig server_state app_state_loopc s|js|||_|j|_|pt|_|j|_||_ |j |_ |j |_ d|_ d|_ d|_d|_d|_t|_t|_t|_d|_d|_d|_d|_t|_g}|jjr^|ttj|j |j|jj!|jj"|jj#|jj$|t%&ddd|_'dd|j(D|_)dS)NFz uvicorn.error) ws_handler ws_servermax_sizeZ max_queueZ ping_intervalZ ping_timeout extensionsloggercSs$g|]\}}|d|dfqS)latin-1decode.0namevaluer&r&r) ssz.WebSocketProtocol.__init__..)*Zloadedloadr6Z loaded_appappasyncioget_event_looploop root_pathr8 connectionstasks transportserverclientschemescopeEventhandshake_started_eventhandshake_completed_event closed_eventinitial_response connect_sent lost_connection_before_handshakeaccepted_subprotocolr!r;Zws_per_message_deflateappendrsuper__init__r:Z ws_max_sizeZ ws_max_queueZws_ping_intervalZws_ping_timeoutlogging getLoggerZ server_headerdefault_headersr5)r(r6r7r8r9r= __class__r&r)r^>sN      zWebSocketProtocol.__init__rOr#csx|j|||_t||_t||_t|rdnd|_|j t r4|jr*d|jnd}|j t d|t |dS)Nwssr"%s:%d - z%sWebSocket connection made)rMaddrOrrPrrQrrRr> isEnabledForrlogr]connection_made)r(rOprefixrbr&r)rjxs    z!WebSocketProtocol.connection_madeexccsz|j||jtr|jrd|jnd}|jtd||j |_ |j t ||dur;|j dSdS)Nrerfz%sWebSocket connection lost)rMremover>rhrrQrirVis_setrZsetr]connection_lostrOclose)r(rlrkrbr&r)rps     z!WebSocketProtocol.connection_lostcCs4d|j_|jr|dn||jdS)NT)r;r-rVrnZfail_connectionsend_500_responserOrqr.r&r&r)shutdowns   zWebSocketProtocol.shutdowntaskcCs|j|dSr%)rNdiscard)r(rur&r&r)on_task_completesz"WebSocketProtocol.on_task_completepathheadersc s|d\}}}tjj|g}|dD]}|dd|dDqdd|D}d|j j dd d |j |j |j |jt||d |d |||jd |_|j|} | |j|j| |jId H|jS)a( This hook is called to determine if the websocket should return an HTTP response and close. Our behavior here is to start the ASGI application, and then wait for either `accept` or `close` in order to determine if we should close the connection. ?zSec-WebSocket-ProtocolcSsg|]}|qSr&)strip)rCtokenr&r&r)rFsz5WebSocketProtocol.process_request..,cSs(g|]\}}|d|jdddfqS)asciisurrogateescape)errors)encoderBr&r&r)rFsZ websocketz2.3)version spec_versionz1.1r~) typeZasgi http_versionrRrPrQrLrxraw_path query_stringry subprotocolsstateN) partition websocketslegacy handshakeZ check_requestget_allextendsplit raw_itemsr6Z asgi_versionrRrPrQrLr rr8copyrSrK create_taskrun_asgiadd_done_callbackrwrNrgrUwaitrX) r(rxryZ path_portion_rrheaderZ asgi_headersrur&r&r)process_requests8    z!WebSocketProtocol.process_requestavailable_subprotocolscCs|jS)z We override the standard 'process_subprotocol' behavior here so that we return whatever subprotocol is sent in the 'accept' message. )r[)r(ryrr&r&r)process_subprotocolsz%WebSocketProtocol.process_subprotocolcCsHd}ddtt|dddd|g}|jd||jdS)NsInternal Server ErrorsMHTTP/1.1 500 Internal Server Error content-type: text/plain; charset=utf-8 scontent-length: r~s sconnection: close )strlenrrOwritejoinrUro)r(msgcontentr&r&r)rssz#WebSocketProtocol.send_500_responseprotocolcs|j|IdHdS)z This is the main handler function for the 'websockets' implementation to call into. We just wait for close then return, and instead allow 'send' and 'receive' events to drive the flow. N)rVro wait_closed)r(rrxr&r&r)r:s zWebSocketProtocol.ws_handlerc s z||j|j|jIdH}Wn9tyI}z-|jd}|jj||d|j s1| n|j IdH|jWYd}~dSd}~ww|j|j sgd}|j|| |jdS|durd}|j|||j IdH|jdSdS)zr Wrapper around the ASGI callable, handling exceptions and unexpected termination states. NzException in ASGI application )exc_infoz1ASGI callable returned without sending handshake.z4ASGI callable should return None, but returned '%s'.)rHrS asgi_receive asgi_send BaseExceptionrWror>errorrUrnrsrVrrOrq)r(resultrlrr&r&r)rs2     zWebSocketProtocol.run_asgimessagerc s|d}|jsz|dkrGtd|}|jd|jdt|jd|_ttt | d|_ d|vr@|j dd |dD|jdS|d krrtd |}|jd |jdt|jtjjgd f|_|j|jdSd}t|||js|jIdH|dkrtd|}| d}| d}|dur|n|}||IdHdS|d krtd |}| dd}| ddpd}|||IdH|jdSd}t||d}t||)Nrzwebsocket.acceptrz%s - "WebSocket %s" [accepted]rQZ subprotocolrycss(|]\}}|d|dfVqdS)r?Nr@rBr&r&r) s  z.WebSocketProtocol.asgi_send..zwebsocket.closerz%s - "WebSocket %s" 403rzLExpected ASGI message 'websocket.accept' or 'websocket.close', but got '%s'.zwebsocket.sendrbytestextcodeireasonrfzJExpected ASGI message 'websocket.send' or 'websocket.close', but got '%s'.z>Unexpected ASGI message '%s', after sending 'websocket.close'.)rUrnr r>inforSrrXrrgetr[r5rrohttp HTTPStatus FORBIDDENrW RuntimeErrorrVrsendrq) r(rZ message_typer bytes_dataZ text_datadatarrr&r&r)rsf              zWebSocketProtocol.asgi_send)rrrc s|js d|_ddiS|jIdH|jrdddS|jr%dddSz |IdH}Wn,tyZ}z |j|j j rJdddWYd}~Sd|j dWYd}~Sd}~wwdd i}t |t rj||d <|S||d <|S) NTrzwebsocket.connectzwebsocket.disconnecti)rrirrzwebsocket.receiverr)rYrVrrZrWrnrecvr ror;r-r isinstancer)r(rrlrr&r&r)rTs2     zWebSocketProtocol.asgi_receiver%)r#N)rrr#N)$r0r1r2rrr__annotations__rr rrrrIAbstractEventLoopr^ Transportrj ExceptionrprtTaskrwr rrrrrrsrr:rrr r __classcell__r&r&rbr)r4;sb  :  -   Fr4)2rIrr_typingrrrrrrrr r urllib.parser rZwebsockets.datastructuresr Zwebsockets.exceptionsr Z(websockets.extensions.permessage_deflaterZwebsockets.legacy.serverrZwebsockets.serverrZwebsockets.typingrZuvicorn._typesrrrrrrrrZuvicorn.configrZuvicorn.loggingrZuvicorn.protocols.utilsrrrrZuvicorn.serverr r!r4r&r&r&r)s&,       (