File: //usr/local/CyberCP/lib64/python3.10/site-packages/websocket/__pycache__/_app.cpython-310.pyc
o
�hp^ � @ s d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZm Z ddl
m Z ddl m
Z
ddlmZmZ ddlmZmZmZ ddlmZ dd lmZ d
gZd ad ed dfd
d�Z G dd� d�Z G dd� de �Z G dd� de �Z G dd� d�Z G dd
� d
�Z!dS )� N)�Any�Callable�Optional�Union� )�_logging)�ABNF)� WebSocket�getdefaulttimeout)�"WebSocketConnectionClosedException�WebSocketException�WebSocketTimeoutException)� SSLEOFError)� parse_url� WebSocketApp�reconnectInterval�returnc C s | a d S �N)� RECONNECT)r � r �A/usr/local/CyberCP/lib/python3.10/site-packages/websocket/_app.py� setReconnect+ s r c @ sf e Zd ZdZdedeeedf ddfdd�Zdeeedf d e ddfd
d �Z
ded e ddfd
d�Z dS )�DispatcherBasez
DispatcherBase
�app� ping_timeoutNr c C s || _ || _d S r )r r )�selfr r r r r �__init__5 s
zDispatcherBase.__init__�seconds�callbackc C s t �|� |� d S r )�time�sleep�r r r r r r �timeout9 s
zDispatcherBase.timeout� reconnectorc
C sh z t �d|� dtt�� �� d�� t�|� |dd� W d S ty3 } z
t �d|� �� |�d }~ww )Nzreconnect() - retrying in z
seconds [� frames in stack]T)� reconnectingz User exited )r �info�len�inspect�stackr r �KeyboardInterrupt)r r r# �er r r � reconnect= s �
��zDispatcherBase.reconnect) �__name__�
__module__� __qualname__�__doc__r r �float�intr r r" r, r r r r r 0 s
r c @ s, e Zd ZdZdejdededdfdd�ZdS ) �
Dispatcherz
Dispatcher
�sock�
read_callback�check_callbackr Nc C s| t �� }|�| jjjt j� z*| jjr+|�| j�r |� s n|� | jjsW |� � d S W |� � d S W |� � d S |� � w r )
� selectors�DefaultSelector�registerr r4 �
EVENT_READ� keep_running�selectr �close�r r4 r5 r6 �selr r r �readN s
� � �zDispatcher.read)r- r. r/ r0 �socketr r@ r r r r r3 I s ����r3 c @ s< e Zd ZdZdejdededdfdd�Zd ejfd
d �Z dS ) �
SSLDispatcherz
SSLDispatcher
r4 r5 r6 r Nc C s� | j jj}t�� }|�|tj� z*| j jr-| �||�r |� s n|� | j jsW |�� d S W |�� d S W |�� d S |�� w r ) r r4 r7 r8 r9 r: r; r< r= r> r r r r@ e s
� � �zSSLDispatcher.readr? c C s@ | j jj}|�� r |gS |�| j�}t|�dkr |d d S d S )Nr )r r4 �pendingr< r r' )r r4 r? �rr r r r< w s
� �zSSLDispatcher.select)
r- r. r/ r0 rA r r@ r7 r8 r< r r r r rB ` s ���
�rB c @ st e Zd ZdZdeeedf ddfdd�Zdejde d e ddfd
d �Z
d ed
e ddfdd�Z d ede ddfdd�Z dS )�WrappedDispatcherz
WrappedDispatcher
r Nr c C s$ || _ || _|| _|�d|j� d S )N� )r r �
dispatcher�signal�abort)r r r rG r r r r � s zWrappedDispatcher.__init__r4 r5 r6 c C s, | j �||� | jo| �| j|� d S d S r )rG r@ r r" )r r4 r5 r6 r r r r@ � s zWrappedDispatcher.readr r c C s | j �||� d S r )rG r" r! r r r r" � s zWrappedDispatcher.timeoutr# c C s | � ||� d S r )r" )r r r# r r r r, � s zWrappedDispatcher.reconnect)
r- r. r/ r0 r r1 r2 r rA r r@ r" r, r r r r rE � s ���
� rE c # @ sV e Zd ZdZ dFdedeeeedf de ee
gdf de ee
gdf de ee
e gdf d e ee
e gdf d
e ee
e e gdf d e e d e e d
e e de de e de e de e de e de e
j
ddf"dd�Zejfdeeef deddfdd�Zdeddfd d �Zdeeef ddfd d �ZdGd d!�ZdGd"d#�ZdGd$d%�ZdGd&d'�Z ( ) * * dHd+ed,ed-ee ef d.ee edf d/ed0ed1eeef d2ed3ed4e e d5e d6ed7ed8e d9ed:ede f"d;d<�Z *dId.ee edf d=e e d>e dee e e!f fd?d@�Z"dAedefdBdC�Z#dGdDdE�Z$dS )Jr z_
Higher level of APIs are provided. The interface is like JavaScript WebSocket object.
NT�url�header�on_open� on_reconnect�
on_message�on_error�on_close�on_ping�on_pong�on_cont_messager; � get_mask_key�cookie� subprotocols�on_datarA r c C s� || _ |dur |ng | _|
| _|| _|| _|| _|| _|| _|| _|| _ | | _
|
| _ d| _ | | _
d| _td�| _td�| _d| _d| _td�| _d| _d| _|| _|| _d| _d| _t� � | _ dS )a�
WebSocketApp initialization
Parameters
----------
url: str
Websocket url.
header: list or dict or Callable
Custom header for websocket handshake.
If the parameter is a callable object, it is called just before the connection attempt.
The returned dict or list is used as custom header value.
This could be useful in order to properly setup timestamp dependent headers.
on_open: function
Callback object which is called at opening websocket.
on_open has one argument.
The 1st argument is this class object.
on_reconnect: function
Callback object which is called at reconnecting websocket.
on_reconnect 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.
socket: socket
Pre-initialized stream socket.
NFr � ) rJ rK rU rL rM rN rW rO rP rQ rR rS r; rT r4 r1 � last_ping_tm� last_pong_tm� ping_thread� stop_ping�
ping_intervalr � ping_payloadrV �prepared_socket� has_errored�has_done_teardown� threading�Lock�has_done_teardown_lock)r rJ rK rL rM rN rO rP rQ rR rS r; rT rU rV rW rA r r r r � s6 T
zWebSocketApp.__init__�data�opcodec 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 � Connection is already closed.N)r4 �sendr )r re rf r r r rh s
�zWebSocketApp.send� text_datac C �& | j r
| j �|tj�dkrtd��dS )z+
Sends UTF-8 encoded text.
r rg N)r4 rh r � OPCODE_TEXTr )r ri r r r � send_text% � �zWebSocketApp.send_textc C rj )z,
Sends a sequence of bytes.
r rg N)r4 rh r �
OPCODE_BINARYr )r re r r r �
send_bytes, rm zWebSocketApp.send_bytesc K s, d| _ | jr| jjdi |�� d| _dS dS )z-
Close websocket connection.
FNr )r; r4 r= )r �kwargsr r r r= 3 s
�zWebSocketApp.closec C s@ t d� | _| _t�� | _tj| jd�| _d| j_ | j�
� d S )Nr )�targetT) r1 rY rZ rb �Eventr\ �Thread�
_send_pingr[ �daemon�start�r r r r �_start_ping_thread< s
z WebSocketApp._start_ping_threadc C s@ | j r| j �� | jr| j�� r| j�d� td� | _| _d S )N� r )r\ �setr[ �is_alive�joinr1 rY rZ rw r r r �_stop_ping_threadC s
z WebSocketApp._stop_ping_threadc
C s� | j �| j�s | jdu rd S | j �| j�sY| jdu r[| jrKt�� | _zt�d� | j� | j
� W n t yJ } zt�d|� �� W Y d }~nd }~ww | j �| j�s]| jdu sd S d S d S d S )NFTz Sending pingzFailed to send ping: ) r\ �waitr] r; r4 r rY r �debug�pingr^ � Exception)r r+ r r r rt J s
��(�zWebSocketApp._send_pingr rX F�sockopt�ssloptr] r r^ �http_proxy_host�http_proxy_port�
http_no_proxy�http_proxy_auth�http_proxy_timeout�skip_utf8_validation�host�origin�suppress_origin�
proxy_typer, c s �
du rt �
|dur|dkrtd��|dur |dk r td��|r*|r*||kr*td���s.d��s2i ��jr9td��|�_|�_|�_d�_d �_d d
tf�fd d �
�d d
t ddf� �������� �
� � �����fdd�
�dt f���
���fdd� � dt f�fdd� � d dt
t t t
ttf d
t dt f���
���fdd�
�t �����|�t�j�d ��zKz"�� �s҈
r҈jr�t�dtt�� �� d�� ���
�� �js�W n t
tfy� } zt�d|� �� �� W Y d}~nd}~ww W �s��� �jS ��s�� w w ) 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.
http_proxy_timeout: int or float
HTTP proxy timeout, default is 60 sec as per python-socks.
http_proxy_auth: tuple
HTTP proxy auth information. tuple of username and password. Default is None.
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.
proxy_type: str
type of proxy from: http, socks4, socks4a, socks5, socks5h
reconnect: int
delay interval when reconnecting
Returns
-------
teardown: bool
False if the `WebSocketApp` is closed or caught KeyboardInterrupt,
True if any other exception was raised during a loop.
Nr zEnsure ping_timeout > 0zEnsure ping_interval >= 0z#Ensure ping_interval > ping_timeoutr zsocket is already openedFT� close_framec s� � j � � jr W d � dS d� _W d � n1 s w Y � �� d� _� jr1� j�� � �| r7| 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.
NTF) rd ra r} r; r4 r= �_get_close_args� _callbackrP )r� �close_status_code� close_reasonrw r r �teardown� s
� �
�z*WebSocketApp.run_forever.<locals>.teardownr% r c s | r
� j r
� j �� t� j��� jd u�
dd�� _ � j �t� � zNt� j�r*� �� n� j}� j j � j
|� j ������ j �� ��
� j
d� t�d� � jrQ� �� | r]� jr]� �� j� n� �� j� ��� j j � � � W d S tttttfy� } z �|| � W Y d }~d S d }~ww )NT)r� r� �fire_cont_framer� �enable_multithread)
rK rU r� r� r� r� r� rV r� r� r� r� rA zWebsocket connected)r4 �shutdownr rT rS �
settimeoutr
�callablerK �connectrJ rU rV r_ r r& r] rx rM r� rL r@ r �ConnectionRefusedErrorr* �
SystemExitr� )r% rK r+ )�checkrG �handleDisconnectr� r� r� r� r� r� r� r� r@ r r� r� r� r� r r �setSock� s\
� �
���z)WebSocketApp.run_forever.<locals>.setSockc
s> �j s�� S z
�j�d�\} }W n tttfy0 } z� r*�|t���W Y d }~S |�d }~ww | tjkr:�|�S | tj krI��
�j |j � dS | tj
kr]t�� �_��
�j|j � dS | tjkr}�jr}��
�j|j |j|j� ��
�j|j |j� dS |j }| tjkr��s�|�d�}��
�j||jd� ��
�j|� dS )NT�utf-8)r; r4 �recv_data_framer r* r �boolr � OPCODE_CLOSE� OPCODE_PINGr� rQ re � OPCODE_PONGr rZ rR � OPCODE_CONTrS rW rf �finrk �decoderN )�op_code�framer+ re )�custom_dispatcherr� r, r r� r� r r r@ sB ���
�
��
z&WebSocketApp.run_forever.<locals>.readc sZ � j r+t�� � j � j k} � j� j dk }� j� j � j k}� jr+| r+|s'|r+td��dS )Nr zping/pong timed outT)r r rY rZ r
)�has_timeout_expired�$has_pong_not_arrived_after_last_ping�has_pong_arrived_too_laterw r r r� , s" �������z'WebSocketApp.run_forever.<locals>.checkr+ c s� d�_ ��� |s���j| � t| ttf�r�� � �r>t�| � d�� � r<t� dt
t � � �� d�� ��
��� d S d S t�| � d�� �� d S )NTz - reconnectz%Calling custom dispatcher reconnect [r$ z
- goodbye)r` r} r� rO �
isinstancer* r� r r& r r' r( r) r, �error)r+ r% )r� rG r, r r� r� r r r� C s"
��
z2WebSocketApp.run_forever.<locals>.handleDisconnectry z Calling dispatcher reconnect [r$ ztearing down on exception r )F)r r r4 r] r r^ ra r; r r� r r r� r* r� r� �create_dispatcherr rJ r r r' r( r) r, r& r` )r r� r� r] r r^ r� r� r� r� r� r� r� r� rG r� r� r, r+ r )r� r� rG r� r� r� r� r� r� r� r� r� r@ r, r r� r� r� r� r� r� r � run_foreverV s� I8 6# ����� � �� �������zWebSocketApp.run_foreverrG �is_sslc C s0 |rt | ||�S |p d}|rt| |�S t| |�S )N�
)rE rB r3 )r r rG r� r" r r r r� y s
z WebSocketApp.create_dispatcherr� c C sx | j r|s ddgS |jr8t|j�dkr8dt|jd � t|jd � }|jdd� }t|t�r4|�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)
NrF � r r r� )rP re r' r2 r� �bytesr� )r r� r� �reasonr r r r� � s
�
z WebSocketApp._get_close_argsc
G s| |r<z || g|�R � W d S t y; } z"t�d|� d|� �� | jr0| �| |� W Y d }~d S W Y d }~d S d }~ww d S )Nzerror from callback z: )r� r r� rO )r r �argsr+ r r r r� � s ����zWebSocketApp._callback)NNNNNNNNNTNNNNN)r N)NNr NrX NNNNNFNNNFNN)NF)%r- r. r/ r0 �strr �list�dictr r r r r� rA r r rk r� r2 rh rl � bytearrayro r= rx r} rt �tupler1 r� r r3 rB rE r� r� r� r r r r r � s� �������� �
� � �
�����
�$q
���
� ���
� �
� � �
�����
� (� ���
�
)"r( r7 rA rb r �typingr r r r rX r �_abnfr �_corer r
� _exceptionsr r r
� _ssl_compatr �_urlr �__all__r r2 r r r3 rB rE r r r r r �<module> s* $