File: //home/arjun/projects/env/lib/python3.10/site-packages/aiohttp/__pycache__/payload.cpython-310.pyc
o
+we5 � @ s$ U d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z d dl
m Z d dl m
Z
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�
ByteString�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__� r0 r0 �H/home/arjun/projects/env/lib/python3.10/site-packages/aiohttp/payload.pyr, = s r, c @ s e Zd ZdZdZdZdS )�Order�normal� try_first�try_lastN)r- r. r/ r3 r4 r5 r0 r0 r0 r1 r2 A s r2 �data�args�kwargs�returnr c O s t j| g|�R i |��S �N)r �get)r6 r7 r8 r0 r0 r1 r G s r ��order�factory�typer= c C s t j| ||d� d S �Nr< )r �register)r> r? r= r0 r0 r1 �register_payloadK s rB c @ sD e Zd Zejd�dededdfdd�Zded ded fd
d �ZdS ) r r< r? r= r9 Nc C s || _ || _d S r: )r? r= )�selfr? r= r0 r0 r1 �__init__R s
zpayload_type.__init__r> r c C s t || j| jd� |S r@ )rB r? r= )rC r> r0 r0 r1 �__call__V s zpayload_type.__call__) r- r. r/ r2 r3 r rD r rE r0 r0 r0 r1 r Q s r c @ s` e Zd ZdZddd�Zed�deded d
d edd f
d
d�Zej d�de
dededdfdd�Z dS )�PayloadRegistryzZPayload registry.
note: we need zope.interface for more efficient adapter search
r9 Nc C s g | _ g | _g | _d S r: )�_first�_normal�_last�rC r0 r0 r1 rD e s
zPayloadRegistry.__init__)�_CHAINr6 r7 rK z!Type[chain[_PayloadRegistryItem]]r8 r c O sT t |t�r|S || j| j| j�D ]\}}t ||�r&||g|�R i |�� S qt� �r: )�
isinstancer rG rH rI r, )rC r6 rK r7 r8 r> r? r0 r0 r1 r; j s
�zPayloadRegistry.getr< r> r? r= c C sh |t ju r| j�||f� d S |t ju r | j�||f� d S |t ju r-| j�||f� d S td|����)NzUnsupported order ) r2 r4 rG �appendr3 rH r5 rI �
ValueError)rC r> r? r= r0 r0 r1 rA y s
zPayloadRegistry.register)r9 N) r- r. r/ �__doc__rD r r r; r2 r3 � PayloadTyperA r0 r0 r0 r1 rF _ s2
�����
������rF 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"e d dfd#d$��Z dS )&r �application/octet-stream�_default_content_typeN�_size�value�headers� content_type�filename�encodingr8 r9 c K s� || _ || _t� | _|| _|tur |d ur || jtj< n"| jd ur7t� | j�d }|d u r0| j
}|| jtj< n| j
| jtj< | j� |pDi � d S �Nr ) � _encoding� _filenamer �_headers�_valuer r � CONTENT_TYPE� mimetypes�
guess_typerR �update)rC rT rU rV rW rX r8 r0 r0 r1 rD � s
zPayload.__init__c C � | j S )zSize of the payload.)rS rJ r0 r0 r1 �size� � z Payload.sizec C rb )zFilename of the payload.)r[ rJ r0 r0 r1 rW � rd zPayload.filenamec C rb )zCustom item headers)r\ rJ r0 r0 r1 rU � rd zPayload.headersc C s$ d� dd� | j�� D ���d�d S )N� c S s g | ] \}}|d | d �qS )z: z
r0 )�.0�k�vr0 r0 r1 �
<listcomp>� s z+Payload._binary_headers.<locals>.<listcomp>�utf-8s
)�joinrU �items�encoderJ r0 r0 r1 �_binary_headers� s ���zPayload._binary_headersc C rb )zPayload encoding)rZ rJ r0 r0 r1 rX � rd zPayload.encodingc C s | j tj S )z Content type)r\ r r^ rJ r0 r0 r1 rV � s zPayload.content_typeTrj �disptype� quote_fields�_charset�paramsc K s" t |f||d�|��| jtj< dS )z$Sets ``Content-Disposition`` header.)rp rq N)r r\ r �CONTENT_DISPOSITION)rC ro rp rq rr r0 r0 r1 �set_content_disposition� s ���z Payload.set_content_disposition�writerc � s �dS )zLWrite payload.
writer is an AbstractStreamWriter instance:
Nr0 �rC ru r0 r0 r1 �write� s z
Payload.write)Trj ) r- r. r/ rR �str�__annotations__rS r �intr r r r r r r r rD �propertyrc rW rU �bytesrn rX rV �boolrt r r rw r0 r0 r0 r1 r � sf
�� �� ��� �
������
� c s@ e Zd Zdedededdf� fdd� Zdeddfd d
�Z� ZS ) r! rT r7 r8 r9 Nc s� t |tttf�stdt|�����d|vrd|d< t� j|g|�R i |�� t |t�r0|j| _ nt
|�| _ | j t krJd| i}t j
dtfi |�� d S d S )Nz%value argument must be byte-ish, not rV rQ �sourcez}Sending a large body directly with raw bytes might lock the event loop. You should probably pass an io.BytesIO object instead)rL r| � bytearray�
memoryview� TypeErrorr? �superrD �nbytesrS �lenr* �warnings�warn�ResourceWarning�rC rT r7 r8 �� __class__r0 r1 rD � s"
�
��zBytesPayload.__init__ru c � s �|� | j�I d H d S r: )rw r] rv r0 r0 r1 rw � s �zBytesPayload.write) r- r. r/ r r rD r rw �
__classcell__r0 r0 r� r1 r! � s r! c
sF e Zd Zddd�dededee dee deddf � fd d
�Z� ZS ) r" N�rX rV rT r7 rX rV r8 r9 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 )Nrj �text/plain; charset=utf-8�charset�text/plain; charset=%sr� )r �
parametersr; r� rD rm )rC rT rX rV r7 r8 �
real_encoding�mimetyper� r0 r1 rD � s& ���
�zStringPayload.__init__)r- r. r/ rx r r rD r� r0 r0 r� r1 r" � s �������r" c s2 e Zd Zdee dededdf� fdd� Z� ZS )r' rT r7 r8 r9 Nc �" t � j|�� g|�R i |�� d S r: )r� rD �readr� r� r0 r1 rD � "zStringIOPayload.__init__)r- r. r/ r rx r rD r� r0 r0 r� r1 r' s *r' c sZ e Zd ZU ee ed< ddee dedededdf
� fd d
�
Zd eddfd d
�Z � Z
S )r# r] �
attachmentrT � dispositionr7 r8 r9 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 )NrW )rW )r r� rD r[ r rs rU rt )rC rT r� r7 r8 r� r0 r1 rD s �zIOBasePayload.__init__ru 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_executorr] r� rw �close)rC ru �loop�chunkr0 r0 r1 rw + s ���2zIOBasePayload.write)r� ) r- r. r/ r r ry rx rD r rw r� r0 r0 r� r1 r# s
������ r# c
sx e Zd ZU eed< ddd�dededee dee ded df � fd
d �Ze d ee
fd d
��Z de d dfdd�Z
� ZS )r& r] Nr� rT r7 rX rV r8 r9 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 )Nrj r� r� r� �rV rX )r r� r; r� rD )rC rT rX rV r7 r8 r� r� r0 r1 rD 9 s$ ���
�zTextIOPayload.__init__c C �4 zt �| j�� �j| j�� W S ty Y d S w r: ��os�fstatr] �fileno�st_size�tell�OSErrorrJ r0 r0 r1 rc U s
�zTextIOPayload.sizeru 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� )rX ) r� r� r� r] r� rZ rm rw r� )rC ru r� r� r6 r0 r0 r1 rw \ s ���� �2 zTextIOPayload.write)r- r. r/ r
ry r r rx rD r{ rz rc r rw r� r0 r0 r� r1 r& 6 s(
������� r& c @ s e Zd Zedefdd��ZdS )r$ r9 c C s. | j �� }| j �dtj�}| j �|� || S rY )r] r� �seekr� �SEEK_END)rC �position�endr0 r0 r1 rc m s
zBytesIOPayload.sizeN)r- r. r/ r{ rz rc r0 r0 r0 r1 r$ l s r$ c @ s" e Zd Zedee fdd��ZdS )r% r9 c C r� r: r� rJ r0 r0 r1 rc v s
�zBufferedReaderPayload.sizeN)r- r. r/ r{ r rz rc r0 r0 r0 r1 r% u s r% c sD e Zd Zddejfdedededededed d
f� fd d �
Z� Z S )
r( rj zapplication/jsonrT rX rV �dumpsr7 r8 r9 Nc s. t � j||��|�g|�R ||d�|�� d S )Nr� )r� rD rm )rC rT rX rV r� r7 r8 r� r0 r1 rD � s
���
�zJsonPayload.__init__)
r- r. r/ �jsonr� r rx r rD r� r0 r0 r� r1 r( � s& ��������r( )�
AsyncIterable�
AsyncIteratorc sR e Zd ZU dZee ed< dedededdf� fdd� Z d e
ddfd
d �Z � Z S ) r) N�_iterrT r7 r8 r9 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}rV rQ ) rL r� r� �formatr? r� rD � __aiter__r� r� r� r0 r1 rD � s
�z AsyncIterablePayload.__init__ru c � sL �| j r$z | j �� I d H }|�|�I d H q ty# d | _ Y d S w d S r: )r� � __anext__rw �StopAsyncIteration)rC ru r� r0 r0 r1 rw � s �� ��zAsyncIterablePayload.write)
r- r. r/ r� r �_AsyncIteratorry �_AsyncIterabler rD r rw r� r0 r0 r� r1 r) � s
r) c s. e Zd Zdedededdf� fdd� Z� ZS )�StreamReaderPayloadrT r7 r8 r9 Nc r� r: )r� rD �iter_anyr� r� r0 r1 rD � r� z StreamReaderPayload.__init__)r- r. r/ r r rD r� r0 r0 r� r1 r� � s &r� )Vr� �enum�ior� r_ r� r� �abcr r � itertoolsr �typingr r r r r r
r r r
r r r � multidictr re r r �helpersr r r r r �streamsr �typedefsr r �__all__r* rz ry r+ � Exceptionr, rx �Enumr2 r r3 rB r rP �_PayloadRegistryItemrF r r! r" r' r# r& r$ r% r( r� r� r| r� r� �collections.abcr) r� r rA r r� �StringIO�
TextIOBase�BytesIO�BufferedReader�BufferedRandom�IOBaser5 r0 r0 r0 r1 �<module> s�
8 ����
�
'U 6