Edit File: web_response.cpython-37.pyc
B �k�a�k � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl m Z d dl m Z mZ d dlmZmZmZmZmZmZmZmZmZmZ d dlmZm Z ddl m Z m Z ddl m!Z! dd l"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ dd l,m-Z-m.Z.m/Z/m0Z0 dd l m1Z1 dd l2m3Z3m4Z4 d Z5e�rDddl6m7Z7 ee8ef Z9nej jZ9e$�s\de j:d< G dd� dej;�Z<G dd� de9e'�Z=G dd� de=�Z>e*fddddddej?d�eee8 ee@ eAee8 ee4 e8e3e>d� dd �ZBdS ) � N)�Executor)�Morsel� SimpleCookie) � TYPE_CHECKING�Any�Dict�Iterator�Mapping�MutableMapping�Optional�Tuple�Union�cast)� CIMultiDict�istr� )�hdrs�payload)�AbstractStreamWriter) �ETAG_ANY�PY_38�QUOTED_ETAG_RE�ETag� HeadersMixin�parse_http_date�rfc822_formatted_time�sentinel�validate_etag_value)� RESPONSES�SERVER_SOFTWARE� HttpVersion10� HttpVersion11)�Payload)� JSONEncoder� LooseHeaders)� ContentCoding�StreamResponse�Response� json_response)� BaseRequestZSameSite�samesitec @ s e Zd ZdZdZdZdS )r% �deflate�gzip�identityN)�__name__� __module__� __qualname__r+ r, r- � r1 r1 �G/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/web_response.pyr% = s r% c s e Zd ZdZdddd�eee ee dd�dd�Ze e d�d d ��Z e d d�d d ��Z e ed�dd��Z e e d�dd��Ze e d�dd��Ze ed�dd��Zdefeee eeeeef f dd�dd�Ze ee d�dd��Zdd�dd �Ze ed�d d ��Ze ed�d d ��Zdsee dd!�d"d#�Zdteee ef dd$�d%d&�Z e d'd�d(d)��Z e d*d�d+d,��Z dddd-ddddd.�eeee ee eeeef eee ee ee ee dd/� d0d1�Z dd-d2�eee edd3�d4d5�Z e ee d�� fd6d7� �Z!e!j"ee dd8�d9d7��Z!e ed�� fd:d;� �Z#e#j"edd8�d<d;��Z#e ee d�� fd=d>� �Z$e$j"ee dd8�d?d>��Z$e ee%j% d�d@dA��Z&e&j"eeee'e%j%ef dd8�dBdA��Z&e ee( d�dCdD��Z)e)j"eee(ef dd8�dEdD��Z)e*j+fe,ddF�dGdH�Z-eddI�dJdK�Z.dLddM�dNdO�Z/dLee0 dM�dPdQ�Z1dLe0dM�dRdS�Z2dd�dTdU�Z3dd�dVdW�Z4e5ddX�dYdZ�Z6dd�d[d\�Z7due5ddX�d^d_�Z8ed�d`da�Z9ee:db�dcdd�Z;ee:dde�dfdg�Z<eddb�dhdi�Z=ed�djdk�Z>e?e d�dldm�Z@ed�dndo�ZAeBe dp�dqdr�ZC� ZDS )vr& T�� N)�status�reason�headers)r4 r5 r6 �returnc C sp d | _ d | _d| _d| _d | _t� | _d | _d | _d| _ d| _ i | _ |d k rXt |�| _ nt � | _ | �||� d S )NFr )�_body� _keep_alive�_chunked� _compression�_compression_forcer �_cookies�_req�_payload_writer� _eof_sent� _body_length�_stater �_headers� set_status)�selfr4 r5 r6 r1 r1 r2 �__init__P s zStreamResponse.__init__)r7 c C s | j d k S )N)r? )rE r1 r1 r2 �preparedk s zStreamResponse.preparedz Optional[asyncio.Task[None]]c C s | j r| j jS d S d S )N)r> �task)rE r1 r1 r2 rH o s zStreamResponse.taskc C s | j S )N)�_status)rE r1 r1 r2 r4 v s zStreamResponse.statusc C s | j S )N)r: )rE r1 r1 r2 �chunkedz s zStreamResponse.chunkedc C s | j S )N)r; )rE r1 r1 r2 � compression~ s zStreamResponse.compressionc C s | j S )N)�_reason)rE r1 r1 r2 r5 � s zStreamResponse.reason)r4 r5 � _RESPONSESr7 c C sV | j rtd��t|�| _|d krLy|| j d }W n tk rJ d}Y nX || _d S )NzGCannot change the response status code after the headers have been sentr � )rG �AssertionError�intrI � ExceptionrL )rE r4 r5 rM r1 r1 r2 rD � s zStreamResponse.set_statusc C s | j S )N)r9 )rE r1 r1 r2 � keep_alive� s zStreamResponse.keep_alivec C s d| _ d S )NF)r9 )rE r1 r1 r2 � force_close� s zStreamResponse.force_closec C s | j S )N)rA )rE r1 r1 r2 � body_length� s zStreamResponse.body_lengthc C s t �dt� | jst�| jjS )Nzoutput_length is deprecated)�warnings�warn�DeprecationWarningr? rO � buffer_size)rE r1 r1 r2 � output_length� s z StreamResponse.output_length)� chunk_sizer7 c C s2 d| _ tj| jkrtd��|dk r.t�dt� dS )z,Enables automatic chunked transfer encoding.Tz>You can't enable chunked encoding when a content length is setNz Chunk size is deprecated #1615)r: r �CONTENT_LENGTHrC � RuntimeErrorrU rV rW )rE rZ r1 r1 r2 �enable_chunked_encoding� s z&StreamResponse.enable_chunked_encoding)�forcer7 c C sT t |�tkr*|rtjntj}t�dt� n|dk rDt|t�sDt d��d| _ || _ dS )z&Enables response compression encoding.z+Using boolean for force is deprecated #3318Nz1force should one of None, bool or ContentEncodingT) �type�boolr% r+ r- rU rV rW � isinstancerO r; r< )rE r^ r1 r1 r2 �enable_compression� s z!StreamResponse.enable_compressionzCIMultiDict[str]c C s | j S )N)rC )rE r1 r1 r2 r6 � s zStreamResponse.headerszSimpleCookie[str]c C s | j S )N)r= )rE r1 r1 r2 �cookies� s zStreamResponse.cookies�/)�expires�domain�max_age�path�secure�httponly�versionr* ) �name�valuere rf rg rh ri rj rk r* r7 c C s� | j �|�} | dk r,| jdkr,| j �|d� || j |< | j | } |dk rR|| d< n| �d�dkrf| d= |dk rv|| d< |dk r�t|�| d< nd| kr�| d= || d< |dk r�|| d< |dk r�|| d < | dk r�| | d < | dk r�| | d < dS ) z�Set or update response cookie. Sets new cookie or updates existent with new value. Also updates only those params which are not None. NrN re z Thu, 01 Jan 1970 00:00:00 GMTrf zmax-agerh ri rj rk r* )r= �getZ coded_value�pop�str) rE rl rm re rf rg rh ri rj rk r* �old�cr1 r1 r2 � set_cookie� s0 zStreamResponse.set_cookie)rf rh )rl rf rh r7 c C s( | j �|d� | j|ddd||d� dS )zBDelete cookie. Creates new empty expired cookie. NrN r z Thu, 01 Jan 1970 00:00:00 GMT)rg re rf rh )r= ro rs )rE rl rf rh r1 r1 r2 � del_cookie s zStreamResponse.del_cookiec s t � jS )N)�super�content_length)rE )� __class__r1 r2 rv s z StreamResponse.content_length)rm r7 c C sD |d k r0t |�}| jr td��t|�| jtj< n| j�tjd � d S )Nz<You can't set content length when chunked encoding is enable)rP r: r\ rp rC r r[ ro )rE rm r1 r1 r2 rv s c s t � jS )N)ru � content_type)rE )rw r1 r2 rx # s zStreamResponse.content_typec C s | j t|�| _| �� d S )N)rx rp � _content_type� _generate_content_type_header)rE rm r1 r1 r2 rx ( s c s t � jS )N)ru �charset)rE )rw r1 r2 r{ . s zStreamResponse.charsetc C sZ | j }|dkrtd��| jd k s$t�|d kr<| j�dd � nt|��� | jd< | �� d S )Nzapplication/octet-streamzYSetting charset for application/octet-stream doesn't make sense, setup content_type firstr{ )rx r\ � _content_dictrO ro rp �lowerrz )rE rm �ctyper1 r1 r2 r{ 3 s c C s t | j�tj��S )zuThe value of Last-Modified HTTP header, or None. This header is represented as a `datetime` object. )r rC rn r � LAST_MODIFIED)rE r1 r1 r2 � last_modifiedB s z StreamResponse.last_modifiedc C s� |d kr| j �tjd � nlt|ttf�rJt�dt� t � |���| j tj< n<t|t j �rpt�d|� � �| j tj< nt|t�r�|| j tj< d S )Nz%a, %d %b %Y %H:%M:%S GMT)rC ro r r ra rP �float�time�strftime�gmtime�math�ceil�datetime� utctimetuplerp )rE rm r1 r1 r2 r� J s c C sZ | j �tj�}|sd S |tkr(ttd�S t�|�}|s:d S |�dd�\}}tt |�|d�S )N)rm r � )�is_weakrm ) rC rn r �ETAGr r r � fullmatch�groupr` )rE Z quoted_value�matchr� rm r1 r1 r2 �etag[ s zStreamResponse.etagc C s� |d kr| j �tjd � n�t|t�r,|tks@t|t�rN|jtkrNt| j tj< n�t|t�rvt |� d|� d�| j tj< ndt|t�r�t|jt�r�t |j� |j r�d|j� d�n d|j� d�}|| j tj< nt dt |�� d���d S )N�"zW/"zUnsupported etag type: z . etag must be str, ETag or None) rC ro r r� ra rp r r rm r r� � ValueErrorr_ )rE rm Z hdr_valuer1 r1 r2 r� k s ")� CONTENT_TYPEr7 c C s^ | j d k st�| jd k s t�d�dd� | j �� D ��}|rJ| jd | }n| j}|| j|< d S )Nz; c s s | ]\}}|� d |� �V qdS )�=Nr1 )�.0�k�vr1 r1 r2 � <genexpr>� s z?StreamResponse._generate_content_type_header.<locals>.<genexpr>)r| rO ry �join�itemsrC )rE r� �paramsr~ r1 r1 r2 rz � s z,StreamResponse._generate_content_type_header)�codingr7 c � sH |t jkrD| jd k st�|j| jtj< | j�|j� | j� tj d � d S )N) r% r- r? rO rm rC r �CONTENT_ENCODINGrb Zpopallr[ )rE r� r1 r1 r2 �_do_start_compression� s z$StreamResponse._do_start_compressionr) )�requestr7 c � s^ | j r| �| j �I d H n@|j�tjd��� }x*tD ]"}|j|kr4| �|�I d H d S q4W d S )NrN ) r< r� r6 rn r ZACCEPT_ENCODINGr} r% rm )rE r� Zaccept_encodingr� r1 r1 r2 �_start_compression� s z!StreamResponse._start_compressionc � s* | j r d S | jd k r| jS | �|�I d H S )N)r@ r? �_start)rE r� r1 r1 r2 �prepare� s zStreamResponse.preparec � sB || _ |j }| _| �� I d H |�| �I d H | �� I d H |S )N)r> r? �_prepare_headersZ _prepare_hook�_write_headers)rE r� �writerr1 r1 r2 r� � s zStreamResponse._startc � s� | j }|d k st�| j}|d k s$t�| j}|d kr8|j}|| _|j}| j}x4| j�� D ]&}|j dd�dd � }|� t j |� qVW | j r�| �|�I d H | jr�|tkr�td�|j���|�� d|t j< t j|kr�|t j= n~| j�rX| j|_|jd k�r:|tk�r4| jdk�r4|�� d|t j< t j|k�r8|t j= nd}n |tk�rX| jdk�rX|t j= | jd k�rr|�t jd � |�t j t � � |�t j t � t j |k�r�|�r�|t!k�r�d |t j < n|tk�r�d |t j < d S ) NrN )�headerr z@Using chunked encoding is forbidden for HTTP/{0.major}.{0.minor}rJ �� F)�d �e �f �g r� )r� i0 zapplication/octet-streamz keep-alive�close)"r> rO r? r9 rR rk rC r= �values�output�addr Z SET_COOKIEr; r� r: r! r\ �formatZenable_chunkingZTRANSFER_ENCODINGr[ � _length_checkrv �lengthr4 � setdefaultr� ZDATEr ZSERVERr Z CONNECTIONr )rE r� r� rR rk r6 Zcookierm r1 r1 r2 r� � sZ z StreamResponse._prepare_headersc � s^ | j }|d k st�| j}|d k s$t�|j}d�|d |d | j| j�}|�|| j�I d H d S )NzHTTP/{}.{} {} {}r r ) r> rO r? rk r� rI rL Z write_headersrC )rE r� r� rk Z status_liner1 r1 r2 r� � s z StreamResponse._write_headers)�datar7 c � sV t |tttf�s tdt|� ��| jr.td��| jd kr@td��| j� |�I d H d S )Nz#data argument must be byte-ish (%r)z%Cannot call write() after write_eof()z$Cannot call write() before prepare()) ra �bytes� bytearray� memoryviewrO r_ r@ r\ r? �write)rE r� r1 r1 r2 r� � s zStreamResponse.writec � sD | j rtd��| jd k s td��tjdtdd� | j�� I d H d S )NzEOF has already been sentz Response has not been startedz2drain method is deprecated, use await resp.write()r� )� stacklevel)r@ rO r? rU rV rW �drain)rE r1 r1 r2 r� s zStreamResponse.drain� c � sn t |tttf�s tdt|� ��| jr*d S | jd k s<td��| j�|�I d H d| _d | _ | jj | _ d | _d S )Nz#data argument must be byte-ish (%r)z Response has not been startedT) ra r� r� r� rO r_ r@ r? � write_eofr> Z output_sizerA )rE r� r1 r1 r2 r� s zStreamResponse.write_eofc C s\ | j r d}n2| jr:| jd k s t�| jj� d| jj� d�}nd}d| jj� d| j� d|� d�S )N�eof� z not prepared�<�>) r@ rG r> rO �methodrh rw r. r5 )rE �infor1 r1 r2 �__repr__ s zStreamResponse.__repr__)�keyr7 c C s | j | S )N)rB )rE r� r1 r1 r2 � __getitem__( s zStreamResponse.__getitem__)r� rm r7 c C s || j |< d S )N)rB )rE r� rm r1 r1 r2 � __setitem__+ s zStreamResponse.__setitem__c C s | j |= d S )N)rB )rE r� r1 r1 r2 � __delitem__. s zStreamResponse.__delitem__c C s t | j�S )N)�lenrB )rE r1 r1 r2 �__len__1 s zStreamResponse.__len__c C s t | j�S )N)�iterrB )rE r1 r1 r2 �__iter__4 s zStreamResponse.__iter__c C s t t| ��S )N)�hash�id)rE r1 r1 r2 �__hash__7 s zStreamResponse.__hash__)�otherr7 c C s | |kS )Nr1 )rE r� r1 r1 r2 �__eq__: s zStreamResponse.__eq__)N)N)r� )Er. r/ r0 r� rP r rp r$ rF �propertyr` rG rH r4 rJ rK r5 r r r rD rR rS rT rY r] r r% rb r6 rc rs rt rv �setterrx r{ r� r� r� r r� r r� r rz r� r� r r� r� r� r� r� r� r� r� r� r r� r� r� r� r r� r� �objectr� � __classcell__r1 r1 )rw r2 r& L s� >' ; r& c sV e Zd Zdddddddddd� eeee ee ee ee ee ee ee dd� � fdd�Z e ee e e f d�dd ��Zejejejfe eedd �d d ��Ze ee d�d d ��Zejedd�dd ��Ze ee d�� fdd� �Zejee dd�dd��Zd"e dd�� fdd� Zded�� fdd� Zedd �d d �Zedd �� fd d!� Z� Z S )#r' Nr3 ) �bodyr4 r5 �textr6 rx r{ �zlib_executor_size� zlib_executor) r� r4 r5 r� r6 rx r{ r� r� r7 c s^ |d k r|d k rt d��|d kr(t� } nt|t�s<t|�} n|} |d k rXd|krXt d��|d k r�tj| kr||sr|r�t d��nRt|t�s�tdt|� ��|d kr�d}|d kr�d}|d | | tj< |�|�}d }nNtj| kr�|d k s�|d k r�t d��n*|d k �r |d k �r|d| 7 }|| tj< t � j ||| d � |d k �rB|| _ n|| _ d | _ || _| | _d S ) Nz&body and text are not allowed togetherr{ z,charset must not be in content_type argumentzPpassing both Content-Type header and content_type or charset params is forbiddenz text argument must be str (%r)z text/plainzutf-8z ; charset=)r4 r5 r6 )r� r ra r r� rp � TypeErrorr_ �encoderu rF r� r� �_compressed_body�_zlib_executor_size�_zlib_executor) rE r� r4 r5 r� r6 rx r{ r� r� Z real_headers)rw r1 r2 rF ? sL zResponse.__init__)r7 c C s | j S )N)r8 )rE r1 r1 r2 r� � s z Response.body)r� r� r[ r7 c C s� |d krd | _ d| _n�t|ttf�r2|| _ d| _n�ytj�|� | _ }W n& tjk rn t dt |� ��Y nX d| _| j }| j s�||kr�|j }|d k r�t|�||< ||kr�|j||< |jr�x&|j�� D ]\}}||kr�|||< q�W d | _d S )NFzUnsupported body type %rT)r8 � _body_payloadra r� r� r ZPAYLOAD_REGISTRYrn � LookupErrorr� r_ rC r: �sizerp rx r6 r� r� )rE r� r� r[ r6 r� r� rm r1 r1 r2 r� � s. c C s | j d krd S | j �| jp d�S )Nzutf-8)r8 �decoder{ )rE r1 r1 r2 r� � s z Response.text)r� r7 c C s` |d ks"t |t�s"tdt|� ��| jdkr2d| _| jd krBd| _|�| j�| _d| _d | _ d S )Nz text argument must be str (%r)zapplication/octet-streamz text/plainzutf-8F) ra rp rO r_ rx r{ r� r8 r� r� )rE r� r1 r1 r2 r� � s c sX | j r d S tj| jkr t� jS | jd k r2t| j�S | jr<d S | j d k rPt| j �S dS d S )Nr ) r: r r[ rC ru rv r� r� r� r8 )rE )rw r1 r2 rv � s zResponse.content_length)rm r7 c C s t d��d S )Nz#Content length is set automatically)r\ )rE rm r1 r1 r2 rv � s r� )r� r7 c � s� | j r d S | jd kr | j}n| j}|r4td|����| jd k sBt�| jd k sPt�|d k r�| jjtjksp| j dkr�t � � � I d H q�| j r�t t|�}|�| j�I d H t � � � I d H q�t � � t t|��I d H nt � � � I d H d S )Nz data arg is not supported, got )r� i0 )r@ r� r8 rO r> r? �_methodr Z METH_HEADrI ru r� r� r r" r� r� )rE r� r� r )rw r1 r2 r� � s" zResponse.write_eofr) )r� r7 c � sX | j sFtj| jkrF| jsF| jd k r:tt| j��| jtj< n d| jtj< t� � |�I d H S )N�0) r: r r[ rC r� r8 rp r� ru r� )rE r� )rw r1 r2 r� � s zResponse._start)� zlib_moder7 c C sB |dks t �tj|d�}| j}|d k s*t �|�|�|�� | _d S )Nr )Zwbits)rO �zlib� compressobjr8 �compress�flushr� )rE r� r� �body_inr1 r1 r2 �_compress_body� s zResponse._compress_body)r� r7 c � s� | j s | jr t� �|�I d H S |tjkr�|tjkr<dtj ntj}| j }|d k sTt �| j d k r�t |�| j kr�t �� �| j| j|�I d H n | �|� | j}|d k s�t �|j| jtj< tt |��| jtj< d S )N� )r� r: ru r� r% r- r, r� � MAX_WBITSr8 rO r� r� �asyncioZget_event_loopZrun_in_executorr� r� r� rm rC r r� rp r[ )rE r� r� r� Zbody_out)rw r1 r2 r� s z Response._do_start_compression)r� ) r. r/ r0 r rP r rp r$ r rF r� r r� r" r� r� r r� r[ r r� rv r� r r� r� r% r� r� r1 r1 )rw r2 r' > s8 4: # r' r3 zapplication/json)r� r� r4 r5 r6 rx �dumps) r� r� r� r4 r5 r6 rx r� r7 c C s6 | t k r"|s|rtd��n|| �}t||||||d�S )Nz3only one of data, text, or body should be specified)r� r� r4 r5 r6 rx )r r� r' )r� r� r� r4 r5 r6 rx r� r1 r1 r2 r( " s r( )Cr� �collections.abc� collectionsr� �enum�jsonr� r� rU r� �concurrent.futuresr Z http.cookiesr r �typingr r r r r r r r r r Z multidictr r rN r r �abcr Zhelpersr r r r r r r r r �httpr r r r! r" Ztypedefsr# r$ �__all__Z web_requestr) rp Z BaseClassZ _reserved�Enumr% r& r' r� r� rP r( r1 r1 r1 r2 �<module> sR 0 , u f(