File: //usr/local/lib/python3.10/dist-packages/httpx/_transports/__pycache__/asgi.cpython-310.pyc
o
���g} � @ s d dl mZ d dlZddlmZmZ ddlmZ ddlm Z ej
r2d dl Z d dl Z ej
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�
ASGITransport�return�boolc C s6 zdd l } | �� dkrW dS W dS ty Y dS w )Nr �trioTF)�sniffio�current_async_library� ImportError)r
� r �A/usr/local/lib/python3.10/dist-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
r c @ s e Zd Zd dd�Zd dd �Zd
S )
�ASGIResponseStream�body� list[bytes]r
�Nonec C s
|| _ d S �N)�_body)�selfr r r r �__init__8 s
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
r )r
r )�__name__�
__module__� __qualname__r r! r r r r r 7 s
r c @ s, e Zd ZdZ dddd�Zddd�ZdS )r a�
A custom AsyncTransport that handles sending requests directly to an ASGI app.
```python
transport = httpx.ASGITransport(
app=app,
root_path="/submount",
client=("1.2.3.4", 123)
)
client = httpx.AsyncClient(transport=transport)
```
Arguments:
* `app` - The ASGI application.
* `raise_app_exceptions` - Boolean indicating if exceptions in the application
should be raised. Default to `True`. Can be set to `False` for use cases
such as testing the content of a client 500 response.
* `root_path` - The root path on which the ASGI application should be mounted.
* `client` - A two-tuple indicating the client IP and port of incoming requests.
```
T� �z 127.0.0.1�{ �app�_ASGIApp�raise_app_exceptionsr � root_path�str�client�tuple[str, int]r
r c C s || _ || _|| _|| _d S r )r( r* r+ r- )r r( r* r+ r- r r r r W s
zASGITransport.__init__�requestr r c
� sJ �t �jt�s J �dddid�jdd� �jjD ��jj�jj�jj �
d�d �jj �jj �jj
f| j| jd � }�j�� �d
�d �d �g � d
�t� �d���fd
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 | ]
\}}|� � |f�qS r )�lower)�.0�k�vr r r �
<listcomp>o s z6ASGITransport.handle_async_request.<locals>.<listcomp>� ?r ) �type�asgi� http_version�method�headers�scheme�path�raw_path� query_string�serverr- r+ Fr
�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 )Nr8 zhttp.disconnectTz http.requestr F)r8 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
�}|r=�jd kr=� �|� |sG��� d S d S d S ) Nr8 zhttp.response.start�statusr< Tzhttp.response.bodyr r rC F�HEAD)�get�is_setr; �append�set)rK r rC )�
body_partsr/ rI �response_headers�response_started� status_coder r �send� s �
�z0ASGITransport.handle_async_request.<locals>.sendi� )r< �stream)r
rB )rK rL r
r )�
isinstancerX r r; r< �raw�urlr= r> r? �split�query�host�portr- r+ r! r r( � Exceptionr* rR rP r r )r r/ �scoperJ rW rX r )rS r/ rG rH rI rT rU rV r �handle_async_requestc sR ��
��
z"ASGITransport.handle_async_requestN)Tr% r& )
r( r) r* r r+ r, r- r. r
r )r/ r r
r )r" r# r$ �__doc__r rb r r r r r ? s � )r
r )r
r ) �
__future__r �typing�_modelsr r �_typesr �baser �
TYPE_CHECKINGr r �Unionr �MutableMappingr, �Any�_Message�Callable� Awaitable�_Receive�_Sendr) �__all__r r r r r r r r �<module> s, � �