U
'7`4 @ s 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*erd dl m+Z+ G dd de,Z-G dd de.ej/Z0eeeddddZ1e0j2ded ee0ddddZ3G dd d Z4G d d d Z5G d d deZ6G d d! d!e6Z7G d"d# d#e7Z8G d$d% d%e8Z9G d&d' d'e6Z:G d(d) d)e:Z;G d*d+ d+e:Z<G d,d- d-e:Z=G d.d/ d/e7Z>er.d d0l m?Z?m@Z@ e@eA ZBe?eA ZCnd d0lDm?Z?m@Z@ e@ZBe?ZCG d1d2 d2e6ZEG d3d4 d4eEZFe5 ZGeGHe7eAeIeJf eGHe8e. eGHe9ejK eGHe;ejL eGHe<ejM eGHe=ejNejOf eGHe:ejP eGHeFe% eGjHeEe?e0jQd dS )5 N)ABCabstractmethod)chain) IO
TYPE_CHECKINGAny
ByteStringDictIterableOptionalTextTextIOTupleTypeUnion) CIMultiDict )hdrs)AbstractStreamWriter)PY_36content_disposition_headerguess_filenameparse_mimetypesentinel) StreamReader) JSONEncoder _CIMultiDict)
PAYLOAD_REGISTRY get_payload payload_typePayload BytesPayload
StringPayload
IOBasePayloadBytesIOPayloadBufferedReaderPayload
TextIOPayloadStringIOPayload JsonPayloadAsyncIterablePayloadi )Listc @ s e Zd ZdS ) LookupErrorN)__name__
__module__ __qualname__ r/ r/ B/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/payload.pyr+ > s r+ c @ s e Zd ZdZdZdZdS )Ordernormal try_firsttry_lastN)r, r- r. r2 r3 r4 r/ r/ r/ r0 r1 B s r1 r )dataargskwargsreturnc O s t j| f||S N)r get)r5 r6 r7 r/ r/ r0 r H s r orderfactorytyper< r8 c C s t j| ||d d S Nr; )r register)r> r? r< r/ r/ r0 register_payloadL s rB c @ s> e Zd ZejdeeddddZed ed ddd ZdS )
r r; N)r? r< r8 c C s || _ || _d S r9 )r? r< )selfr? r< r/ r/ r0 __init__S s zpayload_type.__init__r )r> r8 c C s t || j| jd |S r@ )rB r? r< )rC r> r/ r/ r0 __call__W s zpayload_type.__call__) r, r- r. r1 r2 r rD r rE r/ r/ r/ r0 r R s r c @ sZ e Zd ZdZddddZedeeeeddd d
Zej d e
d eedd d
dZ dS )PayloadRegistryzZPayload registry.
note: we need zope.interface for more efficient adapter search
Nr8 c C s g | _ g | _g | _d S r9 )_first_normal_lastrC r/ r/ r0 rD b s zPayloadRegistry.__init__)_CHAINr )r5 r6 rL r7 r8 c O sR t |tr|S || j| j| jD ]&\}}t ||r ||f|| S q t d S r9 )
isinstancer rH rI rJ r+ )rC r5 rL r6 r7 r> r? r/ r/ r0 r: g s
zPayloadRegistry.getr; r= c C sf |t jkr | j||f nF|t jkr8| j||f n*|t jkrT| j||f ntd|d S )NzUnsupported order ) r1 r3 rH appendr2 rI r4 rJ
ValueError)rC r> r? r< r/ r/ r0 rA r s
zPayloadRegistry.register) r, r- r. __doc__rD r r r: r1 r2 r rA r/ r/ r/ r0 rF \ s" rF c
@ s e Zd ZdZdZdeddfeeee e
e e f e e
e e f f ee ee ee eddddZeee dddZeee dd d
Zee dd d Zeedd
dZeee dddZee dddZde eeddddZeeddddZ dS )r application/octet-streamN)valueheaders content_typefilenameencodingr7 r8 c K s || _ || _t | _|| _|tk r8|d k r8|| jtj< nD| jd k rnt | jd }|d kr`| j
}|| jtj< n| j
| jtj< | j |pi d S Nr ) _encoding _filenamer _headers_valuer r CONTENT_TYPE mimetypesZ
guess_type_default_content_typeupdate)rC rR rS rT rU rV r7 r/ r/ r0 rD s
zPayload.__init__rG c C s | j S )zSize of the payload.)_sizerK r/ r/ r0 size s z Payload.sizec C s | j S )zFilename of the payload.)rY rK r/ r/ r0 rU s zPayload.filenamec C s | j S )zCustom item headers)rZ rK r/ r/ r0 rS s zPayload.headersc C s$ d dd | j D dd S )N c S s g | ]\}}|d | d qS )z: z
r/ ).0kvr/ r/ r0
<listcomp> s z+Payload._binary_headers.<locals>.<listcomp>utf-8s
)joinrS itemsencoderK r/ r/ r0 _binary_headers s zPayload._binary_headersc C s | j S )zPayload encoding)rX rK r/ r/ r0 rV s zPayload.encodingc C s | j tj S )z Content type)rZ r r\ rK r/ r/ r0 rT s zPayload.content_typeT)disptype quote_fieldsparamsr8 c K s t |fd|i|| jtj< dS )z$Sets ``Content-Disposition`` header.rm N)r rZ r CONTENT_DISPOSITION)rC rl rm rn r/ r/ r0 set_content_disposition s z Payload.set_content_dispositionwriterr8 c s dS )zLWrite payload.
writer is an AbstractStreamWriter instance:
Nr/ rC rr r/ r/ r0 write s z
Payload.write)T) r, r- r. r^ r` r r r r r r strr
r rD propertyintra rU rS bytesrk rV rT boolrp r r rt r/ r/ r/ r0 r sL c s8 e Zd Zeeedd fdd ZeddddZ ZS )r! NrR r6 r7 r8 c s t |tttfs"tdt|d|kr2d|d< t j|f|| t |trZ|j | _
n
t || _
| j
t krt
r|d| i}ni }tjdtf| d S )Nz)value argument must be byte-ish, not {!r}rT rQ sourcez}Sending a large body directly with raw bytes might lock the event loop. You should probably pass an io.BytesIO object instead)rM rx bytearray
memoryview TypeErrorformatr? superrD nbytesr` lenTOO_LARGE_BYTES_BODYr warningswarnResourceWarningrC rR r6 r7 __class__r/ r0 rD s(
zBytesPayload.__init__rq c s | | jI d H d S r9 )rt r[ rs r/ r/ r0 rt s zBytesPayload.write) r, r- r. r r rD r rt
__classcell__r/ r/ r r0 r! s r! c s< e Zd Zdddeeee ee edd fddZ ZS )r" NrV rT rR r6 rV rT r7 r8 c sn |d kr2|d krd}d}qFt |}|jdd}n|d krBd| }|}t j||f|||d| d S )Nrg text/plain; charset=utf-8charsettext/plain; charset=%sr )r
parametersr: r rD rj )rC rR rV rT r6 r7 Z
real_encodingmimetyper r/ r0 rD s& zStringPayload.__init__) r, r- r. r r r ru rD r r/ r/ r r0 r" s r" c s, e Zd Zee eedd fdd Z ZS )r' Nrz c s t j| f|| d S r9 )r rD readr r r/ r0 rD s zStringIOPayload.__init__)r, r- r. r ru r rD r r/ r/ r r0 r' s r' c s@ e Zd Zd ee eeedd fdd
ZeddddZ Z S )
r#
attachmentN)rR dispositionr6 r7 r8 c sZ d|krt ||d< t j|f|| | jd k rV|d k rVtj| jkrV| j|| jd d S )NrU )rU )r r rD rY r ro rS rp )rC rR r r6 r7 r r/ r0 rD s zIOBasePayload.__init__rq c
sp t }zJ|d | jjdI d H }|rP||I d H |d | jjdI d H }q"W 5 |d | jjI d H X d S Ni )asyncioget_event_looprun_in_executorr[ closer rt rC rr Zloopchunkr/ r/ r0 rt # s zIOBasePayload.write)r )
r, r- r. r r ru rD r rt r r/ r/ r r0 r# s r# c sb e Zd Zdddeeee ee edd fddZeee dddZ
e dd d
d Z Z
S ) r& Nr r c sd |d kr2|d krd}d}qBt |}|jdd}n|d krBd| }t j|f|||d| d S )Nrg r r r rT rV )r r r: r rD )rC rR rV rT r6 r7 r r r/ r0 rD / s$ zTextIOPayload.__init__rG c C s: z t | j j| j W S tk
r4 Y d S X d S r9 osfstatr[ filenost_sizetellOSErrorrK r/ r/ r0 ra K s zTextIOPayload.sizerq c
sx t }zR|d | jjdI d H }|rX||| jI d H |d | jjdI d H }q"W 5 |d | jjI d H X d S r ) r r r r[ r r rt rj rX r r/ r/ r0 rt R s zTextIOPayload.write)r, r- r. r
r r ru rD rv rw ra r rt r r/ r/ r r0 r& . s r& c @ s e Zd ZeedddZdS )r$ rG c C s. | j }| j dtj}| j | || S rW )r[ r seekr SEEK_END)rC Zpositionendr/ r/ r0 ra ^ s
zBytesIOPayload.sizeN)r, r- r. rv rw ra r/ r/ r/ r0 r$ ] s r$ c @ s" e Zd Zeee dddZdS )r% rG c C s: z t | j j| j W S tk
r4 Y d S X d S r9 r rK r/ r/ r0 ra g s zBufferedReaderPayload.sizeN)r, r- r. rv r rw ra r/ r/ r/ r0 r% f s r% c s8 e Zd Zddejfeeeeeedd fdd
Z Z S )r( rg zapplication/jsonN)rR rV rT dumpsr6 r7 r8 c s, t j|||f|||d| d S )Nr )r rD rj )rC rR rV rT r r6 r7 r r/ r0 rD r s
zJsonPayload.__init__)
r, r- r. jsonr r ru r rD r r/ r/ r r0 r( q s r( )
AsyncIterable
AsyncIteratorc s< e Zd ZdZeeedd fdd ZeddddZ Z S )r) Nrz c sN t |ts tdt|d|kr,d|d< t j|f|| | | _d S )NzOvalue argument must support collections.abc.AsyncIterablebe interface, got {!r}rT rQ ) rM r r~ r r? r rD __aiter___iterr r r/ r0 rD s
z AsyncIterablePayload.__init__rq c sL | j rHz&| j I d H }||I d H qW n tk
rF d | _ Y nX d S r9 )r __anext__rt StopAsyncIteration)rC rr r r/ r/ r0 rt s zAsyncIterablePayload.write)
r, r- r. r _AsyncIterabler rD r rt r r/ r/ r r0 r) s r) c s( e Zd Zeeedd fdd Z ZS )StreamReaderPayloadNrz c s t j| f|| d S r9 )r rD Ziter_anyr r r/ r0 rD s z StreamReaderPayload.__init__)r, r- r. r r rD r r/ r/ r r0 r s r )Rr enumior r] r r abcr r itertoolsr typingr r r r r r
r r r
r r r Z multidictr rb r r Zhelpersr r r r r Zstreamsr Ztypedefsr r __all__r r* Exceptionr+ ru Enumr1 r r2 rB r rF r r! r" r' r# r& r$ r% r( r r rx Z_AsyncIteratorr Zcollections.abcr) r r rA r| r} StringIO
TextIOBaseBytesIOBufferedReaderBufferedRandomIOBaser4 r/ r/ r/ r0 <module> sv 8
#Q" /