U
    <{f,                     @  sd  d dl mZ d dlZd dlZd dlZd dlmZmZmZm	Z	 ddl
mZ ddlmZ ddlmZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZmZmZ ddlmZmZmZ ddlm Z  ddl!m"Z" ddl#m$Z$ dddgZ%G dd de"Z&dddddddddedddddddddddddd d!d"d#ddd$d$d%d&d'dd(d)dZ'd-ddd*dd+d,dZ(dS ).    )annotationsN)AnyOptionalSequenceType   )ClientProtocol)HeadersLike)ClientExtensionFactory) enable_client_permessage_deflate)validate_subprotocols)
USER_AGENT)Response)
CONNECTINGOPENEvent)
LoggerLikeOriginSubprotocol)	parse_uri   )
Connection)Deadlineconnectunix_connectClientConnectionc                      st   e Zd ZdZddddddd fd	d
ZdedfdddddddZddd fddZdd fddZ  Z	S )r   a  
    Threaded implementation of a WebSocket client connection.

    :class:`ClientConnection` provides :meth:`recv` and :meth:`send` methods for
    receiving and sending messages.

    It supports iteration to receive messages::

        for message in websocket:
            process(message)

    The iterator exits normally when the connection is closed with close code
    1000 (OK) or 1001 (going away) or without a close code. It raises a
    :exc:`~websockets.exceptions.ConnectionClosedError` when the connection is
    closed with any other code.

    Args:
        socket: Socket connected to a WebSocket server.
        protocol: Sans-I/O connection.
        close_timeout: Timeout for closing the connection in seconds.

    
   close_timeoutzsocket.socketr   Optional[float]None)socketprotocolr   returnc                  s$   |  t  | _t j|||d d S )Nr   )	threadingr   response_rcvdsuper__init__)selfr!   r"   r   	__class__ U/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/websockets/sync/client.pyr'   1   s    
zClientConnection.__init__NOptional[HeadersLike]Optional[str])additional_headersuser_agent_headertimeoutr#   c              	   C  s   | j tdJ | j | _|dk	r0| jj| |dk	rD|| jjd< | j| j W 5 Q R X | j	|s| 
  | j  td| jdkr| 
  | j  td| jjtk	r| j| j | 
  | j  | jjdk	r| jjdS )z1
        Perform the opening handshake.

        )Zexpected_stateNz
User-Agentztimed out during handshakez"connection closed during handshake)Zsend_contextr   r"   r   requestheadersupdatesend_requestr%   waitZclose_socketZrecv_events_threadjoinTimeoutErrorresponseConnectionErrorstater   r   handshake_exc)r(   r/   r0   r1   r+   r+   r,   	handshake@   s*    




zClientConnection.handshaker   )eventr#   c                   s:   | j dkr*t|tst|| _ | j  nt | dS )z.
        Process one incoming event.

        N)r9   
isinstancer   AssertionErrorr%   setr&   process_event)r(   r>   r)   r+   r,   rB   d   s
    
zClientConnection.process_event)r#   c                   s    zt   W 5 | j   X dS )zI
        Read incoming data from the socket and process events.

        N)r%   rA   r&   recv_events)r(   r)   r+   r,   rC   r   s    zClientConnection.recv_events)
__name__
__module____qualname____doc__r'   r   r=   rB   rC   __classcell__r+   r+   r)   r,   r      s   $Fdeflater   i   )sockssl_contextserver_hostnameunixpathorigin
extensionssubprotocolsr/   r0   compressionopen_timeoutr   max_sizeloggercreate_connectionstrzOptional[socket.socket]zOptional[ssl.SSLContext]r.   boolzOptional[Origin]z*Optional[Sequence[ClientExtensionFactory]]zOptional[Sequence[Subprotocol]]r-   r   zOptional[int]zOptional[LoggerLike]z Optional[Type[ClientConnection]])urirJ   rK   rL   rM   rN   rO   rP   rQ   r/   r0   rR   rS   r   rT   rU   rV   r#   c             	   C  s  t | }|js|dk	rtd|rV|dkr<|dkr<tdqb|dk	rb|dk	rbtdn|dksbt|dk	rrt| |dkrt|}n|dk	rtd| t|}|dkrt}z|dkr|rt		t	j
t	j}||  |dk	st|| nt	|j|jf| }|d |s0|t	jt	jd |jr|dkrJt }|dkrZ|j}||  |j||d}|d t||||t||d	}||||d
}||	|
|  W n* tk
r   |dk	r|   Y nX |S )a)
  
    Connect to the WebSocket server at ``uri``.

    This function returns a :class:`ClientConnection` instance, which you can
    use to send and receive messages.

    :func:`connect` may be used as a context manager::

        async with websockets.sync.client.connect(...) as websocket:
            ...

    The connection is closed automatically when exiting the context.

    Args:
        uri: URI of the WebSocket server.
        sock: Preexisting TCP socket. ``sock`` overrides the host and port
            from ``uri``. You may call :func:`socket.create_connection` to
            create a suitable TCP socket.
        ssl_context: Configuration for enabling TLS on the connection.
        server_hostname: Host name for the TLS handshake. ``server_hostname``
            overrides the host name from ``uri``.
        origin: Value of the ``Origin`` header, for servers that require it.
        extensions: List of supported extensions, in order in which they
            should be negotiated and run.
        subprotocols: List of supported subprotocols, in order of decreasing
            preference.
        additional_headers (HeadersLike | None): Arbitrary HTTP headers to add
            to the handshake request.
        user_agent_header: Value of  the ``User-Agent`` request header.
            It defaults to ``"Python/x.y.z websockets/X.Y"``.
            Setting it to :obj:`None` removes the header.
        compression: The "permessage-deflate" extension is enabled by default.
            Set ``compression`` to :obj:`None` to disable it. See the
            :doc:`compression guide <../../topics/compression>` for details.
        open_timeout: Timeout for opening the connection in seconds.
            :obj:`None` disables the timeout.
        close_timeout: Timeout for closing the connection in seconds.
            :obj:`None` disables the timeout.
        max_size: Maximum size of incoming messages in bytes.
            :obj:`None` disables the limit.
        logger: Logger for this client.
            It defaults to ``logging.getLogger("websockets.client")``.
            See the :doc:`logging guide <../../topics/logging>` for details.
        create_connection: Factory for the :class:`ClientConnection` managing
            the connection. Set it to a wrapper or a subclass to customize
            connection handling.

    Raises:
        InvalidURI: If ``uri`` isn't a valid WebSocket URI.
        OSError: If the TCP connection fails.
        InvalidHandshake: If the opening handshake fails.
        TimeoutError: If the opening handshake times out.

    Nz5ssl_context argument is incompatible with a ws:// URIzmissing path argumentz(path and sock arguments are incompatiblerI   zunsupported compression: T)rL   )rO   rP   rQ   r;   rT   rU   r   )r   secure	TypeErrorr@   r   r   
ValueErrorr   r   r!   AF_UNIXSOCK_STREAM
settimeoutr1   r   rV   hostport
setsockoptIPPROTO_TCPTCP_NODELAYsslcreate_default_contextwrap_socketr   r   r=   	Exceptionclose)rY   rJ   rK   rL   rM   rN   rO   rP   rQ   r/   r0   rR   rS   r   rT   rU   rV   Zwsurideadliner"   
connectionr+   r+   r,   r   ~   s    S









r   )rN   rY   kwargsr#   c                 K  s6   |dkr | ddkrd}nd}tf |d| d|S )a  
    Connect to a WebSocket server listening on a Unix socket.

    This function is identical to :func:`connect`, except for the additional
    ``path`` argument. It's only available on Unix.

    It's mainly useful for debugging servers listening on Unix sockets.

    Args:
        path: File system path to the Unix socket.
        uri: URI of the WebSocket server. ``uri`` defaults to
            ``ws://localhost/`` or, when a ``ssl_context`` is provided, to
            ``wss://localhost/``.

    NrK   zws://localhost/zwss://localhost/T)rY   rM   rN   )getr   )rN   rY   rl   r+   r+   r,   r   /  s
    )NN))
__future__r   r!   re   r$   typingr   r   r   r   clientr   Zdatastructuresr	   Zextensions.baser
   Zextensions.permessage_deflater   r3   r   httpr   Zhttp11r   r"   r   r   r   r   r   r   rY   r   rk   r   utilsr   __all__r   r   r   r+   r+   r+   r,   <module>   sN   
i4 3  