U
    ddȦ                     @   sx  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mZ ddlmZmZmZmZ ddlmZmZmZmZmZmZmZmZ ddlmZmZmZmZm Z m!Z!m"Z" ddl#m$Z$ dd	l%m&Z& dd
l'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2 ddl3m4Z4 ddl5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z< G dd dej=e>e>f Z?G dd dZ@G dd dZAG dd dej=e>e>f ZBdS )    N)Mapping)Cookie	CookieJar   )
ByteStreamUnattachedStreamencode_requestencode_response)SUPPORTED_DECODERSByteChunkerContentDecoderIdentityDecoderLineDecoderMultiDecoderTextChunkerTextDecoder)CookieConflictHTTPStatusErrorRequestNotReadResponseNotReadStreamClosedStreamConsumedrequest_context)(get_multipart_boundary_from_content_type)codes)AsyncByteStreamCookieTypesHeaderTypesQueryParamTypesRequestContentRequestDataRequestExtensionsRequestFilesResponseContentResponseExtensionsSyncByteStream)URL)guess_json_utfis_known_encodingnormalize_header_keynormalize_header_valueobfuscate_sensitive_headersparse_content_type_charsetparse_header_linksc                   @   s  e Zd ZdZd4eje eje ddddZe	edddZ
e
jedd	d
dZ
e	ejejeef  dddZeje dddZeje dddZejeef dddZejejeef  dddZd5eejejdddZd6eeeje dddZd7eje ddddZd ddd Zeed!d"d#Zeedd$d%d&Zedd!d'd(Zejed!d)d*Z ej!ej dd+d,Z"e#dd-d.Z$ejed/d0d1Z%edd2d3Z&dS )8Headersz9
    HTTP headers, as a case-insensitive multi-dict.
    N)headersencodingreturnc                    sj   |d krg | _ nPt|tr(t|j | _ n8t|trL fdd| D | _ n fdd|D | _  | _d S )Nc                    s4   g | ],\}}t |d  dt |d dt| fqS F)lowerr0   Tr)   r*   .0kvr0    1/tmp/pip-unpacked-wheel-fkuvgui5/httpx/_models.py
<listcomp>G   s
   z$Headers.__init__.<locals>.<listcomp>c                    s4   g | ],\}}t |d  dt |d dt| fqS r2   r4   r5   r9   r:   r;   r<   P   s
   )_list
isinstancer.   listr   items	_encoding)selfr/   r0   r:   r9   r;   __init__=   s    



	
	zHeaders.__init__r1   c              
   C   sn   | j dkrhdD ]R}| jD ]<\}}z|| || W q tk
rR   Y  qY qX q|| _  qhqd| _ | j S )zn
        Header encoding is mandated as ascii, but we allow fallbacks to utf-8
        or iso-8859-1.
        N)asciiutf-8z
iso-8859-1)rA   rawdecodeUnicodeDecodeError)rB   r0   keyvaluer:   r:   r;   r0   [   s    

zHeaders.encodingrK   r1   c                 C   s
   || _ d S NrA   rB   rK   r:   r:   r;   r0   t   s    c                 C   s   dd | j D S )zH
        Returns a list of the raw header items, as byte pairs.
        c                 S   s   g | ]\}}}||fqS r:   r:   )r6   Zraw_key_rK   r:   r:   r;   r<   }   s     zHeaders.raw.<locals>.<listcomp>r=   rB   r:   r:   r;   rG   x   s    zHeaders.rawc                    s    fdd j D  S )Nc                    s    i | ]\}}}|  jd qS rM   rH   r0   r6   rP   rJ   rK   rR   r:   r;   
<dictcomp>   s     
 z Headers.keys.<locals>.<dictcomp>)r=   keysrR   r:   rR   r;   rV      s    zHeaders.keysc                 C   s^   i }| j D ]J\}}}|| j}|| j}||krL||  d| 7  < q
|||< q
| S )N, )r=   rH   r0   valuesrB   Zvalues_dictrP   rJ   rK   Zstr_keyZ	str_valuer:   r:   r;   rX      s    
zHeaders.valuesc                 C   s^   i }| j D ]J\}}}|| j}|| j}||krL||  d| 7  < q
|||< q
| S )z
        Return `(key, value)` items of headers. Concatenate headers
        into a single comma separated value when a key occurs multiple times.
        rW   )r=   rH   r0   r@   rY   r:   r:   r;   r@      s    
zHeaders.itemsc                    s    fdd j D S )z
        Return a list of `(key, value)` pairs of headers. Allow multiple
        occurrences of the same key without concatenating into a single
        comma separated value.
        c                    s*   g | ]"\}}}|  j|  jfqS r:   rS   rT   rR   r:   r;   r<      s   z'Headers.multi_items.<locals>.<listcomp>rQ   rR   r:   rR   r;   multi_items   s    
zHeaders.multi_items)rJ   defaultr1   c                 C   s(   z
| | W S  t k
r"   | Y S X dS )z
        Return a header value. If multiple occurrences of the header occur
        then concatenate them together with commas.
        N)KeyError)rB   rJ   r[   r:   r:   r;   get   s    
zHeaders.getF)rJ   split_commasr1   c                    sZ   |  j  fddjD }|s.|S g }|D ]}|dd |dD  q6|S )z
        Return a list of all header values for a given key.
        If `split_commas=True` is passed, then any comma separated header
        values are split into multiple return strings.
        c                    s*   g | ]"\}}}|   kr|jqS r:   )r3   rH   r0   )r6   rP   item_keyZ
item_valueZget_header_keyrB   r:   r;   r<      s   z$Headers.get_list.<locals>.<listcomp>c                 S   s   g | ]}|  qS r:   )strip)r6   itemr:   r:   r;   r<      s     ,)r3   encoder0   r=   extendsplit)rB   rJ   r^   rX   Zsplit_valuesrK   r:   r`   r;   get_list   s    zHeaders.get_list)r/   r1   c                 C   s:   t |}| D ]}|| kr| | q| j|j d S rM   )r.   rV   popr=   re   )rB   r/   rJ   r:   r:   r;   update   s
    zHeaders.updatec                 C   s   t | | jdS )Nr9   )r.   r0   rR   r:   r:   r;   copy   s    zHeaders.copy)rJ   r1   c                    s@   |  j  fddjD }|r4d|S t|dS )z
        Return a single header value.

        If there are multiple headers with the same key, then we concatenate
        them with commas. See: https://tools.ietf.org/html/rfc7230#section-3.2.2
        c                    s&   g | ]\}}}| kr| jqS r:   rS   )r6   rP   
header_keyheader_valueZnormalized_keyrB   r:   r;   r<      s   z'Headers.__getitem__.<locals>.<listcomp>rW   N)r3   rd   r0   r=   joinr\   )rB   rJ   r@   r:   rm   r;   __getitem__   s    
zHeaders.__getitem__rJ   rK   r1   c                    s   | | jpd}| | jpd}|   fddt| jD }t|dd D ]}| j|= qP|r||d }| |f| j|< n| j| |f dS )zs
        Set the header `key` to `value`, removing any duplicate entries.
        Retains insertion order.
        rF   c                    s"   g | ]\}\}}}| kr|qS r:   r:   r6   idxrP   r_   Z
lookup_keyr:   r;   r<      s   z'Headers.__setitem__.<locals>.<listcomp>r   Nr   )rd   rA   r3   	enumerater=   reversedappend)rB   rJ   rK   Zset_key	set_valueZfound_indexesrr   r:   rs   r;   __setitem__   s    

zHeaders.__setitem__c                    sN   |  | j  fddt| jD }|s4t|t|D ]}| j|= q<dS )z*
        Remove the header `key`.
        c                    s&   g | ]\}\}}}|   kr|qS r:   )r3   rq   Zdel_keyr:   r;   r<     s   z'Headers.__delitem__.<locals>.<listcomp>N)r3   rd   r0   rt   r=   r\   ru   )rB   rJ   Zpop_indexesrr   r:   ry   r;   __delitem__   s    
zHeaders.__delitem__c                 C   s$   |  | j}|dd | jD kS )Nc                 S   s   g | ]\}}}|qS r:   r:   )r6   rP   rJ   r:   r:   r;   r<     s     z(Headers.__contains__.<locals>.<listcomp>)r3   rd   r0   r=   )rB   rJ   rk   r:   r:   r;   __contains__  s    zHeaders.__contains__c                 C   s   t |  S rM   )iterrV   rR   r:   r:   r;   __iter__  s    zHeaders.__iter__c                 C   s
   t | jS rM   )lenr=   rR   r:   r:   r;   __len__  s    zHeaders.__len__)otherr1   c                 C   sT   zt |}W n tk
r"   Y dS X dd | jD }dd |jD }t|t|kS )NFc                 S   s   g | ]\}}}||fqS r:   r:   rT   r:   r:   r;   r<     s     z"Headers.__eq__.<locals>.<listcomp>c                 S   s   g | ]\}}}||fqS r:   r:   rT   r:   r:   r;   r<      s     )r.   
ValueErrorr=   sorted)rB   r   Zother_headersZ	self_listZ
other_listr:   r:   r;   __eq__  s    zHeaders.__eq__c                 C   sv   | j j}d}| jdkr"d| j}tt|  }t|}t|t|k}|rb| d|| dS | d|| dS )N rE   z, encoding=())	__class____name__r0   r?   r+   rZ   dictr~   )rB   
class_nameZencoding_strZas_listZas_dictZno_duplicate_keysr:   r:   r;   __repr__#  s    
zHeaders.__repr__)NN)N)F)N)'r   
__module____qualname____doc__typingOptionalr   strrC   propertyr0   setterListTuplebytesrG   KeysViewrV   
ValuesViewrX   	ItemsViewr@   rZ   Anyr]   boolrg   ri   rj   ro   rx   rz   r{   Iteratorr}   intr   r   r   r:   r:   r:   r;   r.   8   s<      

r.   c                   @   s"  e Zd Zdddddddddd	ejeef ejdef eje eje	 eje
 eje eje eje ejej ejeedf eje dddZejeef dddd	Zeed
ddZed
ddZed
ddZed
ddZejeejf d
ddZejeejf ddddZdS )RequestN)	paramsr/   cookiescontentdatafilesjsonstream
extensionsr&   )methodurlr   r/   r   r   r   r   r   r   r   c       	      	   C   s   t |tr|d n| | _t|| _|d k	rD| jj|d| _t|| _	|d krZi n|| _
|rrt||  |
d kr| j	d}t||||	t|r|| j	jnd dd\}}
| | |
| _t |
tr|   n|
| _d S )NrE   )r   zcontent-type)content_type)r   r   r   r   boundary)r>   r   rH   upperr   r&   r   Zcopy_merge_paramsr.   r/   r   Cookiesset_cookie_headerr]   r   r   rd   r0   _preparer   r   read)rB   r   r   r   r/   r   r   r   r   r   r   r   r   r:   r:   r;   rC   4  s:    





zRequest.__init__default_headersr1   c                 C   s   |  D ].\}}| dkr(d| jkr(q| j|| qg }d| jk}d| jkpXd| jk}|sx| jjrx|d| jjf |s| jdkr|d t	|| jj
 | _d S )Ntransfer-encodingzContent-LengthHostzTransfer-Encodings   Host)POSTPUTPATCH)s   Content-Length   0)r@   r3   r/   
setdefaultr   hostrv   netlocr   r.   rG   )rB   r   rJ   rK   Zauto_headersZhas_hostZhas_content_lengthr:   r:   r;   r   s  s    

zRequest._preparerD   c                 C   s   t | dst | jS N_content)hasattrr   r   rR   r:   r:   r;   r     s    
zRequest.contentc                 C   sH   t | dsBt| jtjstd| j| _t| jtsBt| j| _| jS )6
        Read and return the request content.
        r       )	r   r>   r   r   IterableAssertionErrorrn   r   r   rR   r:   r:   r;   r     s    
zRequest.readc                    sX   t | dsRt| jtjstddd | j2 I dH | _t| jtsRt| j| _| jS )r   r   r   c                    s   g | z3 d H W }|q6 S rM   r:   r6   partr:   r:   r;   r<     s     z!Request.aread.<locals>.<listcomp>N)	r   r>   r   r   AsyncIterabler   rn   r   r   rR   r:   r:   r;   aread  s    
zRequest.areadc                 C   s,   | j j}t| j}d| d| jd|dS )N<r   rW   z)>)r   r   r   r   r   )rB   r   r   r:   r:   r;   r     s    
zRequest.__repr__c                 C   s   dd | j  D S )Nc                 S   s   i | ]\}}|d kr||qS ))r   r   r:   r6   namerK   r:   r:   r;   rU     s    z(Request.__getstate__.<locals>.<dictcomp>__dict__r@   rR   r:   r:   r;   __getstate__  s    zRequest.__getstate__stater1   c                 C   s0   |  D ]\}}t| || qi | _t | _d S rM   )r@   setattrr   r   r   rB   r   r   rK   r:   r:   r;   __setstate__  s    zRequest.__setstate__)r   r   r   r   Unionr   r   r   r   r   r   r   r    r"   r   r%   r   r!   rC   Dictr   r   r   r   r   r   r   r   r:   r:   r:   r;   r   3  s<   
?r   c                   @   s  e Zd Zddddddddddd
eeje eje eje eje ej	ej
eedf eje eje ejejd   ej
eejegef f dddZejeef dddd	Zeejd
ddZejejddddZeed
ddZejeddddZeed
ddZeed
ddZeed
ddZeed
ddZeed
ddZ eeje d
ddZ!e!jeddddZ!eeje d
d d!Z"e#d
d"d#Z$ee%d
d$d%Z&ee%d
d&d'Z'ee%d
d(d)Z(ee%d
d*d+Z)ee%d
d,d-Z*ee%d
d.d/Z+ee%d
d0d1Z,dd
d2d3Z-ej	ej	d4d5d6Z.ed7d
d8d9Z/eejeje ejeef f d
d:d;Z0eed
d<d=Z1ed
d>d?Z2ejeej	f d
d@dAZ3ejeej	f ddBdCdDZ4ed
dEdFZ5d^eje ej6e dGdHdIZ7d_eje ej6e dGdJdKZ8ej6e d
dLdMZ9d`eje ej6e dGdNdOZ:dd
dPdQZ;ed
dRdSZ<daeje ej=e dGdTdUZ>dbeje ej=e dGdVdWZ?ej=e d
dXdYZ@dceje ej=e dGdZd[ZAdd
d\d]ZBdS )dResponseNrF   )
r/   r   texthtmlr   r   requestr   historydefault_encoding)status_coder/   r   r   r   r   r   r   r   r   r   c       
         C   s   || _ t|| _|| _d | _|	d kr(i n|	| _|
d kr:g nt|
| _d| _d| _	|| _
|d krt||||\}}| | || _t|tr|   n|| _d| _d S )NFr   )r   r.   r/   _requestZnext_requestr   r?   r   	is_closedis_stream_consumedr   r	   r   r   r>   r   r   _num_bytes_downloaded)rB   r   r/   r   r   r   r   r   r   r   r   r   r:   r:   r;   rC     s"    



zResponse.__init__r   c                 C   s<   |  D ].\}}| dkr(d| jkr(q| j|| qd S )Nr   zcontent-length)r@   r3   r/   r   )rB   r   rJ   rK   r:   r:   r;   r     s    zResponse._preparerD   c                 C   s   t | dstd| jS )ze
        Returns the time taken for the complete request/response
        cycle to complete.
        _elapsedzK'.elapsed' may only be accessed after the response has been read or closed.)r   RuntimeErrorr   rR   r:   r:   r;   elapsed  s
    
zResponse.elapsed)r   r1   c                 C   s
   || _ d S rM   )r   )rB   r   r:   r:   r;   r   	  s    c                 C   s   | j dkrtd| j S )zR
        Returns the request instance associated to the current response.
        Nz7The request instance has not been set on this response.)r   r   rR   r:   r:   r;   r     s
    
zResponse.requestrL   c                 C   s
   || _ d S rM   )r   rO   r:   r:   r;   r     s    c                 C   s8   z| j d }W n tk
r$   Y dS X |jdddS d S )Nhttp_versionzHTTP/1.1rE   ignoreerrors)r   r\   rH   )rB   r   r:   r:   r;   r     s
    zResponse.http_versionc                 C   sB   z| j d }W n  tk
r.   t| j Y S X |jdddS d S )Nreason_phraserE   r   r   )r   r\   r   Zget_reason_phraser   rH   )rB   r   r:   r:   r;   r   %  s
    zResponse.reason_phrasec                 C   s   | j jS )zA
        Returns the URL for which the request was made.
        )r   r   rR   r:   r:   r;   r   .  s    zResponse.urlc                 C   s   t | dst | jS r   )r   r   r   rR   r:   r:   r;   r   5  s    
zResponse.contentc                 C   sN   t | dsH| j}|sd| _n,t| jp&dd}d|| j| g| _| jS )N_textr   rF   r9   )r   r   r   r   r0   rn   rH   flush)rB   r   decoderr:   r:   r;   r   ;  s    
zResponse.textc                 C   sZ   t | dsT| j}|dks t|sJt| jtr4| j}nt | drJ| | j}|pPd| _| jS )a  
        Return an encoding to use for decoding the byte content into text.
        The priority for determining this is given by...

        * `.encoding = <>` has been set explicitly.
        * The encoding as specified by the charset parameter in the Content-Type header.
        * The encoding as determined by `default_encoding`, which may either be
          a string like "utf-8" indicating the encoding to use, or may be a callable
          which enables charset autodetection.
        rA   Nr   rF   )r   charset_encodingr(   r>   r   r   r   rA   )rB   r0   r:   r:   r;   r0   F  s    


zResponse.encodingc                 C   s
   || _ d S rM   rN   rO   r:   r:   r;   r0   \  s    c                 C   s    | j d}|dkrdS t|S )zO
        Return the encoding, as specified by the Content-Type header.
        zContent-TypeN)r/   r]   r,   )rB   r   r:   r:   r;   r   `  s    zResponse.charset_encodingc              	   C   s   t | dsg }| jjddd}|D ]B}|  }zt| }||  W q" tk
rb   Y q"Y q"X q"t|dkr~|d | _	n"t|dkrt
|d| _	nt | _	| j	S )z
        Returns a decoder instance which can be used to decode the raw byte
        content, depending on the Content-Encoding used in the response.
        _decoderzcontent-encodingT)r^   r   r   )children)r   r/   rg   ra   r3   r
   rv   r\   r~   r   r   r   )rB   ZdecodersrX   rK   Zdecoder_clsr:   r:   r;   _get_content_decoderk  s     
zResponse._get_content_decoderc                 C   s   t | jS )zU
        A property which is `True` for 1xx status codes, `False` otherwise.
        )r   is_informationalr   rR   r:   r:   r;   r     s    zResponse.is_informationalc                 C   s   t | jS )zU
        A property which is `True` for 2xx status codes, `False` otherwise.
        )r   
is_successr   rR   r:   r:   r;   r     s    zResponse.is_successc                 C   s   t | jS )a  
        A property which is `True` for 3xx status codes, `False` otherwise.

        Note that not all responses with a 3xx status code indicate a URL redirect.

        Use `response.has_redirect_location` to determine responses with a properly
        formed URL redirection.
        )r   is_redirectr   rR   r:   r:   r;   r     s    
zResponse.is_redirectc                 C   s   t | jS )zU
        A property which is `True` for 4xx status codes, `False` otherwise.
        )r   is_client_errorr   rR   r:   r:   r;   r     s    zResponse.is_client_errorc                 C   s   t | jS )zU
        A property which is `True` for 5xx status codes, `False` otherwise.
        )r   is_server_errorr   rR   r:   r:   r;   r     s    zResponse.is_server_errorc                 C   s   t | jS )z]
        A property which is `True` for 4xx and 5xx status codes, `False` otherwise.
        )r   is_errorr   rR   r:   r:   r;   r     s    zResponse.is_errorc                 C   s(   | j tjtjtjtjtjfko&d| jkS )zs
        Returns True for 3xx responses with a properly formed URL redirection,
        `False` otherwise.
        ZLocation)r   r   MOVED_PERMANENTLYFOUND	SEE_OTHERTEMPORARY_REDIRECTPERMANENT_REDIRECTr/   rR   r:   r:   r;   has_redirect_location  s    zResponse.has_redirect_locationc                 C   st   | j }|dkrtd| jr dS | jr,d}nd}| jd }dddd	d
}||d}|j| |d}t||| ddS )z>
        Raise the `HTTPStatusError` if one occurred.
        NzYCannot call `raise_for_status` as the request instance has not been set on this response.z{error_type} '{0.status_code} {0.reason_phrase}' for url '{0.url}'
Redirect location: '{0.headers[location]}'
For more information check: https://httpstatuses.com/{0.status_code}z{error_type} '{0.status_code} {0.reason_phrase}' for url '{0.url}'
For more information check: https://httpstatuses.com/{0.status_code}d   zInformational responsezRedirect responsezClient errorzServer error)r            zInvalid status code)
error_type)r   response)r   r   r   r   r   r]   formatr   )rB   r   messageZstatus_classZerror_typesr   r:   r:   r;   raise_for_status  s*    
zResponse.raise_for_status)kwargsr1   c                 K   sV   | j d krF| jrFt| jdkrFt| j}|d k	rFtj| j|f|S tj| jf|S )Nr   )r   r   r~   r'   jsonlibloadsrH   r   )rB   r  r0   r:   r:   r;   r     s
    
zResponse.jsonr   c                 C   s$   t | dst | _| j|  | jS )N_cookies)r   r   r  extract_cookiesrR   r:   r:   r;   r     s    
zResponse.cookiesc                 C   sF   | j d}i }|rBt|}|D ] }|dp6|d}|||< q |S )zI
        Returns the parsed header links of the response, if any
        linkrelr   )r/   r]   r-   )rB   headerZldictlinksr  rJ   r:   r:   r;   r
    s    
zResponse.linksc                 C   s   | j S rM   )r   rR   r:   r:   r;   num_bytes_downloaded  s    zResponse.num_bytes_downloadedc                 C   s   d| j  d| j dS )Nz<Response [ ]>)r   r   rR   r:   r:   r;   r     s    zResponse.__repr__c                 C   s   dd | j  D S )Nc                 S   s   i | ]\}}|d kr||qS ))r   r   r   r   r:   r   r:   r:   r;   rU     s    z)Response.__getstate__.<locals>.<dictcomp>r   rR   r:   r:   r;   r     s    zResponse.__getstate__r   c                 C   s6   |  D ]\}}t| || qd| _i | _t | _d S )NT)r@   r   r   r   r   r   r   r:   r:   r;   r     s
    zResponse.__setstate__c                 C   s    t | dsd|  | _| jS )7
        Read and return the response content.
        r   r   )r   rn   
iter_bytesr   rR   r:   r:   r;   r      s    
zResponse.read)
chunk_sizer1   c              	   c   s   t | drT|dkrt| jn|}tdt| jt|dD ]}| j|||  V  q8n|  }t|d}t| jdf | 	 D ]$}|
|}|
|D ]
}|V  qq|| }|
|D ]
}|V  q| D ]
}|V  qW 5 Q R X dS z
        A byte-iterator over the decoded response content.
        This allows us to handle gzip, deflate, and brotli encoded responses.
        r   Nr   r   r  r   )r   r~   r   rangemaxr   r   r   r   iter_rawrH   r   rB   r  ir   chunkerZ	raw_bytesdecodedchunkr:   r:   r;   r  (  s     



zResponse.iter_bytesc              	   c   s   t | jp
dd}t|d}t| jdf |  D ]$}||}||D ]
}|V  qHq0| }||D ]
}|V  qh| D ]
}|V  q|W 5 Q R X dS z
        A str-iterator over the decoded response content
        that handles both gzip, deflate, etc but also detects the content's
        string encoding.
        rF   r9   r  r  N)r   r0   r   r   r   r  rH   r   rB   r  r   r  Zbyte_contentZtext_contentr  r:   r:   r;   	iter_textA  s    


zResponse.iter_textc              	   c   sZ   t  }t| jd> |  D ]}||D ]
}|V  q*q| D ]
}|V  q@W 5 Q R X d S Nr  )r   r   r   r  rH   r   rB   r   r   liner:   r:   r;   
iter_linesV  s    
zResponse.iter_linesc              	   c   s   | j rt | jrt t| jts,tdd| _ d| _t	|d}t
| jd: | jD ],}|  jt|7  _||D ]
}|V  qvqVW 5 Q R X | D ]
}|V  q|   dS )@
        A byte-iterator over the raw response content.
        z5Attempted to call a sync iterator on an async stream.Tr   r  r  N)r   r   r   r   r>   r   r%   r   r   r   r   r   r~   rH   r   closerB   r  r  Zraw_stream_bytesr  r:   r:   r;   r  _  s"    

zResponse.iter_rawc              	   C   sF   t | jtstd| jsBd| _t| jd | j  W 5 Q R X dS )
        Close the response and release the connection.
        Automatically called if the response body is read to completion.
        z3Attempted to call an sync close on an async stream.Tr  N)r>   r   r%   r   r   r   r   r$  rR   r:   r:   r;   r$  {  s    zResponse.closec                    s0   t | ds*ddd |  2 I dH | _| jS )r  r   r   c                    s   g | z3 d H W }|q6 S rM   r:   r   r:   r:   r;   r<     s     z"Response.aread.<locals>.<listcomp>N)r   rn   aiter_bytesr   rR   r:   r:   r;   r     s    
 zResponse.areadc              	   C  s   t | drT|dkrt| jn|}tdt| jt|dD ]}| j|||  V  q8n|  }t|d}t| jdp | 	 2 z,3 dH W }|
|}|
|D ]
}|V  qq|6 | }|
|D ]
}|V  q| D ]
}|V  qW 5 Q R X dS r  )r   r~   r   r  r  r   r   r   r   	aiter_rawrH   r   r  r:   r:   r;   r'    s     


zResponse.aiter_bytesc              	   C  s   t | jp
dd}t|d}t| jdp |  2 z,3 dH W }||}||D ]
}|V  qPq06 | }||D ]
}|V  qr| D ]
}|V  qW 5 Q R X dS r  )r   r0   r   r   r   r'  rH   r   r  r:   r:   r;   
aiter_text  s    

zResponse.aiter_textc              	   C  sd   t  }t| jdH |  2 z"3 d H W }||D ]
}|V  q2q6 | D ]
}|V  qJW 5 Q R X d S r  )r   r   r   r)  rH   r   r   r:   r:   r;   aiter_lines  s    zResponse.aiter_linesc              	   C  s   | j rt | jrt t| jts,tdd| _ d| _t	|d}t
| jdD | j2 z43 dH W }|  jt|7  _||D ]
}|V  q~qV6 W 5 Q R X | D ]
}|V  q|  I dH  dS )r#  z6Attempted to call an async iterator on an sync stream.Tr   r  r  N)r   r   r   r   r>   r   r   r   r   r   r   r   r~   rH   r   acloser%  r:   r:   r;   r(    s"    
zResponse.aiter_rawc              	      sL   t | jtstd| jsHd| _t| jd | j I dH  W 5 Q R X dS )r&  z3Attempted to call an async close on an sync stream.Tr  N)r>   r   r   r   r   r   r   r+  rR   r:   r:   r;   r+    s    zResponse.aclose)N)N)N)N)N)N)Cr   r   r   r   r   r   r   r#   r   r   r   r%   r   r   r$   r   Callabler   rC   r   r   r   datetime	timedeltar   r   r   r   r   r&   r   r   r   r0   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   AsyncIteratorr'  r)  r*  r(  r+  r:   r:   r:   r;   r     s   7


%*	  
 	  
 r   c                   @   s|  e Zd ZdZd1eje ddddZeddddZ	e
dd	d
dZd2eeeeddddZd3eeje eje eje eje dddZd4eeje eje ddddZd5eje eje ddddZd6eje ddddZeeddddZeedd d!Zeddd"d#Zed$d%d&Zeje d$d'd(Zed$d)d*Zed$d+d,ZG d-d. d.ejj
ZG d/d0 d0ZdS )7r   z-
    HTTP Cookies, as a mutable mapping.
    N)r   r1   c                 C   s   |d kst |trDt | _t |tr| D ]\}}| || q,n`t |trrt | _|D ]\}}| || qZn2t |trt | _|jD ]}| j| qn|| _d S rM   )	r>   r   r   jarr@   setr?   r   
set_cookie)rB   r   rJ   rK   cookier:   r:   r;   rC     s    



zCookies.__init__)r   r1   c                 C   s(   |  |}| |j}| j|| dS )zO
        Loads any cookies based on the response `Set-Cookie` headers.
        N)_CookieCompatResponse_CookieCompatRequestr   r0  r  )rB   r   Zurllib_responseurllib_requestr:   r:   r;   r    s    
zCookies.extract_cookiesr   r1   c                 C   s   |  |}| j| dS )zM
        Sets an appropriate 'Cookie:' HTTP header on the `Request`.
        N)r5  r0  add_cookie_header)rB   r   r6  r:   r:   r;   r     s    
zCookies.set_cookie_headerr   /)r   rK   domainpathr1   c                 C   sT   d||dd|t ||d|t |dddddddidd}tf |}| j| dS )zU
        Set a cookie value by name. May optionally include domain and path.
        r   NF.THttpOnly)versionr   rK   portport_specifiedr:  domain_specifieddomain_initial_dotr;  path_specifiedsecureexpiresdiscardcommentcomment_urlrestrfc2109)r   
startswithr   r0  r2  )rB   r   rK   r:  r;  r  r3  r:   r:   r;   r1    s(    
zCookies.set)r   r[   r:  r;  r1   c                 C   sn   d}| j D ]R}|j|kr
|dks*|j|kr
|dks<|j|kr
|dk	rVd| }t||j}q
|dkrj|S |S )z
        Get a cookie by name. May optionally include domain and path
        in order to specify exactly which cookie to retrieve.
        Nz!Multiple cookies exist with name=)r0  r   r:  r;  r   rK   )rB   r   r[   r:  r;  rK   r3  r   r:   r:   r;   r]   1  s    


zCookies.get)r   r:  r;  r1   c                    s\    dk	r dk	r | j  S  fdd| j D }|D ]}| j |j|j|j q<dS )z
        Delete a cookie by name. May optionally include domain and path
        in order to specify exactly which cookie to delete.
        Nc                    s>   g | ]6}|j kr d ks$|j krd ks6|jkr|qS rM   )r   r:  r;  r6   r3  r:  r   r;  r:   r;   r<   W  s   
 
 
z"Cookies.delete.<locals>.<listcomp>)r0  clearr:  r;  r   )rB   r   r:  r;  remover3  r:   rM  r;   deleteJ  s    
zCookies.delete)r:  r;  r1   c                 C   sD   g }|dk	r| | |dk	r4|dk	s*t| | | jj|  dS )z
        Delete all cookies. Optionally include a domain and path in
        order to only delete a subset of all the cookies.
        N)rv   r   r0  rN  )rB   r:  r;  argsr:   r:   r;   rN  b  s    

zCookies.clearc                 C   s$   t |}|jD ]}| j| qd S rM   )r   r0  r2  )rB   r   r3  r:   r:   r;   ri   q  s    
zCookies.update)r   rK   r1   c                 C   s   |  ||S rM   )r1  rB   r   rK   r:   r:   r;   rx   v  s    zCookies.__setitem__)r   r1   c                 C   s   |  |}|d krt||S rM   )r]   r\   rR  r:   r:   r;   ro   y  s    
zCookies.__getitem__c                 C   s
   |  |S rM   )rP  )rB   r   r:   r:   r;   rz     s    zCookies.__delitem__rD   c                 C   s
   t | jS rM   )r~   r0  rR   r:   r:   r;   r     s    zCookies.__len__c                 C   s   dd | j D S )Nc                 s   s   | ]}|j V  qd S rM   )r   rL  r:   r:   r;   	<genexpr>  s     z#Cookies.__iter__.<locals>.<genexpr>r0  rR   r:   r:   r;   r}     s    zCookies.__iter__c                 C   s   | j D ]} dS dS )NTFrT  )rB   rP   r:   r:   r;   __bool__  s    
zCookies.__bool__c                 C   s"   d dd | jD }d| dS )NrW   c              	   S   s*   g | ]"}d |j  d|j d|j dqS )z<Cookie =z for z />)r   rK   r:  rL  r:   r:   r;   r<     s   z$Cookies.__repr__.<locals>.<listcomp>z	<Cookies[r  )rn   r0  )rB   Zcookies_reprr:   r:   r;   r     s    zCookies.__repr__c                       s>   e Zd ZdZedd fddZeedd fddZ  ZS )	zCookies._CookieCompatRequest
        Wraps a `Request` instance up in a compatibility interface suitable
        for use with `CookieJar` operations.
        Nr7  c                    s*   t  jt|jt|j|jd || _d S )N)r   r/   r   )superrC   r   r   r   r/   r   r   )rB   r   r   r:   r;   rC     s    z%Cookies._CookieCompatRequest.__init__rp   c                    s   t  || || jj|< d S rM   )rX  add_unredirected_headerr   r/   )rB   rJ   rK   rY  r:   r;   rZ    s    z4Cookies._CookieCompatRequest.add_unredirected_header)	r   r   r   r   r   rC   r   rZ  __classcell__r:   r:   rY  r;   r5    s   r5  c                   @   s0   e Zd ZdZedddZejjdddZ	dS )	zCookies._CookieCompatResponserW  r   c                 C   s
   || _ d S rM   r\  )rB   r   r:   r:   r;   rC     s    z&Cookies._CookieCompatResponse.__init__rD   c                 C   s,   t j }| jj D ]\}}|||< q|S rM   )emailr   Messager   r/   rZ   )rB   inforJ   rK   r:   r:   r;   r_    s    

z"Cookies._CookieCompatResponse.infoN)
r   r   r   r   r   rC   r]  r   r^  r_  r:   r:   r:   r;   r4    s   r4  )N)r   r9  )NNN)NN)NN)N) r   r   r   r   r   r   r   rC   r   r  r   r   r   r1  r]   rP  rN  ri   rx   ro   rz   r   r   r   r}   r   rU  r   urllibr   r5  r4  r:   r:   r:   r;   r     sP   	         
r   )Cr-  email.messager]  r   r  r   urllib.requestr`  collections.abcr   http.cookiejarr   r   r   r   r   r   r	   	_decodersr
   r   r   r   r   r   r   r   _exceptionsr   r   r   r   r   r   r   Z
_multipartr   Z_status_codesr   _typesr   r   r   r   r   r    r!   r"   r#   r$   r%   Z_urlsr&   _utilsr'   r(   r)   r*   r+   r,   r-   MutableMappingr   r.   r   r   r   r:   r:   r:   r;   <module>   s0   (
$	4$ |     8