Format: application/x-python-code
1 o
2
3 7'h86 � @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZddl m
4 Z
5 ddl m Z m
6 Z
7 mZmZmZ ddlmZmZmZ ddlmZmZ ddlmZ d d
8 lmZ e� d �Z ej ejej ej f Z!G d d
9 � d
10 ej"�Z#G dd� de�Z$G dd� d�Z%G dd� de
11 �Z&dS )� )� annotationsN� )�AsyncNetworkStream)�ConnectionNotAvailable�LocalProtocolError�RemoteProtocolError�
12 WriteError�map_exceptions)�Origin�Request�Response)� AsyncLock�AsyncShieldCancellation)�Trace� )�AsyncConnectionInterfacezhttpcore.http11c @ s e Zd ZdZdZdZdZdS )�HTTPConnectionStater r r � N)�__name__�
13 __module__� __qualname__�NEW�ACTIVE�IDLE�CLOSED� r r �LC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpcore/_async/http11.pyr $ s
14 r c @ s� e Zd ZdZdZ dCdDd d
15 �ZdEdd�ZdFdd�ZdFdd�ZdCdGdd�Z dHd d �Z
16 dId d!�Z dCdJd#d$�Z dKd%d&�Z
17 dKd'd(�ZdLd*d+�ZdMd,d-�ZdMd.d/�ZdMd0d1�ZdMd2d3�ZdNd5d6�ZdNd7d8�ZdOd9d:�Z dPdQdAdB�ZdS )R�AsyncHTTP11Connectioni i � N�originr
18 �streamr �keepalive_expiry� float | None�return�Nonec C sF || _ || _|| _d | _tj| _t� | _d| _ t
19 j t
20 j | j
21 d�| _d S )Nr )Zour_roleZmax_incomplete_event_size)�_origin�_network_stream�_keepalive_expiry�
22 _expire_atr r �_stater
23 � _state_lock�_request_count�h11�
24 ConnectionZCLIENT�MAX_INCOMPLETE_EVENT_SIZE�
25 _h11_state)�selfr r r r r r �__init__/ s �z AsyncHTTP11Connection.__init__�requestr r c
26 � s� �| � |jj�std|jj� d| j� ���| j4 I d H �& | jtjtj fv r5| j
27 d7 _
28 tj | _d | _ nt
29 � �W d �I d H n 1 I d H sHw Y z�d|i}zZtdt||�4 I d H �}| jdi |��I d H W d �I d H n 1 I d H szw Y tdt||�4 I d H �}| jdi |��I d H W d �I d H n 1 I d H s�w Y W n ty� Y nw tdt||�4 I d H �"}| jdi |��I d H \}}}}}||||f|_W d �I d H n 1 I d H s�w Y | j} |dk�s
30 |jd k�rd
31 | k�rd k �rn nt| |�} t||t| |�||| d �d
32 �W S t�yr }
33 zGt� �9 tdt|�4 I d H �}| � � I d H W d �I d H n1 I d H �sNw Y W d � |
34 �W d � |
35 �1 �sgw Y |
36 �d }
37 ~
38 ww )Nz Attempted to send request to z on connection to r r1 Zsend_request_headersZsend_request_bodyZreceive_response_headers�e s CONNECT�� i, )� http_version�
39 reason_phrase�network_stream)�status�headers�content�
40 extensions�response_closedr ) �can_handle_request�urlr � RuntimeErrorr$ r) r( r r r r* r r' r r �logger�_send_request_headers�_send_request_bodyr �_receive_response_headers� return_valuer% �method�AsyncHTTP11UpgradeStreamr �HTTP11ConnectionByteStream�
41 BaseExceptionr �_response_closed) r/ r1 �kwargs�tracer4 r7 r5 r8 �
42 trailing_datar6 �excr r r �handle_async_requestA s� � ��(� �(�(�� � ���(�
43 (
44 ��
45 ,�
46 ��
47 ����z*AsyncHTTP11Connection.handle_async_requestc � sz �|j �di �}|�dd �}ttjti�� tj|j|jj|j d�}W d � n1 s,w Y | j
48 ||d�I d H d S )N�timeout�write)rD �targetr8 �rN ) r: �getr r+ r r rD r= rP r8 � _send_event�r/ r1 �timeoutsrN �eventr r r r@ � s � � �z+AsyncHTTP11Connection._send_request_headersc � s� �|j �di �}|�dd �}t|jtj�sJ �|j2 z3 d H W }tj|d�}| j||d�I d H q6 | jt� � |d�I d H d S )NrN rO )�datarQ )
49 r: rR �
50 isinstancer �typing�
51 AsyncIterabler+ �DatarS � EndOfMessage)r/ r1 rU rN �chunkrV r r r rA � s � � z(AsyncHTTP11Connection._send_request_bodyrV � h11.EventrN c � s4 �| j �|�}|d ur| jj||d�I d H d S d S )NrQ )r. �sendr% rO )r/ rV rN Z
52 bytes_to_sendr r r rS � s
53 � �z!AsyncHTTP11Connection._send_event�:tuple[bytes, int, bytes, list[tuple[bytes, bytes]], bytes]c � s� �|j �di �}|�dd �} | j|d�I d H }t|tj�r n
54 t|tj�r+|jdkr+nqd|j }|j �
55 � }| j j \}}||j|j
56 ||fS )NrN �readTrQ r2 s HTTP/)r: rR �_receive_eventrX r+ r ZInformationalResponse� status_coder4 r8 � raw_itemsr. rK �reason) r/ r1 rU rN rV r4 r8 rK �_r r r rB � s �
57 �
58 �
59
60
61 z/AsyncHTTP11Connection._receive_response_headers�typing.AsyncIterator[bytes]c C sb �|j �di �}|�dd �} | j|d�I d H }t|tj�r%t|j�V n t|tjtj f�r0d S q)NrN ra TrQ )
62 r: rR rb rX r+ r[ �bytesrW r\ ZPAUSEDrT r r r �_receive_response_body� s � �z,AsyncHTTP11Connection._receive_response_body� h11.Event | type[h11.PAUSED]c � s� � t tjti��
63 | j�� }W d � n1 sw Y |tju rG| jj| j|d�I d H }|dkr@| jj tj
64 kr@d}t|��| j� |� n|S q)NTrQ � z/Server disconnected without sending a response.) r r+ r r. Z
65 next_eventZ NEED_DATAr% ra �READ_NUM_BYTES� their_stateZ
66 SEND_RESPONSEZ receive_data)r/ rN rV rW �msgr r r rb � s � �
67 � �z$AsyncHTTP11Connection._receive_eventc � s� �| j 4 I d H �E | jjtju r0| jjtju r0tj| _| j� � | j
68 d ur/t � � }|| j
69 | _
70 n| �� I d H W d �I d H d S W d �I d H d S 1 I d H sSw Y d S �N)r) r. Z our_stater+ ZDONErm r r r( Zstart_next_cycler& �time� monotonicr' �aclose)r/ �nowr r r rH � s �
71
72 ���. z&AsyncHTTP11Connection._response_closedc � s �t j| _| j�� I d H d S ro )r r r( r% rr �r/ r r r rr � s �z AsyncHTTP11Connection.aclose�boolc C s
73 || j kS ro )r$ �r/ r r r r r< s
74 z(AsyncHTTP11Connection.can_handle_requestc C � | j tjkS ro �r( r r rt r r r � is_available s z"AsyncHTTP11Connection.is_availablec C s< t �� }| jd uo
75 || jk}| jtjko| j�d�}|p |S )NZ is_readable)rp rq r' r( r r r% �get_extra_info)r/ rs Zkeepalive_expiredZserver_disconnectedr r r � has_expired s
76 �z!AsyncHTTP11Connection.has_expiredc C rw ro rx rt r r r �is_idle � z AsyncHTTP11Connection.is_idlec C rw ro )r( r r rt r r r � is_closed# r} z AsyncHTTP11Connection.is_closed�strc C s$ t | j�}|�d| jj� d| j� �S )Nz , HTTP/1.1, �, Request Count: )r r$ r( �namer* rv r r r �info& s
77
78 ��zAsyncHTTP11Connection.infoc C s6 | j j}t| j�}d|� d|�d| jj� d| j� d� S )N�<z [z, r� z]>)� __class__r r r$ r( r� r* )r/ �
79 class_namer r r r �__repr__- s
80 ��z AsyncHTTP11Connection.__repr__c � s �| S ro r rt r r r �
81 __aenter__8 s �z AsyncHTTP11Connection.__aenter__�exc_type�type[BaseException] | None� exc_value�BaseException | None� traceback�types.TracebackType | Nonec � s �| � � I d H d S ro )rr )r/ r� r� r� r r r � __aexit__; s �z AsyncHTTP11Connection.__aexit__ro )r r
82 r r r r! r" r# )r1 r r" r )r1 r r" r# )rV r^ rN r! r" r# )r1 r r" r` )r1 r r" rg )rN r! r" rj �r" r# )r r
83 r" ru )r" ru )r" r )r" r )NNN)r� r� r� r� r� r� r" r# )r r r rl r- r0 rM r@ rA rS rB ri rb rH rr r< ry r{ r| r~ r� r� r� r� r r r r r + s6 �
84
85 K
86
87
88 �
89
90
91
92
93
94
95
96
97
98
99 �r c @ s* e Zd Zddd�Zdd
100 d �Zdd d
101 �ZdS )rF �
102 connectionr r1 r r" r# c C s || _ || _d| _d S )NF)� _connection�_request�_closed)r/ r� r1 r r r r0 E s
103 z#HTTP11ConnectionByteStream.__init__rg c C s� �d| j i}z;tdt| j |�4 I d H �! | jjdi |��2 z 3 d H W }|V q 6 W d �I d H W d S 1 I d H s:w Y W d S tyj } z t� � | �� I d H W d � |�1 s_w Y |�d }~ww )Nr1 Zreceive_response_bodyr )r� r r? r� ri rG r rr )r/ rI r] rL r r r � __aiter__J s" �
104 �2�
105 ����z$HTTP11ConnectionByteStream.__aiter__c � sj �| j s3d| _ tdt| j�4 I d H � | j�� I d H W d �I d H d S 1 I d H s,w Y d S d S )NTr; )r� r r? r� r� rH rt r r r rr X s �.��z!HTTP11ConnectionByteStream.acloseN)r� r r1 r r" r# )r" rg r� )r r r r0 r� rr r r r r rF D s
106
107 rF c @ sR e Zd Zd dd�Zd!d"dd�Zd!d#dd�Zd$dd�Z d%d&dd�Zd'd d �Zd S )(rE r r � leading_datarh r" r# c C s || _ || _d S ro )�_stream�
108 _leading_data)r/ r r� r r r r0 ` s
109 z!AsyncHTTP11UpgradeStream.__init__N� max_bytes�intrN r! c � s> �| j r| j d |� }| j |d � | _ |S | j�||�I d H S ro )r� r� ra )r/ r� rN �bufferr r r ra d s �z AsyncHTTP11UpgradeStream.readr� c � s �| j �||�I d H d S ro )r� rO )r/ r� rN r r r rO l s �z AsyncHTTP11UpgradeStream.writec � s �| j �� I d H d S ro )r� rr rt r r r rr o s �z AsyncHTTP11UpgradeStream.aclose� ssl_context�ssl.SSLContext�server_hostname�
110 str | Nonec � s �| j �|||�I d H S ro )r� � start_tls)r/ r� r� rN r r r r� r s �z"AsyncHTTP11UpgradeStream.start_tlsr� r �
111 typing.Anyc C s | j �|�S ro )r� rz )r/ r� r r r rz z r} z'AsyncHTTP11UpgradeStream.get_extra_info)r r r� rh r" r# ro )r� r� rN r! r" rh )r� rh rN r! r" r# r� )NN)r� r� r� r� rN r! r" r )r� r r" r� ) r r r r0 ra rO rr r� rz r r r r rE _ s
112
113 �rE )'�
114 __future__r �enum�logging�sslrp �typesrY r+ Z_backends.baser � _exceptionsr r r r r Z_modelsr
115 r r Z_synchronizationr
116 r �_tracer Z
117 interfacesr � getLoggerr? �Unionr[ r\ Z H11SendEvent�IntEnumr r rF rE r r r r �<module> s6
118 ��