ZS @ 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) CIMultiDictCIMultiDictProxy )hdrspayload) HeadersMixinrfc822_formatted_timesentinel) RESPONSESSERVER_SOFTWARE
HttpVersion10
HttpVersion11
ContentCodingStreamResponseResponse
json_responsec @ s" e Z d Z d Z d Z d Z d S)r deflategzipidentityN)__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 Tstatus reasonNheadersc 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)Ntask)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/ AssertionErrorintr2 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)warningswarnDeprecationWarningr' 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) typeboolr 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.cookiesexpiresdomainmax_agepath/securehttponlyversionc 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) zSet 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_valuepop) r- namevaluerJ rK rL rM rO rP rQ oldcr 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)supercontent_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 datetimetimezoneZutc)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 floattimestrftimegmtimemathceilrf 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 ).0ir 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 rd | | <n |
| k rd | | <d
j |
d |
d | j | j } | j | | | S) Nheaderr6 r z@Using chunked encoding is forbidden for HTTP/{0.major}.{0.minor}r3 Fzapplication/octet-streamz
keep-aliveclosezHTTP/{}.{} {} {}
r )r& r! r: rQ r' r+ r% valuesoutputaddr# rz r" rB formatZenable_chunking
_length_checkr[ length
setdefaultr
r2 r5 Z
write_headers)r- ry r r
CONNECTIONDATESERVERru 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)Neofz{} {} 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)hashid)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 qt | 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 rt 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)N0)
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 compressobjcompressr 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/jsondumpsc
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* "