�
L�gN � � � d dl Z d dlZd dlZd dlmZ ddlmZmZ ddlm Z m
Z
m Z ddl m
Z
mZ ej ej dfgZ eed� � r"e� ej ej df� � eed� � r"e� ej ej d f� � eed
� � r"e� ej ej d f� � eed � � r"e� ej ej d
f� � dag d�Z G d� d� � Zdee e df ddfd�Z dee e df fd�Z dej de de fd�Z!dej de fd�Z"dej dee e#f de fd�Z$dS )� N)�Union� )�"WebSocketConnectionClosedException�WebSocketTimeoutException)�SSLError�SSLWantReadError�SSLWantWriteError)�extract_error_code�extract_err_message� SO_KEEPALIVE� TCP_KEEPIDLE� �
TCP_KEEPINTVL�
� TCP_KEEPCNT� )�DEFAULT_SOCKET_OPTION�sock_opt�setdefaulttimeout�getdefaulttimeout�recv� recv_line�sendc �" � e Zd Zdededdfd�ZdS )r �sockopt�sslopt�returnNc �@ � |�g }|�i }|| _ || _ d | _ d S �N)r r �timeout)�selfr r s �a/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/websocket/_socket.py�__init__zsock_opt.__init__8 s/ � � �?��G� �>��F� �� � �� ��� � � � )�__name__�
__module__� __qualname__�list�dictr# � r$ r" r r 7 s= � � � � � � �� �d� �t� � � � � � r$ r r r c �
� | a dS )z�
Set the global timeout setting to connect.
Parameters
----------
timeout: int or float
default socket timeout time (in seconds)
N��_default_timeout)r s r" r r B s � � ���r$ c � � t S )z�
Get default timeout
Returns
----------
_default_timeout: int or float
Return the global timeout setting (in seconds) to connect.
r, r* r$ r" r r O s
� � �r$ �sock�bufsizec �� � �� � st d� � ��� fd�} � � � � dk r� � �� � }n
|� � }n�# t $ r t d� � �t
j $ r#}t |� � }t |� � �d }~wt $ r=}t |� � }t |t � � rd|v rt |� � �� d }~ww xY w|st d� � �|S )N�socket is already closed.c �� �� �� �� � S # t $ r Y nEt j $ r4} t | � � }|t
j t
j fvr� Y d } ~ nd } ~ ww xY wt j � � }|�
�t j � � |� ��
� � � � }|� � � |r�� �� � S d S r )r r �socket�errorr
�errno�EAGAIN� EWOULDBLOCK� selectors�DefaultSelector�register�
EVENT_READ�select�
gettimeout�close)�exc�
error_code�sel�rr0 r/ s ��r" �_recvzrecv.<locals>._recv_ s� �� � ��9�9�W�%�%� %�� � � � � �D��|� � � �+�C�0�0�J��%�,��0A�!B�B�B�� C�B�B�B�B����� ����
�'�)�)�� � � �T�9�/�0�0�0� �J�J�t���(�(� )� )�� � � � � � � � &��9�9�W�%�%� %� &� &s � �
A&�A&�*A!�!A&r zConnection timed out� timed outz#Connection to remote host was lost.) r r> r � TimeoutErrorr r4 r r r �
isinstance�str)r/ r0 rD �bytes_�e�messages `` r" r r [ sB ��� � � N�0�1L�M�M�M�&� &� &� &� &� &�&� �?�?� � �� !� !��Y�Y�w�'�'�F�F��U�W�W�F��� � @� @� @�'�(>�?�?�?� �>� 1� 1� 1�%�a�(�(��'��0�0�0����� � � � �%�a�(�(�� �g�s� #� #� � �w�(>�(>�+�G�4�4� 4� ����� ���� � X�0�1V�W�W�W� �Ms# �8A �'C#�; B�
C#�&8C � C#c � � g } t | d� � }|� |� � |dk rn�-d� |� � S )NTr �
r$ )r �append�join)r/ �line�cs r" r r � sK � �
�D�� ��q�M�M�� � � �A���� ��:�:� � �
�8�8�D�>�>�r$ �datac �� � �� t �t � � r�� d� � �� st d� � ��� fd�} � � � � dk r� � �� � S |� � S # t j $ r#}t |� � }t |� � �d }~wt $ r=}t |� � }t |t � � rd|v rt |� � �� d }~ww xY w)Nzutf-8r2 c �� �� �� �� � S # t $ r Y nHt j $ r7} t | � � }|�� |t
j t
j fvr� Y d } ~ nd } ~ ww xY wt j � � }|�
�t j � � |� ��
� � � � }|� � � |r�� �� � S d S r )r r r4 r5 r
r6 r7 r8 r9 r: r; � EVENT_WRITEr= r> r? )r@ rA rB �wrR r/ s ��r" �_sendzsend.<locals>._send� s �� � ��9�9�T�?�?� "�� � � � � �D��|� � � �+�C�0�0�J��!���%�,��0A�!B�B�B�� C�B�B�B�B����� ���� �'�)�)�� � � �T�9�0�1�1�1� �J�J�t���(�(� )� )�� � � � � � � � #��9�9�T�?�?� "� #� #s � �
A)�A)�-A$�$A)r rE ) rG rH �encoder r> r r4 r r r � Exception)r/ rR rW rJ rK s `` r" r r � s ��� ��$�� � � $��{�{�7�#�#�� � N�0�1L�M�M�M�#� #� #� #� #� #�*
� �?�?� � �� !� !��9�9�T�?�?� "��5�7�7�N�� �>� 1� 1� 1�%�a�(�(��'��0�0�0����� � � � �%�a�(�(�� �g�s� #� #� � �w�(>�(>�+�G�4�4� 4� ����� ���s* �,A< �2 A< �<C3� B)�)
C3�68C.�.C3)%r6 r9 r4 �typingr � _exceptionsr r � _ssl_compatr r r �_utilsr
r �SOL_TCP� TCP_NODELAYr �hasattrrN �
SOL_SOCKETr r
r r r- �__all__r �int�floatr r �bytesr r rH r r* r$ r" �<module>rf s� �� � � � � � � � �
�
�
�
� � � � � � �� � � � � � � � G� F� F� F� F� F� F� F� F� F� ;� ;� ;� ;� ;� ;� ;� ;��& !�.�&�*<�a�@�A� �
�7�6�>�"�"� N�� � �&�"3�V�5H�!�!L�M�M�M�
�7�6�>�"�"� L�� � �&�.�&�2E�r�!J�K�K�K�
�7�6�?�#�#� M�� � �&�.�&�2F��!K�L�L�L�
�7�6�=�!�!� J�� � �&�.�&�2D�a�!H�I�I�I�� � � � �� � � � � � � � �
�u�S�%��%5� 6�
�4�
�
�
�
� �5��e�T�!1�2� � � � �+�v�}� +�s� +�u� +� +� +� +�\�F�M� �e� � � � �)�v�}� )�E�%��*�$5� )�#� )� )� )� )� )� )r$