U
'�7`Zf � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z d dl m
Z
d dlmZmZ d dlmZmZmZmZmZmZmZmZmZmZ d dl m Z m Z ddl m Z m!Z! dd l"m#Z# dd
l$m%Z%m&Z&m'Z'm(Z( dd l)m*Z*m+Z+m,Z,m-Z- dd l!m.Z. dd
l/m0Z0m1Z1 dZ2e�r<ddl3m4Z4 ee5ef Z6nej"jZ6e%�sTdej7d< G dd� dej8�Z9G dd� de6e&�Z:G dd� de:�Z;e(fddddddej<d�eee5 ee= e>ee5 ee1 e5e0e;d� d d �Z?dS ) � N)�Executor)� parsedate)�Morsel� SimpleCookie)
�
TYPE_CHECKING�Any�Dict�Iterator�Mapping�MutableMapping�Optional�Tuple�Union�cast)� CIMultiDict�istr� )�hdrs�payload)�AbstractStreamWriter)�PY_38� HeadersMixin�rfc822_formatted_time�sentinel)� 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) � r- r- �G/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/web_response.pyr! 4 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 ��Zdpee dd!�d"d#�Zdqeee
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(j)fe*ddC�dDdE�Z+eddF�dGdH�Z,dIddJ�dKdL�Z-dIee. dJ�dMdN�Z/dIe.dJ�dOdP�Z0dd�dQdR�Z1dd�dSdT�Z2e3ddU�dVdW�Z4dd�dXdY�Z5dre3ddU�d[d\�Z6ed�d]d^�Z7ee8d_�d`da�Z9ee8ddb�dcdd�Z:edd_�dedf�Z;ed�dgdh�Z<e=e d�didj�Z>ed�dkdl�Z?e@e
dm�dndo�ZA� ZBS )sr" T�� N��status�reason�headers)r1 r2 r3 �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)�selfr1 r2 r3 r- r- r. �__init__G s zStreamResponse.__init__�r4 c C s
| j d k S �N)r< �rB r- r- r. �preparedb s zStreamResponse.preparedzasyncio.Task[None]c C s t | jdd �S )N�task)�getattrr; rF r- r- r. rH f s zStreamResponse.taskc C s | j S rE )�_statusrF r- r- r. r1 j s zStreamResponse.statusc C s | j S rE )r7 rF r- r- r. �chunkedn s zStreamResponse.chunkedc C s | j S rE )r8 rF r- r- r. � compressionr s zStreamResponse.compressionc C s | j S rE )�_reasonrF r- r- r. r2 v s zStreamResponse.reason)r1 r2 �
_RESPONSESr4 c C sV | j rtd��t|�| _|d krLz|| 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�intrJ � ExceptionrM )rB r1 r2 rN r- r- r. rA z s �
zStreamResponse.set_statusc C s | j S rE �r6 rF r- r- r. �
keep_alive� s zStreamResponse.keep_alivec C s
d| _ d S )NFrS rF r- r- r. � force_close� s zStreamResponse.force_closec C s | j S rE )r> rF r- r- r. � body_length� s zStreamResponse.body_lengthc C s t �dt� | jst�| jjS )Nzoutput_length is deprecated)�warnings�warn�DeprecationWarningr< rP Z buffer_sizerF r- r- r. �
output_length� s
z StreamResponse.output_length)�
chunk_sizer4 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)r7 r �CONTENT_LENGTHr@ � RuntimeErrorrW rX rY )rB r[ r- r- r. �enable_chunked_encoding� s �z&StreamResponse.enable_chunked_encoding)�forcer4 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) rW rX rY �
isinstancerP r8 r9 )rB r_ r- r- r. �enable_compression� s � �z!StreamResponse.enable_compressionzCIMultiDict[str]c C s | j S rE )r@ rF r- r- r. r3 � s zStreamResponse.headerszSimpleCookie[str]c C s | j S rE )r: rF r- r- r. �cookies� s zStreamResponse.cookies�/)�expires�domain�max_age�path�secure�httponly�versionr& ) �name�valuerf rg rh ri rj rk rl r& r4 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.
NrO rf � Thu, 01 Jan 1970 00:00:00 GMTrg zmax-ageri rj rk rl r& )r: �getZ coded_value�pop�str)
rB rm rn rf rg rh ri rj rk rl r& �old�cr- r- r. �
set_cookie� s0
zStreamResponse.set_cookie)rg ri )rm rg ri r4 c C s( | j �|d� | j|ddd||d� dS )zBDelete cookie.
Creates new empty expired cookie.
NrO r ro )rh rf rg ri )r: rq ru )rB rm rg ri r- r- r. �
del_cookie� s �zStreamResponse.del_cookiec s t � jS rE )�super�content_lengthrF �� __class__r- r. rx s z StreamResponse.content_length�rn r4 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)rQ r7 r] rr r@ r r\ rq �rB rn r- r- r. rx s �c s t � jS rE )rw � content_typerF ry r- r. r} s zStreamResponse.content_typec C s | j t|�| _| �� d S rE )r} rr �
_content_type� _generate_content_type_headerr| r- r- r. r} s
c s t � jS rE )rw �charsetrF ry r- r. r� # s zStreamResponse.charsetc C sZ | j }|dkrtd��| jd k s$t�|d kr<| j�dd � nt|��� | jd< | �� d S )N�application/octet-streamzYSetting charset for application/octet-stream doesn't make sense, setup content_type firstr� )r} r] �
_content_dictrP rq rr �lowerr )rB rn �ctyper- r- r. r� ( s �c C sF | j �tj�}|dk rBt|�}|dk rBtj|dd� dtjji�S dS )zuThe value of Last-Modified HTTP header, or None.
This header is represented as a `datetime` object.
N� Ztzinfo)r@ rp r �
LAST_MODIFIEDr �datetime�timezoneZutc)rB ZhttpdateZ timetupler- r- r. �
last_modified7 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)r@ rq r r� rb rQ �float�time�strftime�gmtime�mathZceilr� Z utctimetuplerr r| r- r- r. r� D s � �
)� CONTENT_TYPEr4 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 )�=Nr- )�.0�k�vr- r- r. � <genexpr>Z s z?StreamResponse._generate_content_type_header.<locals>.<genexpr>)r� rP r~ �join�itemsr@ )rB r� Zparamsr� r- r- r. r U s z,StreamResponse._generate_content_type_header��codingr4 c � sH |t jkrD| jd k st�|j| jtj< | j�|j� | j� tj
d � d S rE ) r! r) r< rP rn r@ r �CONTENT_ENCODINGrc Zpopallr\ )rB r� r- r- r. �_do_start_compressiona s
z$StreamResponse._do_start_compressionr% ��requestr4 c � s\ | j r| �| j �I d H n>|j�tjd��� }tD ]$}|j|kr2| �|�I d H d S q2d S )NrO ) r9 r� r3 rp r ZACCEPT_ENCODINGr� r! rn )rB r� Zaccept_encodingr� r- r- r. �_start_compressionj s
z!StreamResponse._start_compressionc � s* | j r
d S | jd k r| jS | �|�I d H S rE )r= r< �_start�rB r� r- r- r. �preparet s
zStreamResponse.preparec � sB || _ |j }| _| �� I d H |�| �I d H | �� I d H |S rE )r; r< �_prepare_headersZ
_prepare_hook�_write_headers)rB r� �writerr- r- r. r� | s zStreamResponse._startc � s� | j }|d k st�| j}|d k s$t�| j}|d kr8|j}|| _|j}| j}| j�� D ]&}|j dd�dd � }|�
t j |� qT| j
r�| �|�I d H | jr�|tkr�td�|j���|�� d|t j< t j|kr�|t j= nr| j�rH| j|_|jd k�r*|tk�r$|�� d|t j< t j|k�r(|t j= nd}n |tk�rH| jdk�rH|t j= |�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 ) NrO )�headerr z@Using chunked encoding is forbidden for HTTP/{0.major}.{0.minor}rK F)�d �e �f �g �� r� z
keep-alive�close)"r; rP r< r6 rT rl r@ r: �values�output�addr Z
SET_COOKIEr8 r� r7 r r] �formatZenable_chunkingZTRANSFER_ENCODINGr\ �
_length_checkrx Zlengthr1 �
setdefaultr� ZDATEr ZSERVERr Z
CONNECTIONr )rB r� r� rT rl r3 Zcookiern r- r- r. r� � s\ ��
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; rP r< rl r� rJ rM Z
write_headersr@ )rB r� r� rl Z status_liner- r- r. r� � s �z StreamResponse._write_headers��datar4 c � sV t |tttf�s tdt|� ��| jr.td��| jd kr@td��| j� |�I d H d S )N�#data argument must be byte-ish (%r)z%Cannot call write() after write_eof()z$Cannot call write() before prepare())
rb �bytes� bytearray�
memoryviewrP r` r= r] r< �write�rB r� r- r- r. r� � s �
�
zStreamResponse.writec � sD | j rtd��| jd k s td��tjdtdd� | j�� I d H d S )NzEOF has already been sent� Response has not been startedz2drain method is deprecated, use await resp.write()� )�
stacklevel)r= rP r< rW rX rY �drainrF r- r- r. 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 )Nr� r� T) rb r� r� r� rP r` r= r< � write_eofr; Z output_sizer> r� r- r- r. 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; rP �methodri rz r* r2 )rB �infor- r- r. �__repr__� s zStreamResponse.__repr__)�keyr4 c C s
| j | S rE �r? �rB r� r- r- r. � __getitem__� s zStreamResponse.__getitem__)r� rn r4 c C s || j |< d S rE r� )rB r� rn r- r- r. � __setitem__� s zStreamResponse.__setitem__c C s | j |= d S rE r� r� r- r- r. � __delitem__ s zStreamResponse.__delitem__c C s
t | j�S rE )�lenr? rF r- r- r. �__len__ s zStreamResponse.__len__c C s
t | j�S rE )�iterr? rF r- r- r. �__iter__ s zStreamResponse.__iter__c C s t t| ��S rE )�hash�idrF r- r- r. �__hash__ s zStreamResponse.__hash__)�otherr4 c C s | |kS rE r- )rB r� r- r- r. �__eq__ s zStreamResponse.__eq__)N)N)r� )Cr* r+ r, r� rQ r rr r rC �propertyra rG rH r1 rK rL r2 r r
r
rA rT rU rV rZ r^ r r! rc r3 rd ru rv rx �setterr} r� r� r� r� r r� r r r� r� r r� r� r� r� r� r� r� r� r� r r� r� r� r� r r� r� �objectr� �
__classcell__r- r- ry r. r" C s� �� �� �� �� 5 � � ���
:
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# Nr/ ) �bodyr1 r2 �textr3 r} r� �zlib_executor_size�
zlib_executor)
r� r1 r2 r� r3 r} r� r� r� r4 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 forbidden� text argument must be str (%r)�
text/plain�utf-8z
; charset=r0 )�
ValueErrorr rb r r� rr � TypeErrorr` �encoderw rC r� r� �_compressed_body�_zlib_executor_size�_zlib_executor) rB r� r1 r2 r� r3 r} r� r� r� Z real_headersry r- r. rC sP
�
�
zResponse.__init__rD c C s | j S rE )r5 rF r- r- r. r� Y s z
Response.body)r� r� r\ r4 c C s� |d krd | _ d| _n�t|ttf�r2|| _ d| _n�ztj�|� | _ }W n& tjk
rn t dt
|� ��Y nX d| _| j }| j s�||kr�|j
}|d k r�t|�||< ||kr�|j||< |jr�|j�� D ]\}}||kr�|||< q�d | _d S )NFzUnsupported body type %rT)r5 �
_body_payloadrb r� r� r ZPAYLOAD_REGISTRYrp � LookupErrorr� r` r@ r7 �sizerr r} r3 r� r� )rB r� r� r\ r3 r� r� rn r- r- r. r� ] s.
c C s | j d krd S | j �| jp d�S )Nr� )r5 �decoder� rF r- r- r. r� � s
z
Response.text)r� r4 c C s` |d ks"t |t�s"tdt|� ��| jdkr2d| _| jd krBd| _|�| j�| _d| _d | _ d S )Nr� r� r� r� F)
rb rr rP r` r} r� r� r5 r� r� )rB r� r- r- r. 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 )
r7 r r\ r@ rw rx r� r� r� r5 rF ry r- r. rx � s
zResponse.content_lengthr{ c C s t d��d S )Nz#Content length is set automatically)r] r| r- r- r. rx � s r� r� 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� r5 rP r; r< �_methodr Z METH_HEADrJ rw r� r� r r r� r� )rB r� r� r ry r- r. r� � s"
zResponse.write_eofr% r� c � sX | j sFtj| jkrF| jsF| jd k r:tt| j��| jtj< n d| jtj< t� � |�I d H S )N�0)
r7 r r\ r@ r� r5 rr r� rw r� r� ry r- r. r� � s
zResponse._start)� zlib_moder4 c C sB |dks t �tj|d�}| j}|d k s*t �|�|�|�� | _d S )Nr )Zwbits)rP �zlib� compressobjr5 �compress�flushr� )rB r� r� �body_inr- r- r. �_compress_body� s
zResponse._compress_bodyr� 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� r7 rw r� r! r) r( r� Z MAX_WBITSr5 rP r� r� �asyncioZget_event_loopZrun_in_executorr� r� r� rn r@ r r� rr r\ )rB r� r� r� Zbody_outry r- r. r� � s*
� � � �
z Response._do_start_compression)r� ) r* r+ r, r rQ r rr r r rC r� r r� r r� r� r r� r\ r r� rx r� r r� r� r! r� r� r- r- ry r. r# sX ��F ��(
r# r/ zapplication/json)r� r� r1 r2 r3 r} �dumps) r� r� r� r1 r2 r3 r} r� r4 c C s6 | t k r"|s|rtd��n|| �}t||||||d�S )Nz3only one of data, text, or body should be specified)r� r� r1 r2 r3 r} )r r� r# )r� r� r� r1 r2 r3 r} r� r- r- r. r$ � s
�r$ )@r� Zcollections.abc� collectionsr� �enumZjsonr� r� rW r� �concurrent.futuresr Z email.utilsr Z http.cookiesr r �typingr r r r r
r r r
r r Z multidictr r rO r r �abcr Zhelpersr r r r Zhttpr r r r r Ztypedefsr r �__all__Z web_requestr% rr Z BaseClassZ _reserved�Enumr! r" r# r� r� rQ r$ r- r- r- r. �<module> sj 0
R f���