File: //usr/local/lib/python3.10/dist-packages/aiohttp/__pycache__/payload.cpython-310.pyc
o
���g�= � @ s( U d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z m
Z
d dl m Z d dl
mZmZmZmZmZmZmZmZmZmZmZ d dlmZ ddlm Z ddlm Z dd l m Z m Z m!Z!m"Z"m#Z# dd
l$m%Z% dd l&m'Z'm(Z( d Z)d
Z*ee+ e,d< er�d dl
m-Z- G dd� de.�Z/G dd� de0ej1�Z2dedededdfdd�Z3e2j4d�d ed d ed e2ddfd d �Z5G d!d"� d"�Z6ed Z7ee7ef Z8G d#d$� d$�Z9G d%d� de �Z:G d&d'� d'e:�Z;G d(d)� d)e;�Z<G d*d+� d+e<�Z=G d,d-� d-e:�Z>G d.d/� d/e>�Z?G d0d1� d1e>�Z@G d2d3� d3e>�ZAG d4d5� d5e;�ZBe�r/d d6l
mCZCmDZD eDeE ZFeCeE ZGn d d6lHmCZCmDZD eDZFeCZGG d7d8� d8e:�ZIG d9d:� d:eI�ZJe9� ZKeK�Le;eEeMeNf� eK�Le<e0� eK�Le=ejO� eK�Le?ejP� eK�Le@ejQ� eK�LeAejRejSf� eK�Le>ejT� eK�LeJe%� eKjLeIeCe2jUd� dS );� N)�ABC�abstractmethod)�chain) �IO�
TYPE_CHECKING�Any�Dict�Final�Iterable�Optional�TextIO�Tuple�Type�Union)� CIMultiDict� )�hdrs)�AbstractStreamWriter)� _SENTINEL�content_disposition_header�guess_filename�parse_mimetype�sentinel)� StreamReader)� JSONEncoder� _CIMultiDict)
�PAYLOAD_REGISTRY� get_payload� payload_type�Payload� BytesPayload�
StringPayload�
IOBasePayload�BytesIOPayload�BufferedReaderPayload�
TextIOPayload�StringIOPayload� JsonPayload�AsyncIterablePayloadi �TOO_LARGE_BYTES_BODY)�Listc @ s e Zd ZdS )� LookupErrorN)�__name__�
__module__� __qualname__� r/ r/ �:/usr/local/lib/python3.10/dist-packages/aiohttp/payload.pyr+ = s r+ c @ s e Zd ZdZdZdZdS )�Order�normal� try_first�try_lastN)r, r- r. r2 r3 r4 r/ r/ r/ r0 r1 A s r1 �data�args�kwargs�returnr c O s t j| g|�R i |��S �N)r �get)r5 r6 r7 r/ r/ r0 r G s r ��order�factory�typer< c C s t j| ||d� d S �Nr; )r �register)r= r> r< r/ r/ r0 �register_payloadK s rA c @ sD e Zd Zejd�dededdfdd�Zded ded fd
d �ZdS ) r r; r> r< r8 Nc C s || _ || _d S r9 )r> r< )�selfr> r< r/ r/ r0 �__init__R s
zpayload_type.__init__r= r c C s t || j| jd� |S r? )rA r> r< )rB r= r/ r/ r0 �__call__V s zpayload_type.__call__) r, r- r. r1 r2 r rC r rD r/ r/ r/ r0 r Q s r c @ sd e Zd ZdZdZddd�Zed�ded ed
d d edd
f
dd�Ze j
d�de dede ddfdd�Z dS )�PayloadRegistryzZPayload registry.
note: we need zope.interface for more efficient adapter search
��_first�_normal�_last�_normal_lookupr8 Nc C s g | _ g | _g | _i | _d S r9 rF �rB r/ r/ r0 rC g s
zPayloadRegistry.__init__)�_CHAINr5 r6 rL z!Type[chain[_PayloadRegistryItem]]r7 r c O s� | j r | j D ]\}}t||�r ||g|�R i |�� S q| j�t|�� }r2||g|�R i |��S t|t�r9|S || j| j�D ]\}}t||�rV||g|�R i |�� S q@t� �r9 ) rG �
isinstancerJ r: r> r rH rI r+ )rB r5 rL r6 r7 r= �type_�lookup_factoryr/ r/ r0 r: m s
�
�zPayloadRegistry.getr; r= r> r< c C s� |t ju r| j�||f� d S |t ju r4| j�||f� t|t�r-|D ]}|| j|< q#d S || j|< d S |t j u rC| j
�||f� d S t d|����)NzUnsupported order ) r1 r3 rG �appendr2 rH rM r
rJ r4 rI �
ValueError)rB r= r> r< �tr/ r/ r0 r@ � s
�
zPayloadRegistry.register)r8 N)
r, r- r. �__doc__� __slots__rC r r r: r1 r2 � PayloadTyper@ r/ r/ r/ r0 rE _ s4
�����
������rE c @ s\ e Zd ZU dZeed< dZee ed< de ddfde
dee e e
eef eeeef f de edef dee d ee d
e
d dfd d
�Zed ee fdd��Zed ee fdd��Zed e fdd��Zed efdd��Zed ee fdd��Zed efdd��Z d)d ed ed ed e
d df
d d!�Ze d*d ed#ed efd$d%��Z e d&e d dfd'd(��Z dS )+r �application/octet-stream�_default_content_typeN�_size�value�headers� content_type�filename�encodingr7 r8 c K s� || _ || _t� | _|| _|tur |d ur || jtj< n-| jd urBtj dkr*t
j }nt
j }|| j�d }|d u r;| j
}|| jtj< n| j
| jtj< |rS| j�|� d S d S )N)� �
r )� _encoding� _filenamer �_headers�_valuer r � CONTENT_TYPE�sys� version_info� mimetypes�guess_file_type�
guess_typerW �update)rB rY rZ r[ r\ r] r7 �guesserr/ r/ r0 rC � s$
�zPayload.__init__c C � | j S )zSize of the payload.)rX rK r/ r/ r0 �size� � z Payload.sizec C rl )zFilename of the payload.)ra rK r/ r/ r0 r\ � rn zPayload.filenamec C rl )zCustom item headers)rb rK r/ r/ r0 rZ � rn zPayload.headersc C s$ d� dd� | j�� D ���d�d S )N� c S s g | ] \}}|d | d �qS )z: z
r/ )�.0�k�vr/ r/ r0 �
<listcomp>� s z+Payload._binary_headers.<locals>.<listcomp>�utf-8s
)�joinrZ �items�encoderK r/ r/ r0 �_binary_headers� s ���zPayload._binary_headersc C rl )zPayload encoding)r` rK r/ r/ r0 r] � rn zPayload.encodingc C s | j tj S )z Content type)rb r rd rK r/ r/ r0 r[ � s zPayload.content_typeTrt �disptype� quote_fields�_charset�paramsc K s" t |f||d�|��| jtj< dS )z$Sets ``Content-Disposition`` header.)rz r{ N)r rb r �CONTENT_DISPOSITION)rB ry rz r{ r| r/ r/ r0 �set_content_disposition� s ���z Payload.set_content_disposition�strict�errorsc C s dS )z~Return string representation of the value.
This is named decode() to allow compatibility with bytes objects.
Nr/ �rB r] r� r/ r/ r0 �decode� s zPayload.decode�writerc � s �dS )zLWrite payload.
writer is an AbstractStreamWriter instance:
Nr/ �rB r� r/ r/ r0 �write� s z
Payload.write)Trt �rt r ) r, r- r. rW �str�__annotations__rX r �intr r r r r r
r
r rC �propertyrm r\ rZ �bytesrx r] r[ �boolr~ r r� r r� r/ r/ r/ r0 r � sj
�� �� ��� �
� �����
� c sl e Zd ZU eed< deeeef dededdf� fdd� Z dd e
d e
de
fd
d�Z de ddfdd�Z
� ZS )r rc rY r6 r7 r8 Nc s� d|vrd|d< t � j|g|�R i |�� t|t�r |j| _nt|ttf�r,t|�| _n t dt
|�����| jt krJd| i}t j
dtfi |�� d S d S )Nr[ rV z%value argument must be byte-ish, not �sourcez}Sending a large body directly with raw bytes might lock the event loop. You should probably pass an io.BytesIO object instead)�superrC rM �
memoryview�nbytesrX r� � bytearray�len� TypeErrorr> r) �warnings�warn�ResourceWarning�rB rY r6 r7 �� __class__r/ r0 rC � s"
�
��zBytesPayload.__init__rt r r] r� c C s | j �||�S r9 )rc r� r� r/ r/ r0 r� s zBytesPayload.decoder� c � s �|� | j�I d H d S r9 )r� rc r� r/ r/ r0 r� s �zBytesPayload.writer� )r, r- r. r� r� r r� r� r rC r� r� r r� �
__classcell__r/ r/ r� r0 r � s
����r c
sF e Zd Zddd�dededee dee deddf � fd d
�Z� ZS ) r! N�r] r[ rY r6 r] r[ r7 r8 c sp |d u r|d u r
d}d}nt |�}|j�dd�}n
|d u r!d| }|}t� j|�|�g|�R ||d�|�� d S )Nrt �text/plain; charset=utf-8�charset�text/plain; charset=%sr� )r �
parametersr: r� rC rw )rB rY r] r[ r6 r7 �
real_encoding�mimetyper� r/ r0 rC s& ���
�zStringPayload.__init__)r, r- r. r� r r rC r� r/ r/ r� r0 r! s �������r! c s2 e Zd Zdee dededdf� fdd� Z� ZS )r& rY r6 r7 r8 Nc �" t � j|�� g|�R i |�� d S r9 )r� rC �readr� r� r/ r0 rC : � "zStringIOPayload.__init__)r, r- r. r r� r rC r� r/ r/ r� r0 r& 9 s *r& c sp e Zd ZU ejed< ddee dedededdf
� fd d
�
Z d e
ddfd d
�Z ddededefdd�Z � Z
S )r" rc �
attachmentrY � dispositionr6 r7 r8 Nc sl d|vr
t |�|d< t� j|g|�R i |�� | jd ur0|d ur2tj| jvr4| j|| jd� d S d S d S d S )Nr\ )r\ )r r� rC ra r r} rZ r~ )rB rY r� r6 r7 r� r/ r0 rC A s �zIOBasePayload.__init__r� c
� s� �t �� }z@|�d | jjd�I d H }|r8|�|�I d H |�d | jjd�I d H }|sW |�d | jj�I d H d S W |�d | jj�I d H d S |�d | jj�I d H w )N� )�asyncio�get_event_loop�run_in_executorrc r� r� �close)rB r� �loop�chunkr/ r/ r0 r� M s ���2zIOBasePayload.writert r r] r� c s d� � �fdd�| j�� D ��S )Nro c 3 s � | ] }|� � ��V qd S r9 )r� )rp �r�r] r� r/ r0 � <genexpr>X s � z'IOBasePayload.decode.<locals>.<genexpr>)ru rc � readlinesr� r/ r� r0 r� W s zIOBasePayload.decode)r� r� )r, r- r. �io�IOBaser� r r r� rC r r� r� r� r/ r/ r� r0 r" > s
������
r" c
s� e Zd ZU ejed< ddd�dededee dee ded df � fd
d �Z
e d ee fd d
��Z
dde de d e fdd�Zded dfdd�Z� ZS )r% rc Nr� rY r6 r] r[ r7 r8 c sf |d u r|d u r
d}d}nt |�}|j�dd�}n|d u r!d| }t� j|g|�R ||d�|�� d S )Nrt r� r� r� �r[ r] )r r� r: r� rC )rB rY r] r[ r6 r7 r� r� r/ r0 rC ^ s$ ���
�zTextIOPayload.__init__c C s4 zt �| j�� �j| j�� W S ty Y d S w r9 )�os�fstatrc �fileno�st_size�tell�OSErrorrK r/ r/ r0 rm z s
�zTextIOPayload.sizert r r� c C s
| j �� S r9 )rc r� r� r/ r/ r0 r� � s
zTextIOPayload.decoder� c
� s� �t �� }zN|�d | jjd�I d H }|rF| jr |j| jd�n|�� }|�|�I d H |�d | jjd�I d H }|sW |�d | jj�I d H d S W |�d | jj�I d H d S |�d | jj�I d H w )Nr� )r] ) r� r� r� rc r� r` rw r� r� )rB r� r� r� r5 r/ r/ r0 r� � s ���� �2 zTextIOPayload.writer� )r, r- r. r� �
TextIOBaser� r r r r� rC r� r� rm r� r r� r� r/ r/ r� r0 r% [ s*
������� r% c @ sB e Zd ZU ejed< edefdd��Zd de de de fd d
�Z
d S )
r# rc r8 c C s. | j �� }| j �dtj�}| j �|� || S )Nr )rc r� �seekr� �SEEK_END)rB �position�endr/ r/ r0 rm � s
zBytesIOPayload.sizert r r] r� c C � | j �� �||�S r9 �rc r� r� r� r/ r/ r0 r� � � zBytesIOPayload.decodeNr� ) r, r- r. r� �BytesIOr� r� r� rm r� r� r/ r/ r/ r0 r# � s
r# c @ sF e Zd ZU ejed< edee fdd��Z d de
de
de
fd d
�Z d S )
r$ rc r8 c C s8 zt �| j�� �j| j�� W S ttfy Y d S w r9 )r� r� rc r� r� r� r� �AttributeErrorrK r/ r/ r0 rm � s
�zBufferedReaderPayload.sizert r r] r� c C r� r9 r� r� r/ r/ r0 r� � r� z BufferedReaderPayload.decodeNr� ) r, r- r. r� �BufferedIOBaser� r� r r� rm r� r� r/ r/ r/ r0 r$ � s
r$ c sD e Zd Zddejfdedededededed d
f� fd d �
Z� Z S )
r' rt zapplication/jsonrY r] r[ �dumpsr6 r7 r8 Nc s. t � j||��|�g|�R ||d�|�� d S )Nr� )r� rC rw )rB rY r] r[ r� r6 r7 r� r/ r0 rC � s
���
�zJsonPayload.__init__)
r, r- r. �jsonr� r r� r rC r� r/ r/ r� r0 r' � s& ��������r' )�
AsyncIterable�
AsyncIteratorc sr e Zd ZU dZee ed< eed< dedededdf� fdd � Z d
e
ddfd d �Z dde de de fdd�Z
� ZS )r( N�_iterrc rY r6 r7 r8 c sT t |t�std�t|����d|vrd|d< t� j|g|�R i |�� |�� | _d S )NzMvalue argument must support collections.abc.AsyncIterable interface, got {!r}r[ rV ) rM r� r� �formatr> r� rC � __aiter__r� r� r� r/ r0 rC � s
�z AsyncIterablePayload.__init__r� c � sL �| j r$z | j �� I d H }|�|�I d H q ty# d | _ Y d S w d S r9 )r� � __anext__r� �StopAsyncIteration)rB r� r� r/ r/ r0 r� � s �� ��zAsyncIterablePayload.writert r r] r� c C s t d��)NzUnable to decode.)r� r� r/ r/ r0 r� � s zAsyncIterablePayload.decoder� )r, r- r. r� r �_AsyncIteratorr� �_AsyncIterabler rC r r� r� r� r� r/ r/ r� r0 r( � s
r( c s. e Zd Zdedededdf� fdd� Z� ZS )�StreamReaderPayloadrY r6 r7 r8 Nc r� r9 )r� rC �iter_anyr� r� r/ r0 rC � r� z StreamReaderPayload.__init__)r, r- r. r r rC r� r/ r/ r� r0 r� � s &r� )Vr� �enumr� r� rg r� re r� �abcr r � itertoolsr �typingr r r r r r
r r r
r r � multidictr ro r r �helpersr r r r r �streamsr �typedefsr r �__all__r) r� r� r* � Exceptionr+ r� �Enumr1 r r2 rA r rU �_PayloadRegistryItemrE r r r! r& r" r% r# r$ r' r� r� r� r� r� �collections.abcr( r� r r@ r� r� �StringIOr� r� �BufferedReader�BufferedRandomr� r4 r/ r/ r/ r0 �<module> s�
4 ����
�
7a# 9
#