File: //usr/lib/python3/dist-packages/websocket/__pycache__/_app.cpython-310.pyc
o
���a�: � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
ddl T ddl m
Z
dgZG d d
� d
�ZG d d � d �ZG d
d� d�ZdS )z
� N� )�ABNF)� WebSocket�getdefaulttimeout)�*)�_logging� WebSocketAppc @ s e Zd ZdZdd� Zdd� ZdS )�
Dispatcherz
Dispatcher
c C � || _ || _d S �N��app� ping_timeout��selfr
r � r �0/usr/lib/python3/dist-packages/websocket/_app.py�__init__) �
zDispatcher.__init__c C s\ | j jr,t�� }|�| j jjtj� |�| j�}|r |� s d S |� |� � | j jsd S d S r )
r
� keep_running� selectors�DefaultSelector�register�sock�
EVENT_READ�selectr �close)r r �
read_callback�check_callback�sel�rr r r �read- s �zDispatcher.readN)�__name__�
__module__� __qualname__�__doc__r r! r r r r r % s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �
SSLDispatcherz
SSLDispatcher
c C r
r r r r r r r > r zSSLDispatcher.__init__c C s4 | j jr| �� }|r|� sd S |� | j jsd S d S r )r
r r )r r r r r r r r r! B s �zSSLDispatcher.readc C s^ | j jj}|�� r |gS t�� }|�|tj� |�| j�}|� � t
|�dkr-|d d S d S )Nr ) r
r �pendingr r r r r r r �len)r r r r r r r r J s
�zSSLDispatcher.selectN)r" r# r$ r% r r! r r r r r r& : s
r& c @ sx e Zd ZdZ ddd�Zejfdd�Zdd � Zd
d � Z
ddd�Z
dd� Z dd� Z dd� Z
dS )r z_
Higher level of APIs are provided. The interface is like JavaScript WebSocket object.
NTc C sv || _ |dur |ng | _| | _|| _|| _|| _|| _|| _|| _|| _ | | _
d| _ | | _ d| _
d| _d| _|
| _dS )a�
WebSocketApp initialization
Parameters
----------
url: str
Websocket url.
header: list or dict
Custom header for websocket handshake.
on_open: function
Callback object which is called at opening websocket.
on_open has one argument.
The 1st argument is this class object.
on_message: function
Callback object which is called when received data.
on_message has 2 arguments.
The 1st argument is this class object.
The 2nd argument is utf-8 data received from the server.
on_error: function
Callback object which is called when we get error.
on_error has 2 arguments.
The 1st argument is this class object.
The 2nd argument is exception object.
on_close: function
Callback object which is called when connection is closed.
on_close has 3 arguments.
The 1st argument is this class object.
The 2nd argument is close_status_code.
The 3rd argument is close_msg.
on_cont_message: function
Callback object which is called when a continuation
frame is received.
on_cont_message has 3 arguments.
The 1st argument is this class object.
The 2nd argument is utf-8 string which we get from the server.
The 3rd argument is continue flag. if 0, the data continue
to next frame data
on_data: function
Callback object which is called when a message received.
This is called before on_message or on_cont_message,
and then on_message or on_cont_message is called.
on_data has 4 argument.
The 1st argument is this class object.
The 2nd argument is utf-8 string which we get from the server.
The 3rd argument is data type. ABNF.OPCODE_TEXT or ABNF.OPCODE_BINARY will be came.
The 4th argument is continue flag. If 0, the data continue
keep_running: bool
This parameter is obsolete and ignored.
get_mask_key: function
A callable function to get new mask keys, see the
WebSocket.set_mask_key's docstring for more information.
cookie: str
Cookie value.
subprotocols: list
List of available sub protocols. Default is None.
NFr )�url�header�cookie�on_open�
on_message�on_data�on_error�on_close�on_ping�on_pong�on_cont_messager � get_mask_keyr � last_ping_tm� last_pong_tm� subprotocols)r r) r* r, r- r/ r0 r1 r2 r3 r r4 r+ r7 r. r r r r ^ s" ?
zWebSocketApp.__init__c C s$ | j r | j �||�dkrtd��dS )a
send message
Parameters
----------
data: str
Message to send. If you set opcode to OPCODE_TEXT,
data must be utf-8 string or unicode.
opcode: int
Operation code of data. Default is OPCODE_TEXT.
r z Connection is already closed.N)r �send�"WebSocketConnectionClosedException)r �data�opcoder r r r8 � s
��zWebSocketApp.sendc K s, d| _ | jr| jjdi |�� d| _dS dS )z-
Close websocket connection.
FNr )r r r )r �kwargsr r r r � s
�zWebSocketApp.closec
C st |� |�s8t�� | _| jr1z| j�|� W n ty0 } zt�d�|�� W Y d }~d S d }~ww |� |�rd S d S )Nz send_ping routine terminated: {}) �wait�timer5 r �ping� Exceptionr �warning�format)r �interval�event�payload�exr r r �
_send_ping� s
���zWebSocketApp._send_pingr � Fc s� �dur �dkr t d��|dur|dk rt d���r$|r$|�kr$t d��|s(g }|s,i }�jr3t d��d�d�_d�_d�_d� ��fdd � �zkt�j||�jdu|
dd
��_�j�t � � �jj
�j �j �j
|||| �j| | ||d � |
s{����}
���j� |r�t�� � tj�j|� |fd ��d�_��� ��fd
d�}��fdd�}|
��jj||� W dS tttfy� } z ���j |� t |t�rʂ �� t |t� W Y d}~S d}~ww )a�
Run event loop for WebSocket framework.
This loop is an infinite loop and is alive while websocket is available.
Parameters
----------
sockopt: tuple
Values for socket.setsockopt.
sockopt must be tuple
and each element is argument of sock.setsockopt.
sslopt: dict
Optional dict object for ssl socket option.
ping_interval: int or float
Automatically send "ping" command
every specified period (in seconds).
If set to 0, no ping is sent periodically.
ping_timeout: int or float
Timeout (in seconds) if the pong message is not received.
ping_payload: str
Payload message to send with each ping.
http_proxy_host: str
HTTP proxy host name.
http_proxy_port: int or str
HTTP proxy port. If not set, set to 80.
http_no_proxy: list
Whitelisted host names that don't use the proxy.
skip_utf8_validation: bool
skip utf8 validation.
host: str
update host header.
origin: str
update origin header.
dispatcher: Dispatcher object
customize reading data from socket.
suppress_origin: bool
suppress outputting origin header.
Returns
-------
teardown: bool
False if caught KeyboardInterrupt, True if other exception was raised during a loop
Nr zEnsure ping_timeout > 0zEnsure ping_interval >= 0z#Ensure ping_interval > ping_timeoutzsocket is already openedTc sb �r�� � r� �� ��� d�_�jr�j�� ��| r | nd�\}}d�_���j||� dS )a
Tears down the connection.
Parameters
----------
close_frame: ABNF frame
If close_frame is set, the on_close handler is invoked
with the statusCode and reason from the provided frame.
FN) �is_alive�set�joinr r r �_get_close_args� _callbackr0 )� close_frame�close_status_code� close_reason)rD r �threadr r �teardown s
�z*WebSocketApp.run_forever.<locals>.teardown)�sockopt�sslopt�fire_cont_frame�skip_utf8_validation�enable_multithread) r* r+ �http_proxy_host�http_proxy_port�
http_no_proxy�http_proxy_authr7 �host�origin�suppress_origin�
proxy_type)�target�argsc s� � j s�� S � j�d�\} }| tjkr�|�S | tjkr&� �� j|j� dS | tj kr:t
�
� � _ � �� j |j� dS | tj
krZ� jrZ� �� j|j|j|j� � �� j|j|j� dS |j}| tjkrg|�d�}� �� j||jd� � �� j|� dS )NT�utf-8)r r �recv_data_framer � OPCODE_CLOSE� OPCODE_PINGrM r1 r: � OPCODE_PONGr> r6 r2 � OPCODE_CONTr3 r. r; �fin� OPCODE_TEXT�decoder- )�op_code�framer: )r rR r r r! K s4
�
� �� �
z&WebSocketApp.run_forever.<locals>.readc sT � r(t � � �j � k} �j�j dk }�j�j � k}�jr(| r(|s$|r(td��dS )Nr zping/pong timed outT)r> r5 r6 �WebSocketTimeoutException)�has_timeout_expired�$has_pong_not_arrived_after_last_ping�has_pong_arrived_too_late)r r r r �checke s ���z'WebSocketApp.run_forever.<locals>.checkr ) �WebSocketExceptionr r r5 r6 r r4 r3 �
settimeoutr �connectr) r* r+ r7 �create_dispatcherrM r, � threading�Event�ThreadrG �daemon�startr! r@ �KeyboardInterrupt�
SystemExitr/ �
isinstance)r rS rT �
ping_intervalr � ping_payloadrX rY rZ r[ rV r\ r] �
dispatcherr^ r_ r! rq �er )rD r r rR rQ r � run_forever� sl 4� �
�
��zWebSocketApp.run_foreverc C s& |pd}| j �� rt| |�S t| |�S )N�
)r �is_sslr& r )r r �timeoutr r r ru z s
z WebSocketApp.create_dispatcherc C sb | j r|s ddgS |jr-t|j�dkr-d|jd |jd }|jdd� �d�}||gS ddgS )z�
_get_close_args extracts the close code and reason from the close body
if it exists (RFC6455 says WebSocket Connection Close Code is optional)
N� � r r rb )r0 r: r( rj )r rN rO �reasonr r r rL � s
z WebSocketApp._get_close_argsc
G sx |r:z || g|�R � W d S t y9 } z t�d�||�� | jr.| �| |� W Y d }~d S W Y d }~d S d }~ww d S )Nzerror from callback {}: {})r@ r �errorrB r/ )r �callbackra r� r r r rM � s ����zWebSocketApp._callback)
NNNNNNNNTNNNN)NNr NrH NNNNFNNNFN)r" r# r$ r% r r ri r8 r rG r� ru rL rM r r r r r Y s4
�R
� ' )r% r �sysrv r> � traceback�_abnfr �_corer r � _exceptionsrH r �__all__r r& r r r r r �<module> s