Format: application/x-python-code
1 o
2
3 C'h} � @ s d dl mZ d dlZddlmZmZ ddlmZ ddlm Z ej
4 r2d dl Z d dl Z ej
5 e je jf Zejeejf Zejg eje f Zejejeejf gejd f Zejejeejf eegejd f ZdgZdd d �Zddd�ZG dd� de�ZG dd� de �Z dS )� )� annotationsN� )�Request�Response)�AsyncByteStream� )�AsyncBaseTransport�
6 ASGITransport�return�boolc C s6 zdd l } | �� dkrW dS W dS ty Y dS w )Nr �trioTF)�sniffioZcurrent_async_library� ImportError)r
7 � r �LC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpx/_transports/asgi.py�is_running_trio s � ��r �Eventc C s& t � r dd l} | �� S dd l}|�� S )Nr )r r r �asyncio)r r r r r � create_event, s
8 r c @ s e Zd Zd dd�Zd dd �Zd
9 S )
10 �ASGIResponseStream�body� list[bytes]r
11 �Nonec C s
12 || _ d S �N)�_body)�selfr r r r �__init__8 s
13 zASGIResponseStream.__init__�typing.AsyncIterator[bytes]c C s �d� | j�V d S )N� )�joinr )r r r r � __aiter__; s �z ASGIResponseStream.__aiter__N)r r r
14 r )r
15 r )�__name__�
16 __module__� __qualname__r r r r r r r 7 s
17 r c @ s, e Zd ZdZ dddd�Zddd�ZdS )r a�
18 A custom AsyncTransport that handles sending requests directly to an ASGI app.
19
20 ```python
21 transport = httpx.ASGITransport(
22 app=app,
23 root_path="/submount",
24 client=("1.2.3.4", 123)
25 )
26 client = httpx.AsyncClient(transport=transport)
27 ```
28
29 Arguments:
30
31 * `app` - The ASGI application.
32 * `raise_app_exceptions` - Boolean indicating if exceptions in the application
33 should be raised. Default to `True`. Can be set to `False` for use cases
34 such as testing the content of a client 500 response.
35 * `root_path` - The root path on which the ASGI application should be mounted.
36 * `client` - A two-tuple indicating the client IP and port of incoming requests.
37 ```
38 T� �z 127.0.0.1�{ �app�_ASGIApp�raise_app_exceptionsr � root_path�str�client�tuple[str, int]r
39 r c C s || _ || _|| _|| _d S r )r' r) r* r, )r r' r) r* r, r r r r W s
40 zASGITransport.__init__�requestr r c
41 � sJ �t �jt�s J �dddid�jdd� �jjD ��jj�jj�jj �
42 d�d �jj �jj �jj
43 f| j| jd � }�j�� �d
44 �d �d �g � d
45 �t� �d���fd
46 d� }d� �����fdd� }z | �|||�I d H W n ty� | jru� ��� �d u rd��d u r�i �Y nw ��� s�J ��d us�J ��d us�J �t� �}t��|d�S )N�http�versionz3.0z1.1c S s g | ]
47 \}}|� � |f�qS r )�lower)�.0�k�vr r r �
48 <listcomp>o s z6ASGITransport.handle_async_request.<locals>.<listcomp>� ?r ) �typeZasgi� http_version�method�headers�scheme�path�raw_pathZ query_string�serverr, r* Fr
49 �dict[str, typing.Any]c � s^ ��r�� � I d H ddiS z � �� I d H } W n ty( d�dddd� Y S w d| dd�S )Nr7 zhttp.disconnectTz http.requestr F)r7 r � more_body)�wait� __anext__�StopAsyncIteration)r )�request_body_chunks�request_complete�response_completer r �receive� s � � z3ASGITransport.handle_async_request.<locals>.receive�message�&typing.MutableMapping[str, typing.Any]r c � s� �| d dkr�r J �| d �| � dg ��d�d S | d dkrE��� r%J �| � dd�}| � d d
50 �}|r=�jd kr=� �|� |sG��� d S d S d S ) Nr7 zhttp.response.start�statusr: Tzhttp.response.bodyr r r@ F�HEAD)�get�is_setr9 �append�set)rH r r@ )�
51 body_partsr. rF �response_headers�response_started� status_coder r �send� s �
52 �z0ASGITransport.handle_async_request.<locals>.sendi� )r: �stream)r
53 r? )rH rI r
54 r )�
55 isinstancerU r r9 r: �raw�urlr; r< r= �split�query�host�portr, r* r r r' � Exceptionr) rO rM r r )r r. �scoperG rT rU r )rP r. rD rE rF rQ rR rS r �handle_async_requestc sR ��
56 ��
57 z"ASGITransport.handle_async_requestN)Tr$ r% )
58 r' r( r) r r* r+ r, r- r
59 r )r. r r
60 r )r! r" r# �__doc__r r_ r r r r r ? s � )r
61 r )r
62 r ) �
63 __future__r �typingZ_modelsr r �_typesr �baser �
64 TYPE_CHECKINGr r �Unionr �MutableMappingr+ �AnyZ_Message�Callable� AwaitableZ_ReceiveZ_Sendr( �__all__r r r r r r r r �<module> s, � �
65
66