���Z�Q � @ 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 Z d d l Z d d l
Z
d d l m Z d d l
m Z d d l m Z d d l m Z d d l Z d d l m Z m Z m Z m Z d d l m Z d d l m Z m Z d d
l m Z m Z m Z d d l! m" Z" d d l# m$ Z$ d
d d f Z% e j& d d d d � Gd d � d � � Z' e j( e j) d Z* d j+ d e* � Z, d j+ d j- d d � d d d f e. e/ d d � � D� � � Z0 d Z1 d! j+ d" e0 d# e1 � Z2 d$ j+ d% e, d& e2 � Z3 e j4 d' � Z5 e j4 e3 � Z6 Gd( d
� d
e j7 e � Z8 Gd) d � d e8 � Z9 d S)*� N)� parsedate)� SimpleCookie)�MappingProxyType)� parse_qsl)� CIMultiDict�CIMultiDictProxy� MultiDict�MultiDictProxy)�URL� )�hdrs� multipart)� HeadersMixin�reify�sentinel)�EmptyStreamReader)�HTTPRequestEntityTooLarge� BaseRequest� FileField�Request�frozenT�slotsc @ sm e Z d Z e j d e � Z e j d e � Z e j d e j � Z
e j d e � Z e j d e � Z
d S)r �typeN)�__name__�
__module__� __qualname__�attrZib�str�name�filename�io�BufferedReader�file� content_typer �headers� r% r% �/web_request.pyr s
z!#$%&'*+.^_`|~-z
[{tchar}]+Ztcharz[{}]� c c s | ] } t | � Vq d S)N)�chr)�.0�cr% r% r& � <genexpr>- s r+ � � �! �# � z
\\[\t !-~]z "(?:{quoted_pair}|{qdtext})*"ZqdtextZ quoted_pairz#({token})=({token}|{quoted_string})�tokenZ
quoted_stringz \\([\t !-~])c @ s� e Z d Z e j e j e j e j e j h Z e
j e d d d d d d d d d d
d d d
d d g � BZ d dx d d d d d d d d d d � Z
d e d e d e d e d e d e 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 e d+ d, � � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z e d7 d8 � � Z e d9 d: � � Z e d; d< � � Z e d= d> � � Z! e d? d@ � � Z" e dA dB � � Z# e dC dD � � Z$ e dE dF � � Z% e dG dH � � Z& e dI dJ � � Z' e dK dL � � Z( e dM dN � � Z) e dO dP � � Z* e dQ dR � � Z+ e dS dT � � Z, e e j- dU dV � � Z. e dW dX � � Z/ e dY dZ � � Z0 e d[ e j1 d\ d] � � Z2 e d^ d_ � � Z3 e d` da � � Z4 e db dc � � Z5 e dd de � � Z6 df dg � Z7 dh di � Z8 dj dk � Z9 dl e: j; dm dn � Z: do e< j= dp dq � Z< dr ds � Z> dt du � Z? e@ jA dv dw � � ZB d S)yr �_message� _protocol�_payload_writer�_payload�_headers�_method�_version�_rel_url�_post� _read_bytes�_state�_cache�_task�_client_max_size�_loop�client_max_sizei � �stateN�scheme�host�remotec C s� | d k r i } | | _ | | _ | | _ | | _ | j | _ | j | _ | j | _ | j
| _ d | _ d | _
| | _ i | _ | | _ | | _ | | _ | d k r� | | j d <|
d k r� |
| j d <| d k r� | | j d <d S)NrD rE rF )r2 r3 r4 r5 r$ r6 �methodr7 �versionr8 �urlr9 r: r; r< r= r> r? r@ ) �self�messageZpayload�protocolZpayload_writer�task�looprA rC rD rE rF r% r% r& �__init__O s.
zBaseRequest.__init__rG �rel_urlr$ c
C sB | j r t d � � i } | t k r1 | | d <| t k rc t | � } | | d <t | � | d <| t k r� t | � | d <t d d � | j � D� � | d <| j j | � } i } | t k r� | | d <| t k r� | | d
<| t k r� | | d <| j
| | j | j | j
| j | j d | j d
| j j � | �S)a Clone itself with replacement some attributes.
Creates and returns a new instance of Request object. If no parameters
are given, an exact copy is returned. If a parameter is not passed, it
will reuse the one from the current request object.
z/Cannot clone request after reading it's contentrG rI �pathr$ c s s3 | ]) \ } } | j d � | j d � f Vq d S)zutf-8N)�encode)r) �k�vr% r% r& r+ � s z$BaseRequest.clone.<locals>.<genexpr>� raw_headersrD rE rF rA rC )r; � RuntimeErrorr r
r r �tuple�itemsr2 �_replace� __class__r5 r3 r4 r> r@ r? r< �copy)
rJ rG rP r$ rD rE rF �dctrK �kwargsr% r% r& �cloneo s>
zBaseRequest.clonec C s | j S)N)r> )rJ r% r% r& rM � s zBaseRequest.taskc C s | j S)N)r3 )rJ r% r% r& rL � s zBaseRequest.protocolc C s | j d k r d S| j j S)N)r3 � transport)rJ r% r% r& r_ � s zBaseRequest.transportc C s | j S)N)r4 )rJ r% r% r& �writer� s zBaseRequest.writerc C s | j S)N)r2 )rJ r% r% r& rK � s zBaseRequest.messagec C s | j S)N)r9 )rJ r% r% r& rP � s zBaseRequest.rel_urlc C s | j S)N)r@ )rJ r% r% r& rN � s zBaseRequest.loopc C s | j | S)N)r< )rJ �keyr% r% r& � __getitem__� s zBaseRequest.__getitem__c C s | | j | <d S)N)r< )rJ ra �valuer% r% r& � __setitem__� s zBaseRequest.__setitem__c C s | j | =d S)N)r< )rJ ra r% r% r& � __delitem__� s zBaseRequest.__delitem__c C s
t | j � S)N)�lenr< )rJ r% r% r& �__len__� s zBaseRequest.__len__c C s
t | j � S)N)�iterr< )rJ r% r% r& �__iter__� s zBaseRequest.__iter__c C s
| j d k S)z5A bool indicating if the request is handled with SSL.�https)rD )rJ r% r% r& �secure� s zBaseRequest.securec
C s� g } x�| j j j t j f � D]�} t | � } d } d } i } | j t j | � � xZd | k ov | k n r�t j
| | � } | d k r$| r� | j d | � } q�| j � \ } } | d d k r� t
j d | d d � � } | | | j � <| t | j d � � 7} d } q_ | | d k rcd } i } | j t j | � � | d 7} q_ | | d k r�d } | d 7} q_ | | d
k r�| d 7} q_ | j d | � } q_ Wq" Wt | � S) a� A tuple containing all parsed Forwarded header(s).
Makes an effort to parse Forwarded headers as specified by RFC 7239:
- It adds one (immutable) dictionary per Forwarded 'field-value', ie
per proxy. The element corresponds to the data in the Forwarded
field-value added by the first proxy encountered by the client. Each
subsequent item corresponds to those added by later proxies.
- It checks that every value has valid syntax in general as specified
in section 4: either a 'token' or a 'quoted-string'.
- It un-escapes found escape sequences.
- It does NOT validate 'by' and 'for' contents as specified in section
6.
- It does NOT validate 'host' contents (Host ABNF).
- It does NOT validate 'proto' contents for valid URI scheme names.
Returns a tuple containing one or more immutable dicts
r FN�,�"z\1r T�;z ���)r2 r$ Zgetallr Z FORWARDEDrf �append�typesr �_FORWARDED_PAIR_RE�match�find�groups�_QUOTED_PAIR_REPLACE_RE�sub�lower�grouprW )
rJ ZelemsZ field_value�length�posZneed_separator�elemrs r rc r% r% r& � forwarded� s> "
zBaseRequest.forwardedc C s | j j d � r d Sd Sd S)a A string representing the scheme of the request.
Hostname is resolved in this order:
- overridden value by .clone(scheme=new_scheme) call.
- type of connection to peer: HTTPS if socket is SSL, HTTP otherwise.
'http' or 'https'.
Z
sslcontextrj ZhttpN)r_ �get_extra_info)rJ r% r% r& rD s zBaseRequest.schemec C s | j S)zyRead only property for getting HTTP method.
The value is upper-cased str like 'GET', 'POST', 'PUT' etc.
)r7 )rJ r% r% r& rG ! s zBaseRequest.methodc C s | j S)zxRead only property for getting HTTP version of request.
Returns aiohttp.protocol.HttpVersion instance.
)r8 )rJ r% r% r& rH ) s zBaseRequest.versionc C s6 | j j j t j � } | d k r( | St j � Sd S)z�Hostname of the request.
Hostname is resolved in this order:
- overridden value by .clone(host=new_host) call.
- HOST HTTP header
- socket.getfqdn() value
N)r2 r$ �getr ZHOST�socket�getfqdn)rJ rE r% r% r& rE 1 s
zBaseRequest.hostc C sJ | j d k r d S| j j d � } t | t t f � rB | d S| Sd S)z�Remote IP of client initiated HTTP request.
The IP is resolved in this order:
- overridden value by .clone(remote=new_remote) call.
- peername of opened socket
N�peernamer )r_ r~ �
isinstance�listrW )rJ r� r% r% r& rF A s zBaseRequest.remotec C s. t j d | j d | j � } | j | j � S)NrD rE )r
ZbuildrD rE �joinr9 )rJ rI r% r% r& rI R s zBaseRequest.urlc C s
| j j S)z_The URL including *PATH INFO* without the host or scheme.
E.g., ``/app/blog``
)r9 rQ )rJ r% r% r& rQ W s zBaseRequest.pathc C s
t | j � S)zXThe URL including PATH_INFO and the query string.
E.g, /app/blog?id=10
)r r9 )rJ r% r% r& �path_qs_ s zBaseRequest.path_qsc C s
| j j S)z� The URL including raw *PATH INFO* without the host or scheme.
Warning, the path is unquoted and may contains non valid URL characters
E.g., ``/my%2Fpath%7Cwith%21some%25strange%24characters``
)r2 rQ )rJ r% r% r& �raw_pathg s zBaseRequest.raw_pathc C s
| j j S)z7A multidict with all the variables in the query string.)r9 �query)rJ r% r% r& r� p s zBaseRequest.queryc C s
| j j S)z:The query string in the URL.
E.g., id=10
)r9 � query_string)rJ r% r% r& r� u s zBaseRequest.query_stringc C s | j S)z4A case-insensitive multidict proxy with all headers.)r6 )rJ r% r% r& r$ } s zBaseRequest.headersc C s
| j j S)z#A sequence of pars for all headers.)r2 rU )rJ r% r% r& rU � s zBaseRequest.raw_headersc C s] | j j | � } | d k rY t | � } | d k rY t j | d d � d t j j � Sd S)zyThe value of If-Modified-Since HTTP header, or None.
This header is represented as a `datetime` object.
N� Ztzinfo)r$ r r �datetime�timezoneZutc)rJ Z_IF_MODIFIED_SINCEZhttpdateZ timetupler% r% r& �if_modified_since� s
z BaseRequest.if_modified_sincec C s | j j S)z Is keepalive enabled by client?)r2 Z should_close)rJ r% r% r& �
keep_alive� s zBaseRequest.keep_alivec C sA | j j t j d � } t | � } t d d � | j � D� � S)zMReturn request cookies.
A read-only dictionary-like object.
r' c S s" i | ] \ } } | j | � q Sr% )rc )r) ra �valr% r% r& �
<dictcomp>� s z'BaseRequest.cookies.<locals>.<dictcomp>)r$ r r ZCOOKIEr r rX )rJ �rawZparsedr% r% r& �cookies� s zBaseRequest.cookies�_RANGEc C s2 | j j | � } d \ } } | d k r"y& d } t j | | � d \ } } Wn t k
rp t d � � Yn X| r� t | � n d } | r� t | � n d } | d k r� | d k r� | } | d k r� | d k r� | d 7} | | k r� t d � � | | k od k n r"t d � � t | | d � S) zMThe content of Range HTTP header.
Return a slice instance.
Nz^bytes=(\d*)-(\d*)$r z range not in acceptible formatr zstart cannot be after endz"No start or end of range specified)NN)r6 r �re�findall�
IndexError�
ValueError�int�slice)rJ r� �rng�start�end�patternr% r% r& �
http_range� s&
zBaseRequest.http_rangec C s | j S)zReturn raw payload stream.)r5 )rJ r% r% r& �content� s zBaseRequest.contentc C s$ t j d t d d �| j j � S)z@Return True if request's HTTP BODY can be read, False otherwise.z$Deprecated, use .can_read_body #2005�
stacklevelrB )�warnings�warn�DeprecationWarningr5 �at_eof)rJ r% r% r& �has_body� s
zBaseRequest.has_bodyc C s | j j � S)z@Return True if request's HTTP BODY can be read, False otherwise.)r5 r� )rJ r% r% r& �
can_read_body� s zBaseRequest.can_read_bodyc C s t | j � t k S)z6Return True if request has HTTP BODY, False otherwise.)r r5 r )rJ r% r% r& � body_exists� s zBaseRequest.body_existsc � s, x% | j j � s' | j j � Id Hq Wd S)zKRelease request.
Eat unread part of HTTP BODY if present.
N)r5 r� �readany)rJ r% r% r& �release� s zBaseRequest.releasec � s� | j d k rz t � } xP | j j � Id H} | j | � | j r` t | � | j k r` t � | s Pq Wt | � | _ | j S)z_Read request body if present.
Returns bytes object with full request content.
N) r; � bytearrayr5 r� �extendr? rf r �bytes)rJ �body�chunkr% r% r& �read� s
zBaseRequest.readc � s- | j � Id H} | j p d } | j | � S)z1Return BODY as text using encoding from .charset.Nzutf-8)r� �charset�decode)rJ Z
bytes_body�encodingr% r% r& �text� s zBaseRequest.text�loadsc � s | j � Id H} | | � S)zReturn BODY as JSON.N)r� )rJ r� r� r% r% r& �json� s zBaseRequest.json�readerc � s | | j | j � S)z3Return async iterator to process BODY as multipart.)r6 r5 )rJ r� r% r% r& r
s zBaseRequest.multipartc
� s� | j d k r | j S| j | j k rA t t � � | _ | j S| j } | d k ro t t � � | _ | j St � } | d k rw| j � Id H} | j � Id H} x"| d k rsd } | j } | j j
t j � } | j
r�t j � } | j d d � Id H} xu | r|| j | � } | j | � | t | � 7} d | k oQ| k n rbt d � � | j d d � Id H} qW| j d � t | j | j
| | | j � } | j | j | � n� | j d
d � Id H}
| d k s�| j d � r| j d
d � } |
j | � }
| j | j |
� | t |
� 7} d | k oN| k n r_t d � � | j � Id H} q� WnT | j � Id H} | r�| j p�d } | j t | j � j | � d d d | �� t | � | _ | j S)zReturn POST parameters.Nr' �!application/x-www-form-urlencoded�multipart/form-datar �sizerB � z"Maximum request body size exceededr� Tztext/�defaultzutf-8Zkeep_blank_valuesr� )r' r� r� i i ) r: r7 � POST_METHODSr r r# r
�nextr? r$ r r Z CONTENT_TYPEr �tempfile�
TemporaryFileZ
read_chunkr� �writerf r� �seekr r �addr� �
startswithZ get_charsetr� r� r �rstrip)
rJ r# �outr
Zfieldr� �max_size�tmpr� Zffrc r� �datar% r% r& �post sn
zBaseRequest.postc C s: | j j d d � j d � } d j | j j | j | � S)N�ascii�backslashreplacez <{} {} {} >)rQ rR r� �formatrZ r r7 )rJ Zascii_encodable_pathr% r% r& �__repr__L s zBaseRequest.__repr__c c s d Sd V)Nr% )rJ �responser% r% r& �
_prepare_hookR s zBaseRequest._prepare_hooki )Cr r r r Z
METH_PATCHZ METH_POSTZMETH_PUTZ
METH_TRACEZ METH_DELETEr� r �ATTRS� frozensetrO r r^ �propertyrM rL r_ r` rK rP rN rb rd re rg ri rk r r} rD rG rH rE rF rI rQ r� r� r� r� r$ rU ZIF_MODIFIED_SINCEr� r� r� �RANGEr� r� r� r� r� r� r� r� r� r� r
ZMultipartReaderr� r� �asyncio� coroutiner� r% r% r% r& r E st . <
! E c s� e Z d Z e j e d g � BZ � f d d � Z � f d d � Z d e d e d e d e d
e d e � f d d
� Z e
d d � � Z e
d d � � Z d d � Z
� S)r � _match_infoc s t � j | | � d | _ d S)N)�superrO r� )rJ �argsr] )rZ r% r& rO \ s zRequest.__init__c sN | | j k r7 t j d j | j j | � t d d �t � j | | � d S)Nz-Setting custom {}.{} attribute is discouragedr� rB ) r� r� r� r� rZ r r� r� � __setattr__)rJ r r� )rZ r% r& r� c s zRequest.__setattr__rG rP r$ rD rE rF c
sC t � j d | d | d | d | d | d | � } | j | _ | S)NrG rP r$ rD rE rF )r� r^ r� )rJ rG rP r$ rD rE rF Zret)rZ r% r& r^ l s z
Request.clonec C s | j S)zResult of route resolving.)r� )rJ r% r% r& �
match_infox s zRequest.match_infoc C s
| j j S)zApplication instance.)r� Z current_app)rJ r% r% r& �app} s z Request.appc � sI | j } | d k r d Sx) | j D] } | j j | | � Id Hq# Wd S)N)r� ZappsZon_response_prepare�send)rJ r� r� r� r% r% r& r� � s
zRequest._prepare_hook)r r r r r� r� rO r� r r^ r� r� r� r� r% r% )rZ r& r X s
):r� � collectionsr� r r� r� r� �stringr� rq r� Z email.utilsr Z http.cookiesr r Z urllib.parser r Z multidictr r r r Zyarlr
r' r r
Zhelpersr r r Zstreamsr Zweb_exceptionsr �__all__�sr �digits�
ascii_lettersZ_TCHARr� Z_TOKENr� rW �rangeZ_QDTEXTZ _QUOTED_PAIRZ_QUOTED_STRINGZ_FORWARDED_PAIR�compilerv rr �MutableMappingr r r% r% r% r& �<module> sN " + ; � �