o ]i @s~dZddlmZddlmZddlmZmZmZddl m Z er2ddl m Z ddl mZdd lmZd ZGd d d Zd ZdS)z. wsproto ~~~~~~~ A WebSocket implementation. ) annotations) TYPE_CHECKING) ConnectionConnectionStateConnectionType) H11Handshake) Generator)Event)Headersz1.3.1c@sPeZdZdZdddZedd d Zd ddZd!ddZd"ddZ d#ddZ dS)$ WSConnectionzN Represents the local end of a WebSocket connection to a remote peer. connection_typerreturnNonecCs |tju|_t||_d|_dS)z Constructor :param wsproto.connection.ConnectionType connection_type: Controls whether the library behaves as a client or as a server. N)rCLIENTclientr handshake connection)selfr rU/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/wsproto/__init__.py__init__s   zWSConnection.__init__rcCs|jdur |jjS|jjS)z_ :returns: Connection state :rtype: wsproto.connection.ConnectionState N)rrstaterrrrr's zWSConnection.stateheadersr path bytes | strcCs|j||dS)N)rinitiate_upgrade_connection)rrrrrrr1sz(WSConnection.initiate_upgrade_connectioneventr bytescCs@d}|jdur||j|7}|jj|_|S||j|7}|S)a Generate network data for the specified event. When you want to communicate with a WebSocket peer, you should construct an event and pass it to this method. This method will return the bytes that you should send to the peer. :param wsproto.events.Event event: The event to generate data for :returns bytes: The data to send to the peer N)rrsend)rrdatarrrr!6s  zWSConnection.sendr" bytes | NonecCs4|jdur|j||jj|_dS|j|dS)z Feed network data into the connection instance. After calling this method, you should call :meth:`events` to see if the received data triggered any new events. :param bytes data: Data received from remote peer N)rr receive_data)rr"rrrr$Is zWSConnection.receive_dataGenerator[Event, None, None]ccs4|jEdH|jdur|jEdHdSdS)z A generator that yields pending events. Each event is an instance of a subclass of :class:`wsproto.events.Event`. N)reventsrrrrrr&Xs  zWSConnection.eventsN)r rrr)rr)rr rrrr)rr rr)r"r#rr)rr%) __name__ __module__ __qualname____doc__rpropertyrrr!r$r&rrrrr s    r )rr N)r* __future__rtypingrrrrrrrcollections.abcr r&r r __version__r __all__rrrrs      M