���Z�S � @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l
m Z d d l m
Z
m Z d d l m Z m Z d d l m Z m Z m Z d d l m Z m Z m Z m Z d d
d d f Z Gd
d � d e j � Z Gd d
� d
e j e � Z Gd d � d e � Z e d d d d d d d d d d d d d e j! d d �Z" d S)� N)� parsedate)� SimpleCookie)� CIMultiDict�CIMultiDictProxy� )�hdrs�payload)� HeadersMixin�rfc822_formatted_time�sentinel)� RESPONSES�SERVER_SOFTWARE�
HttpVersion10�
HttpVersion11�
ContentCoding�StreamResponse�Response�
json_responsec @ s" e Z d Z d Z d Z d Z d S)r �deflate�gzip�identityN)�__name__�
__module__� __qualname__r r r � r r �/web_response.pyr s c s( e Z d Z d Z d d d d d d d d � Z e d d
� � Z e d d � � Z e d
d � � Z e d d � � Z e d d � � Z
e d d � � Z d e d d � Z
e d d � � Z d d � Z e d d � � Z e d d � � Z d d d � Z d d! d" � Z e d# d$ � � Z e d% d& � � Z d' d d( d d) d d* d+ d, d d- d d. d d/ d0 � Z d( d d* d+ d1 d2 � Z e � f d3 d4 � � Z e j d5 d4 � � Z e � f d6 d7 � � Z e j d8 d7 � � Z e � f d9 d: � � Z e j d; d: � � Z e e j d< d= � � Z e j d> d= � � Z e j d? d@ � Z dA dB � Z! dC dD � Z" dE dF � Z# e$ e% e j& e j' e j( e j e j) e j* e+ e j, dG dH �
Z- dI dJ � Z. dK dL � Z/ dM dN dO � Z0 dP dQ � Z1 dR dS � Z2 dT dU � Z3 dV dW � Z4 dX dY � Z5 dZ d[ � Z6 d\ d] � Z7 � S)^r T�status�� �reasonN�headersc C s� d | _ d | _ d | _ d | _ d | _ t � | _ d | _ d | _ d | _ d | _
i | _ | d k r� t | � | _
n t � | _
| j | | � d S)NFr )�_body� _keep_alive�_chunked� _compression�_compression_forcer �_cookies�_req�_payload_writer� _eof_sent� _body_length�_stater �_headers�
set_status)�selfr r r r r r �__init__) s zStreamResponse.__init__c C s
| j d k S)N)r' )r- r r r �prepared> s zStreamResponse.preparedc C s t | j d d � S)N�task)�getattrr& )r- r r r r0 B s zStreamResponse.taskc C s | j S)N)�_status)r- r r r r F s zStreamResponse.statusc C s | j S)N)r" )r- r r r �chunkedJ s zStreamResponse.chunkedc C s | j S)N)r# )r- r r r � compressionN s zStreamResponse.compressionc C s | j S)N)�_reason)r- r r r r R s zStreamResponse.reasonc C sn | j s t d � � t | � | _ | d k ra y | | j d } Wn t k
r` d } Yn X| | _ d S)NzGCannot change the response status code after the headers have been sentr � )r/ �AssertionError�intr2 � Exceptionr5 )r- r r Z
_RESPONSESr r r r, V s
zStreamResponse.set_statusc C s | j S)N)r! )r- r r r �
keep_aliveb s zStreamResponse.keep_alivec C s
d | _ d S)NF)r! )r- r r r � force_closef s zStreamResponse.force_closec C s | j S)N)r) )r- r r r � body_lengthi s zStreamResponse.body_lengthc C s t j d t � | j j S)Nzoutput_length is deprecated)�warnings�warn�DeprecationWarningr' Z buffer_size)r- r r r �
output_lengthm s z StreamResponse.output_lengthc C sG d | _ t j | j k r' t d � � | d k rC t j d t � d S)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_LENGTHr+ � RuntimeErrorr= r> r? )r- Z
chunk_sizer r r �enable_chunked_encodingr s
z&StreamResponse.enable_chunked_encodingc C sj t | � t k r- | r! t j n t j } n' | d k rT t | t � sT t d � � d | _ | | _ d S)z&Enables response compression encoding.Nz1force should one of None, bool or ContentEncodingT) �type�boolr r r �
isinstancer7 r# r$ )r- �forcer r r �enable_compression| s z!StreamResponse.enable_compressionc C s | j S)N)r+ )r- r r r r � s zStreamResponse.headersc C s | j S)N)r% )r- r r r �cookies� s zStreamResponse.cookies�expires�domain�max_age�path�/�secure�httponly�versionc C s! | j j | � }
|
d k r@ |
j d k r@ | j j | d � | | j | <| j | } | d k rs | | d <n | j d � d k r� | d =| d k r� | | d <| d k r� | | d <n d | k r� | d =| | d <| d k r� | | d <| d k r| | d <| d k r | | d
<d S) z�Set or update response cookie.
Sets new cookie or updates existent with new value.
Also updates only those params which are not None.
Nr6 rJ z Thu, 01 Jan 1970 00:00:00 GMTrK zmax-agerM rO rP rQ )r% �getZ coded_value�pop) r- �name�valuerJ rK rL rM rO rP rQ �old�cr r r �
set_cookie� s,
zStreamResponse.set_cookiec C s? | j j | d � | j | d d d d d d | d | �d S) zBDelete cookie.
Creates new empty expired cookie.
Nr6 rL r rJ z Thu, 01 Jan 1970 00:00:00 GMTrK rM )r% rS rX )r- rT rK rM r r r �
del_cookie� s zStreamResponse.del_cookiec s
t � j S)N)�super�content_length)r- )� __class__r r r[ � s z StreamResponse.content_lengthc C s` | d k rF t | � } | j r- t d � � t | � | j t j <n | j j t j d � d S)Nz<You can't set content length when chunked encoding is enable)r8 r" rB �strr+ r rA rS )r- rU r r r r[ � s c s
t � j S)N)rZ � content_type)r- )r\ r r r^ � s zStreamResponse.content_typec C s$ | j t | � | _ | j � d S)N)r^ r] �
_content_type� _generate_content_type_header)r- rU r r r r^ � s c s
t � j S)N)rZ �charset)r- )r\ r r ra � s zStreamResponse.charsetc C sj | j } | d k r! t d � � | d k rC | j j d d � n t | � j � | j d <| j � d S)Nzapplication/octet-streamzYSetting charset for application/octet-stream doesn't make sense, setup content_type firstra )r^ rB �
_content_dictrS r] �lowerr` )r- rU �ctyper r r ra � s c C s] | j j | � } | d k rY t | � } | d k rY t j | d d � d t j j � Sd S)zuThe value of Last-Modified HTTP header, or None.
This header is represented as a `datetime` object.
N� Ztzinfo)r rR r �datetime�timezoneZutc)r- Z_LAST_MODIFIEDZhttpdateZ timetupler r r �
last_modified� s
z StreamResponse.last_modifiedc C s� | d k r% | j j t j d � n� t | t t f � rk t j d t j t
j | � � � | j t j <nV t | t j � r� t j d | j
� � | j t j <n t | t � r� | | j t j <d S)Nz%a, %d %b %Y %H:%M:%S GMT)r rS r �
LAST_MODIFIEDrF r8 �float�time�strftime�gmtime�math�ceilrf Z utctimetupler] )r- rU r r r rh � s + c C sY d j d d � | j j � D� � } | r? | j d | } n | j } | | j | <d S)Nz; c s s | ] } d | Vq d S)z%s=%sNr )�.0�ir r r � <genexpr> s z?StreamResponse._generate_content_type_header.<locals>.<genexpr>)�joinrb �itemsr_ r )r- � CONTENT_TYPEZparamsrd r r r r`
s
% z,StreamResponse._generate_content_type_headerc C sO | t j k rK | j | j t j <| j j | j � | j j t j
d � d S)N) r r rU r r �CONTENT_ENCODINGr' rH r+ ZpopallrA )r- �codingr r r �_do_start_compression s z$StreamResponse._do_start_compressionc C so | j r | j | j � nO | j j t j d � j � } x. t D]& } | j | k rA | j | � d SqA Wd S)Nr6 ) r$ rx r rR r ZACCEPT_ENCODINGrc r rU )r- �requestZaccept_encodingrw r r r �_start_compression s
z!StreamResponse._start_compressionc � sB | j r
d S| j d k r# | j S| j | � Id H| j | � S)N)r( r' Z
_prepare_hook�_start)r- ry r r r �prepare% s zStreamResponse.preparec C s | | _ | j } | d k r' | j } | | _ | j }
| j } | _ | j } xC | j j � D]2 } | j d d � d d � } | j | | � qb W| j
r� | j | � | j r|
| k r� t
d j | j � � � | j � d | | <| | k re| | =n` | j re| j | _ | j d k re|
| k r_| j � d | | <| | k re| | =n d } | j | d � | j | t � � | j | |
� | | k r�| r�|
| k r�d | | <n |
| k r�d | | <d
j |
d |
d | j | j � } | j | | � | S) N�headerr6 r z@Using chunked encoding is forbidden for HTTP/{0.major}.{0.minor}r3 Fzapplication/octet-streamz
keep-alive�closezHTTP/{}.{} {} {}
r )r& r! r: rQ r' r+ r% �values�output�addr# rz r" rB �formatZenable_chunking�
_length_checkr[ �length�
setdefaultr
r2 r5 Z
write_headers)r- ry r r �
CONNECTION�DATE�SERVERru rA �
SET_COOKIEr
�TRANSFER_ENCODINGr: rQ �writerr ZcookierU Z status_liner r r r{ . sX
zStreamResponse._startc � sw t | t t t f � s. t d t | � � � | j rC t d � � | j d k r^ t d � � | j j | � Id Hd S)Nz#data argument must be byte-ish (%r)z%Cannot call write() after write_eof()z$Cannot call write() before prepare())
rF �bytes� bytearray�
memoryviewr7 rD r( rB r' �write)r- �datar r r r� s s zStreamResponse.writec � s] | j s t d � � | j d k s1 t d � � t j d t d d �| j j � Id Hd S)NzEOF has already been sentz Response has not been startedz2drain method is deprecated, use await resp.write()�
stacklevel� )r( r7 r' r= r> r? �drain)r- r r r r� ~ s zStreamResponse.drain� c � s� t | t t t f � s. t d t | � � � | j r; d S| j d k sV t d � � | j j | � Id Hd | _ d | _ | j j
| _ d | _ d S)Nz#data argument must be byte-ish (%r)z Response has not been startedT) rF r� r� r� r7 rD r( r' � write_eofr& Z output_sizer) )r- r� r r r r� � s zStreamResponse.write_eofc C s^ | j r d } n0 | j r< d j | j j | j j � } n d } d j | j j | j | � S)N�eofz{} {} z not preparedz
<{} {} {}>) r( r/ r� r& �methodrM r\ r r )r- �infor r r �__repr__� s !zStreamResponse.__repr__c C s | j | S)N)r* )r- �keyr r r � __getitem__� s zStreamResponse.__getitem__c C s | | j | <d S)N)r* )r- r� rU r r r � __setitem__� s zStreamResponse.__setitem__c C s | j | =d S)N)r* )r- r� r r r � __delitem__� s zStreamResponse.__delitem__c C s
t | j � S)N)�lenr* )r- r r r �__len__� s zStreamResponse.__len__c C s
t | j � S)N)�iterr* )r- r r r �__iter__� s zStreamResponse.__iter__c C s t t | � � S)N)�hash�id)r- r r r �__hash__� s zStreamResponse.__hash__)8r r r r� r. �propertyr/ r0 r r3 r4 r r r, r: r; r< r@ rC rH r rI rX rY r[ �setterr^ ra r ri rh ru r` rx rz r| r r r� r� r� rA r� r
r� r{ r� r� r� r� r� r� r� r� r� r� r r )r\ r r % sl
%
;
c s e Z d Z d d d d d d d d d d d d d d � f d
d � Z e d d
� � Z e j e j e j d d
� � Z e d d � � Z
e
j d d � � Z
e � f d d � � Z e j d d � � Z � f d d � Z � f d d � Z
� f d d � Z � S)r �bodyNr r r �textr r^ ra c s� | d k r$ | d k r$ t d � � | d k r< t � } n! t | t t f � s] t | � } | d k r� d | k r� t d � � | d k r-t j | k r� | s� | r*t d � � q�t | t � s� t d t | � � � | d k r� d } | d k r d } | d | | t j <| j | � } d } ni t j | k rc| d k sT| d k r�t d � � n3 | d k r�| d k r�| d | 7} | | t j <t
� j d | d
| d | � | d k r�| | _ n | | _
d | _ d S) Nz&body and text are not allowed togetherra 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=r r r )�
ValueErrorr rF r r ru r] � TypeErrorrD �encoderZ r. r� r� �_compressed_body)r- r� r r r� r r^ ra )r\ r r r. � sD
zResponse.__init__c C s | j S)N)r )r- r r r r� � s z
Response.bodyc C sH | d k r! d | _ d | _ nt | t t f � rK | | _ d | _ n� y t j j | � | _ } Wn+ t j k
r� t d t
| � � � Yn Xd | _ | j } | j r� | | k r� | j
} | d k r� t | � | | <| | k r� | j | | <| j r;x3 | j j � D]" \ } } | | k r| | | <qWd | _ d S)NFzUnsupported body type %rT)r �
_body_payloadrF r� r� r ZPAYLOAD_REGISTRYrR � LookupErrorr� rD r+ r" �sizer] r^ r rt r� )r- r� ru rA r r� r� rU r r r r� � s.
c C s, | j d k r d S| j j | j p( d � S)Nzutf-8)r �decodera )r- r r r r� s z
Response.textc C s� | d k s1 t | t � s1 t d t | � � � | j d k rI d | _ | j d k ra d | _ | j | j � | _ d | _ d | _ d S)Nz text argument must be str (%r)zapplication/octet-streamz
text/plainzutf-8F)
rF r] r7 rD r^ ra r� r r� r� )r- r� r r r r� s c sv | j r
d St j | j k r) t � j S| j d k rE t | j � S| j rR d S| j d k rn t | j � Sd Sd S)Nr )
r" r rA r rZ r[ r� r� r� r )r- )r\ r r r[ + s
zResponse.content_lengthc C s t d � � d S)Nz#Content length is set automatically)rB )r- rU r r r r[ > s c � s� | j r
d S| j d k r( | j } n | j } | d k r� | j j t j k sa | j d k rv t � j � Id Hq� | j
r� | j | j � Id Ht � j � Id Hq� t � j | � Id Hn t � j � Id Hd S)N�� �0 )r� r� )
r( r� r r& �_methodr Z METH_HEADr2 rZ r� r� r� r' )r- r� )r\ r r r� B s zResponse.write_eofc sv | j rf t j | j k rf | j sf | j d k rV t t | j � � | j t j <n d | j t j <t � j | � S)N�0)
r" r rA r+ r� r r] r� rZ r{ )r- ry )r\ r r r{ U s "zResponse._startc s� | j s | j r" t � j | � S| t j k r� | j d k rM d t j n t j } t j d | � } | j
| j � | j � | _
| j | j t j <t t | j
� � | j t j <d S)Nr � Zwbits)r� r" rZ rx r r rU �zlib� MAX_WBITS� compressobj�compressr �flushr� r+ r rv r] r� rA )r- rw Z zlib_moder� )r\ r r rx _ s &z Response._do_start_compression)r r r r. r� r� r� r ru rA r� r[ r� r{ rx r r )r\ r r � s 5#
r� r� r r r r r^ zapplication/json�dumpsc
C s^ | t k r3 | s | r' t d � � n | | � } t d | d | d | d | d | d | � S)Nz3only one of data, text, or body should be specifiedr� r� r r r r^ )r r� r )r� r� r� r r r r^ r� r r r r o s )#� collectionsrf �enumZjsonrn rk r= r� Z email.utilsr Z http.cookiesr Z multidictr r r6 r r Zhelpersr r
r Zhttpr r
r r �__all__�Enumr �MutableMappingr r r� r r r r r �<module> s* " � ��