Edit File: payload.cpython-37.pyc
B �k�a�5 � @ s U d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z d dl m Z d dl m Z mZmZmZmZmZmZmZmZmZmZ d dlmZ ddlmZ ddlm Z dd l m Z m Z m Z m!Z!m"Z" dd l#m$Z$ dd l%m&Z&m'Z'm(Z( d Z)d Z*e&e+ e,d< e�rd dl m-Z- G dd� de.�Z/G dd� de0ej1�Z2eeedd�dd�Z3e2j4d�ed ee2dd�dd�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 d+d,� d,e>�Z@G d-d.� d.e>�ZAG d/d0� d0e;�ZBe�rPd d1l mCZCmDZD eDeE ZFeCeE ZGnd d1lHmCZCmDZD eDZFeCZGG d2d3� d3e:�ZIG d4d5� d5eI�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 )6� N)�ABC�abstractmethod)�chain) �IO� TYPE_CHECKING�Any� ByteString�Dict�Iterable�Optional�TextIO�Tuple�Type�Union)� CIMultiDict� )�hdrs)�AbstractStreamWriter)�PY_36�content_disposition_header�guess_filename�parse_mimetype�sentinel)� StreamReader)�Final� 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 �B/opt/alt/python37/lib64/python3.7/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 @ s r2 r )�data�args�kwargs�returnc O s t j| f|�|�S )N)r �get)r6 r7 r8 r0 r0 r1 r F s r )�order)�factory�typer; r9 c C s t j| ||d� d S )N)r; )r �register)r< r= r; r0 r0 r1 �register_payloadJ s r? c @ s> e Zd Zejd�eedd�dd�Zed ed d�dd �ZdS ) r )r; N)r= r; r9 c C s || _ || _d S )N)r= r; )�selfr= r; r0 r0 r1 �__init__Q s zpayload_type.__init__r )r< r9 c C s t || j| jd� |S )N)r; )r? r= r; )r@ r< r0 r0 r1 �__call__U s zpayload_type.__call__) r- r. r/ r2 r3 r rA r rB r0 r0 r0 r1 r P s r c @ sV e Zd ZdZdd�dd�Zed�eededd �d d �Zej d �e eedd �dd�Z dS )�PayloadRegistryzZPayload registry. note: we need zope.interface for more efficient adapter search N)r9 c C s g | _ g | _g | _d S )N)�_first�_normal�_last)r@ r0 r0 r1 rA d s zPayloadRegistry.__init__)�_CHAINz!Type[chain[_PayloadRegistryItem]]r )r6 r7 rG r8 r9 c O sR t |t�r|S x8|| j| j| j�D ]"\}}t ||�r"||f|�|�S q"W t� �d S )N)� isinstancer rD rE rF r, )r@ r6 rG r7 r8 r< r= r0 r0 r1 r: i s zPayloadRegistry.get)r; )r< r= r; r9 c C sf |t jkr | j�||f� nF|t jkr8| j�||f� n*|t jkrT| j�||f� ntd|����d S )NzUnsupported order ) r2 r4 rD �appendr3 rE r5 rF � ValueError)r@ r< r= r; r0 r0 r1 r> x s zPayloadRegistry.register) r- r. r/ �__doc__rA r r r: r2 r3 � PayloadTyper> r0 r0 r0 r1 rC ^ s rC c @ s e Zd ZdZdZdeddfeeee e e e f e e e e f f ee ee ee edd�dd�Zeee d�dd��Zeee d�d d ��Zee d�d d ��Zeed�d d��Zeee d�dd��Zee d�dd��Zde ee edd�dd�Zeedd�dd��Z dS ) r zapplication/octet-streamN)�value�headers� content_type�filename�encodingr8 r9 c K s� || _ || _t� | _|| _|tk r8|d k r8|| jtj< nD| jd k rnt� | j�d }|d kr`| j }|| jtj< n| j | jtj< | j� |p�i � d S )Nr ) � _encoding� _filenamer �_headers�_valuer r � CONTENT_TYPE� mimetypesZ guess_type�_default_content_type�update)r@ rM rN rO rP rQ r8 r0 r0 r1 rA � s zPayload.__init__)r9 c C s | j S )zSize of the payload.)�_size)r@ r0 r0 r1 �size� s z Payload.sizec C s | j S )zFilename of the payload.)rS )r@ r0 r0 r1 rP � s zPayload.filenamec C s | j S )zCustom item headers)rT )r@ r0 r0 r1 rN � s 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>zutf-8s )�joinrN �items�encode)r@ r0 r0 r1 �_binary_headers� s zPayload._binary_headersc C s | j S )zPayload encoding)rR )r@ r0 r0 r1 rQ � s zPayload.encodingc C s | j tj S )z Content type)rT r rV )r@ r0 r0 r1 rO � s zPayload.content_typeT�utf-8)�disptype� quote_fields�_charset�paramsr9 c K s" t |f||d�|��| jtj< dS )z$Sets ``Content-Disposition`` header.)rg rh N)r rT r �CONTENT_DISPOSITION)r@ rf rg rh ri r0 r0 r1 �set_content_disposition� s z Payload.set_content_disposition)�writerr9 c � s dS )zLWrite payload. writer is an AbstractStreamWriter instance: Nr0 )r@ rl r0 r0 r1 �write� s z Payload.write)Tre ) r- r. r/ rX rZ r r r r r r �strr r rA �property�intr[ rP rN �bytesrd rQ rO �boolrk r r rm r0 r0 r0 r1 r � s4 < c s8 e Zd Zeeedd�� fdd� Zedd�dd�Z� ZS )r! N)rM r7 r8 r9 c s� t |tttf�s"tdt|�����d|kr2d|d< t� j|f|�|� t |t�rZ|j| _ n t |�| _ | j t kr�t r|d| i}ni }t jdtf|� d S )Nz%value argument must be byte-ish, not rO zapplication/octet-stream�sourcez}Sending a large body directly with raw bytes might lock the event loop. You should probably pass an io.BytesIO object instead)rH rq � bytearray� memoryview� TypeErrorr= �superrA �nbytesrZ �lenr* r �warnings�warn�ResourceWarning)r@ rM r7 r8 )� __class__r0 r1 rA � s zBytesPayload.__init__)rl r9 c � s |� | j�I d H d S )N)rm rU )r@ rl r0 r0 r1 rm � s zBytesPayload.write) r- r. r/ r r rA r rm � __classcell__r0 r0 )r} r1 r! � s r! c s< e Zd Zddd�eeee ee edd�� fdd�Z� ZS )r" N)rQ rO )rM r7 rQ rO r8 r9 c sn |d kr2|d krd}d}qFt |�}|j�dd�}n|d krBd| }|}t� j|�|�f|�||d�|�� d S )Nzutf-8ztext/plain; charset=utf-8�charsetztext/plain; charset=%s)rQ rO )r � parametersr: rw rA rc )r@ rM rQ rO r7 r8 Z real_encoding�mimetype)r} r0 r1 rA � s zStringPayload.__init__)r- r. r/ rn r r rA r~ r0 r0 )r} r1 r" � s r" c s, e Zd Zee eedd�� fdd� Z� ZS )r' N)rM r7 r8 r9 c s t � j|�� f|�|� d S )N)rw rA �read)r@ rM r7 r8 )r} r0 r1 rA s zStringIOPayload.__init__)r- r. r/ r rn r rA r~ r0 r0 )r} r1 r' s r' c sN e Zd ZU ee ed< d ee eeedd�� fdd� Zedd�dd �Z � Z S ) r# rU � attachmentN)rM � dispositionr7 r8 r9 c sZ d|krt |�|d< t� j|f|�|� | jd k rV|d k rVtj| jkrV| j|| jd� d S )NrP )rP )r rw rA rS r rj rN rk )r@ rM r� r7 r8 )r} r0 r1 rA ! s zIOBasePayload.__init__)rl r9 c � st t �� }zN|�d | jjd�I d H }x0|rR|�|�I d H |�d | jjd�I d H }q$W W d |�d | jj�I d H X d S )Ni )�asyncio�get_event_loop�run_in_executorrU r� rm �close)r@ rl �loop�chunkr0 r0 r1 rm - s zIOBasePayload.write)r� ) r- r. r/ r r �__annotations__rn rA r rm r~ r0 r0 )r} r1 r# s r# c sl e Zd ZU eed< ddd�eeee ee edd�� fdd�Ze ee d�dd ��Z e dd �d d �Z � ZS ) r& rU N)rQ rO )rM r7 rQ rO r8 r9 c sd |d kr2|d krd}d}qBt |�}|j�dd�}n|d krBd| }t� j|f|�||d�|�� d S )Nzutf-8ztext/plain; charset=utf-8r ztext/plain; charset=%s)rO rQ )r r� r: rw rA )r@ rM rQ rO r7 r8 r� )r} r0 r1 rA ; s zTextIOPayload.__init__)r9 c C s6 y t �| j�� �j| j�� S tk r0 d S X d S )N)�os�fstatrU �fileno�st_size�tell�OSError)r@ r0 r0 r1 r[ W s zTextIOPayload.size)rl r9 c � s� t �� }zj|�d | jjd�I d H }xL|rn| jr<|j| jd�n|�� }|�|�I d H |�d | jjd�I d H }q$W W d |�d | jj�I d H X d S )Ni )rQ ) r� r� r� rU r� rR rc rm r� )r@ rl r� r� r6 r0 r0 r1 rm ^ s zTextIOPayload.write)r- r. r/ r r� r r rn rA ro rp r[ r rm r~ r0 r0 )r} r1 r& 8 s r& c @ s e Zd Zeed�dd��ZdS )r$ )r9 c C s. | j �� }| j �dtj�}| j �|� || S )Nr )rU r� �seekr� �SEEK_END)r@ �position�endr0 r0 r1 r[ o s zBytesIOPayload.sizeN)r- r. r/ ro rp r[ r0 r0 r0 r1 r$ n s r$ c @ s" e Zd Zeee d�dd��ZdS )r% )r9 c C s6 y t �| j�� �j| j�� S tk r0 d S X d S )N)r� r� rU r� r� r� r� )r@ r0 r0 r1 r[ x s zBufferedReaderPayload.sizeN)r- r. r/ ro r rp r[ r0 r0 r0 r1 r% w s r% c s8 e Zd Zddejfeeeeeedd�� fdd� Z� Z S )r( zutf-8zapplication/jsonN)rM rQ rO �dumpsr7 r8 r9 c s, t � j||��|�f|�||d�|�� d S )N)rO rQ )rw rA rc )r@ rM rQ rO r� r7 r8 )r} r0 r1 rA � s zJsonPayload.__init__) r- r. r/ �jsonr� r rn r rA r~ r0 r0 )r} r1 r( � s r( )� AsyncIterable� AsyncIteratorc s< e Zd ZdZeeedd�� fdd� Zedd�dd�Z� Z S )r) N)rM r7 r8 r9 c sN t |t�s td�t|����d|kr,d|d< t� j|f|�|� |�� | _d S )NzOvalue argument must support collections.abc.AsyncIterablebe interface, got {!r}rO zapplication/octet-stream) rH r� rv �formatr= rw rA � __aiter__�_iter)r@ rM r7 r8 )r} r0 r1 rA � s z AsyncIterablePayload.__init__)rl r9 c � sP | j rLy*x$| j �� I d H }|�|�I d H q W W n tk rJ d | _ Y nX d S )N)r� � __anext__rm �StopAsyncIteration)r@ rl r� r0 r0 r1 rm � s zAsyncIterablePayload.write) r- r. r/ r� �_AsyncIterabler rA r rm r~ r0 r0 )r} r1 r) � s r) c s( e Zd Zeeedd�� fdd� Z� ZS )�StreamReaderPayloadN)rM r7 r8 r9 c s t � j|�� f|�|� d S )N)rw rA Ziter_any)r@ rM r7 r8 )r} r0 r1 rA � s z StreamReaderPayload.__init__)r- r. r/ r r rA r~ r0 r0 )r} r1 r� � s r� )Vr� �enum�ior� rW r� rz �abcr r � itertoolsr �typingr r r r r r r r r r r Z multidictr r\ r r Zhelpersr r r r r Zstreamsr Ztypedefsr r r �__all__r* rp r� r+ � Exceptionr, rn �Enumr2 r r3 r? r rL Z_PayloadRegistryItemrC r r! r" r' r# r& r$ r% r( r� r� rq Z_AsyncIteratorr� �collections.abcr) r� r r> rt ru �StringIO� TextIOBase�BytesIO�BufferedReader�BufferedRandom�IOBaser5 r0 r0 r0 r1 �<module> sr 4 'U 6