Format: application/x-python-code
1 o
2
3 7'hl � @ s� d dl mZ d dlZd dlZd dlZddlmZmZmZm Z m
4 Z
5 m Z m Z m
6 Z
7 ddlmZmZmZ G dd� de�ZG d d
8 � d
9 e�ZdS ) � )� annotationsN� )� ConnectError�ConnectTimeout�ExceptionMapping� ReadError� ReadTimeout�
10 WriteError� WriteTimeout�map_exceptions� )�
11 SOCKET_OPTION�AsyncNetworkBackend�AsyncNetworkStreamc @ s\ e Zd Zd#dd�Zd$d%d
12 d�Zd$d&dd�Zd'dd�Z d(d)dd�Zd*d d �Zd+d!d"�Z dS ),�
13 TrioStream�stream�trio.abc.Stream�return�Nonec C s
14 || _ d S �N)�_stream��selfr � r �MC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpcore/_backends/trio.py�__init__ s
15 zTrioStream.__init__N� max_bytes�int�timeout� float | None�bytesc � s� �|d u r t d�n|}tjttjttjti}t|��1 t�|�� | j j
16 |d�I d H }|W d � W d � S 1 s>w Y W d � d S 1 sNw Y d S )N�inf)r ) �float�trio� TooSlowErrorr �BrokenResourceErrorr �ClosedResourceErrorr �
17 fail_afterr Z receive_some)r r r �timeout_or_inf�exc_map�datar r r �read s ��
18 ��"�zTrioStream.read�bufferc � s� �|sd S |d u r
19 t d�n|}tjttjttjti}t|��0 t�|�� | j j
20 |d�I d H W d � n1 s9w Y W d � d S W d � d S 1 sQw Y d S )Nr! )r* ) r" r# r$ r
21 r% r r& r r' r Zsend_all)r r, r r( r) r r r �write% s ��
22 ��"�zTrioStream.writec � s �| j �� I d H d S r )r �aclose)r r r r r. 3 � �zTrioStream.aclose� ssl_context�ssl.SSLContext�server_hostname�
23 str | Noner c � s� �|d u r t d�n|}tjttjti}tj| j||ddd�}t|��? z t� |�� |�
24 � I d H W d � n1 s;w Y W n t yV } z | � � I d H |�d }~ww W d � t
25 |�S 1 sdw Y t
26 |�S )Nr! TF)r0 r2 Zhttps_compatible� server_side)r" r# r$ r r% r � SSLStreamr r r' � do_handshake� Exceptionr. r )r r0 r2 r r( r) Z
27 ssl_stream�excr r r � start_tls6 s8 ���
28 �����
29 ��zTrioStream.start_tls�info�str�
30 typing.Anyc C s� |dkrt | jtj�r| jjS |dkr| �� j�� S |dkr%| �� j�� S |dkrF| j}t |tj�r;|j }t |tj�s2t |tj
31 �sCJ �|jS |dkrS| � d�}|� � S d S )N�
32 ssl_object� client_addrZ server_addr�socket� is_readable)
33 �
34 isinstancer r# r5 Z _ssl_object�_get_socket_streamr? � getsockname� getpeername�transport_stream� SocketStream�get_extra_infor@ )r r: r r? r r r rG Q s" �
35 zTrioStream.get_extra_info�trio.SocketStreamc C s8 | j }t|tj�r|j}t|tj�s t|tj�sJ �|S r )r rA r# r5 rE rF r r r r rB e s �z TrioStream._get_socket_stream)r r r r r )r r r r r r )r, r r r r r )r r �NN)r0 r1 r2 r3 r r r r )r: r; r r< )r rH )
36 �__name__�
37 __module__� __qualname__r r+ r- r. r9 rG rB r r r r r s
38
39 �
40 r c @ s8 e Zd Z dddd�Z dddd�Zd dd�ZdS ) � TrioBackendN�hostr; �portr r r �
41 local_addressr3 �socket_options�%typing.Iterable[SOCKET_OPTION] | Noner r c
42 � s� �|d u rg }|d u rt d�n|}tjttjttti}t|��? t�|�� tj |||d�I d H }|D ]} |j
43 | � q3W d � n1 sEw Y W d � t |�S W d � t |�S 1 saw Y t |�S )Nr! )rN rO rP ) r" r# r$ r r% r �OSErrorr r' Zopen_tcp_stream�
44 setsockoptr )
45 r rN rO r rP rQ r( r) r �optionr r r � connect_tcpn s0 �
46 �
47 � � �
48 ��
49 ��zTrioBackend.connect_tcp�pathc � s� �|d u rg }|d u rt d�n|}tjttjttti}t|��< t�|�� t� |�I d H }|D ]}|j
50 |� q0W d � n1 sBw Y W d � t |�S W d � t |�S 1 s^w Y t |�S )Nr! ) r" r# r$ r r% r rS r r' Zopen_unix_socketrT r )r rW r rQ r( r) r rU r r r �connect_unix_socket� s, ��
51 � �
52 ��
53 ��z TrioBackend.connect_unix_socket�secondsr" r c � s �t �|�I d H d S r )r# �sleep)r rY r r r rZ � r/ zTrioBackend.sleep)NNN) rN r; rO r r r rP r3 rQ rR r r rI )rW r; r r rQ rR r r )rY r" r r )rJ rK rL rV rX rZ r r r r rM m s � �rM )�
54 __future__r �ssl�typingr# � _exceptionsr r r r r r r
55 r �baser
56 r r r rM r r r r �<module> s (
57 X