U nj@sddlmZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddlmZmZddlmZmZmZmZddlmZmZmZmZmZmZmZmZddlm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&dd l'm(Z(dd l)m*Z*dd l+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6dd l7m8Z8dd l9m:Z:m;Z;ddddgZd2ddddddZ?d3ddddd d!Z@ddd"d#d$ZAdd%d&d'd(ZBd)d*d+d,d-ZCGd.dde jDeEeEfZFGd/ddZGGd0ddZHGd1dde jDeEeEfZIdS)4) annotationsN)Mapping)Cookie CookieJar) ByteStreamUnattachedStreamencode_requestencode_response)SUPPORTED_DECODERS ByteChunkerContentDecoderIdentityDecoder LineDecoder MultiDecoder TextChunker TextDecoder)CookieConflictHTTPStatusErrorRequestNotReadResponseNotRead StreamClosedStreamConsumedrequest_context)(get_multipart_boundary_from_content_type)codes) AsyncByteStream CookieTypes HeaderTypesQueryParamTypesRequestContent RequestDataRequestExtensions RequestFilesResponseContentResponseExtensionsSyncByteStream)URL)to_bytes_or_strto_strCookiesHeadersRequestResponse authorizationzproxy-authorizationstrbool)encodingreturncCs*zt|Wntk r$YdSXdS)z7 Return `True` if `encoding` is a known codec. FT)codecslookup LookupErrorr1r71/tmp/pip-unpacked-wheel-7ig9zwdf/httpx/_models.py_is_known_encoding8s r9z str | bytes str | Nonebytes)keyr1r2cCst|tr|S||pdS)zE Coerce str/bytes into a strictly byte-wise HTTP header key. ascii) isinstancer;encode)r<r1r7r7r8_normalize_header_keyCsr@)valuer1r2cCs8t|tr|St|ts*tdt|||p4dS)zG Coerce str/bytes into a strictly byte-wise HTTP header value. z'Header value must be str or bytes, not r=)r>r;r/ TypeErrortyper?)rAr1r7r7r8_normalize_header_valueJs   rD) content_typer2cCstj}||d<|jddS)N content-type)failobj)emailmessageMessageget_content_charset)rEmsgr7r7r8_parse_content_type_charsetUs rMzlist[dict[str, str]]rAr2c Csg}d}||}|s|Std|D]}z|dd\}}Wntk r\|d}}YnXd|di}|dD]F}z|d\}}Wntk rYqYnX|||||<qv||q&|S) a7 Returns a list of parsed link headers, for more info see: https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Link The generic syntax of those is: Link: < uri-reference >; param1=value1; param2="value2" So for instance: Link; '; type="image/jpeg",;' would return [ {"url": "http:/.../front.jpeg", "type": "image/jpeg"}, {"url": "http://.../back.jpeg"}, ] :param value: HTTP Link entity-header field :return: list of parsed link headers z '"z, *<;rurlz<> '"=)stripresplit ValueErrorappend) rAlinks replace_charsvalrQparamslinkparamr<r7r7r8_parse_header_links]s&   r^z4typing.Iterable[tuple[typing.AnyStr, typing.AnyStr]]z4typing.Iterator[tuple[typing.AnyStr, typing.AnyStr]])itemsr2ccs8|D].\}}t|tkr(td|d}||fVqdS)Nz[secure])Z match_type_of)r)lowerSENSITIVE_HEADERSr()r_kvr7r7r8_obfuscate_sensitive_headerss  rdc@sZeZdZdZdBddddddZed d d d Zejd dd dd Zedd ddZdd ddZ dd ddZ dd ddZ dd ddZ dCd dddd d!Z dDd d#d$d%d&d'ZdEddd(d)d*Zdd d+d,Zd d d-d.d/Zd d dd0d1d2Zd dd-d3d4Zdd#d-d5d6Zd7d d8d9Zd:d d;d<Zdd#d=d>d?Zd d d@dAZdS)Fr+z9 HTTP headers, as a case-insensitive multi-dict. NHeaderTypes | Noner:None)headersr1r2cCsg|_t|trt|j|_nt|trf|D]2\}}t||}t||}|j|| |fq0n@|dk r|D]2\}}t||}t||}|j|| |fqr||_ dSN) _listr>r+listrr_r@rDrWr` _encoding)selfrgr1rbrcZ bytes_keyZ bytes_valuer7r7r8__init__s       zHeaders.__init__r/r2c Csn|jdkrhdD]R}|jD]<\}}z||||Wqtk rRYqYqXq||_qhqd|_|jS)zn Header encoding is mandated as ascii, but we allow fallbacks to utf-8 or iso-8859-1. N)r=utf-8z iso-8859-1)rkrawdecodeUnicodeDecodeError)rlr1r<rAr7r7r8r1s  zHeaders.encodingrNcCs ||_dSrh)rkrlrAr7r7r8r1szlist[tuple[bytes, bytes]]cCsdd|jDS)zH Returns a list of the raw header items, as byte pairs. cSsg|]\}}}||fqSr7r7).0Zraw_key_rAr7r7r8 szHeaders.raw..rirlr7r7r8rpsz Headers.rawztyping.KeysView[str]csfddjDS)Ncs i|]\}}}|jdqSrhrqr1rtrur<rArxr7r8 s z Headers.keys..)rikeysrxr7rxr8r|sz Headers.keysztyping.ValuesView[str]cCs^i}|jD]J\}}}||j}||j}||krL||d|7<q |||<q |S)N, )rirqr1valuesrlZ values_dictrur<rAZstr_keyZ str_valuer7r7r8r~s   zHeaders.valuesztyping.ItemsView[str, str]cCs^i}|jD]J\}}}||j}||j}||krL||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. r})rirqr1r_rr7r7r8r_s   z Headers.itemszlist[tuple[str, str]]csfddjDS)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. cs*g|]"\}}}|j|jfqSr7ryrzrxr7r8rvsz'Headers.multi_items..rwrxr7rxr8 multi_itemss zHeaders.multi_items typing.Any)r<defaultr2cCs(z ||WStk r"|YSXdS)z Return a header value. If multiple occurrences of the header occur then concatenate them together with commas. N)KeyError)rlr<rr7r7r8gets z Headers.getFr0z list[str])r< split_commasr2csZ|jfddjD}|s.|Sg}|D]}|dd|dDq6|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. cs*g|]"\}}}|kr|jqSr7)r`rqr1)rtruitem_keyZ item_valueZget_header_keyrlr7r8rvs z$Headers.get_list..cSsg|] }|qSr7)rS)rtitemr7r7r8rvs,)r`r?r1riextendrU)rlr<rr~Z split_valuesrAr7rr8get_lists zHeaders.get_list)rgr2cCs:t|}|D]}||kr||q|j|jdSrh)r+r|poprir)rlrgr<r7r7r8updates   zHeaders.updatecCst||jdS)Nr6)r+r1rxr7r7r8copysz Headers.copy)r<r2cs@|jfddjD}|r4d|St|dS)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 cs&g|]\}}}|kr|jqSr7ry)rtru header_key header_valueZnormalized_keyrlr7r8rv%sz'Headers.__getitem__..r}N)r`r?r1rijoinr)rlr<r_r7rr8 __getitem__s  zHeaders.__getitem__r<rAr2cs||jp d}||jpd}|fddt|jD}t|ddD] }|j|=qP|r||d}||f|j|<n|j||fdS)zs Set the header `key` to `value`, removing any duplicate entries. Retains insertion order. rocs"g|]\}\}}}|kr|qSr7r7rtidxrurZ lookup_keyr7r8rv9s z'Headers.__setitem__..rNr)r?rkr` enumeraterireversedrW)rlr<rAZset_key set_valueZ found_indexesrr7rr8 __setitem__0s  zHeaders.__setitem__csN||jfddt|jD}|s4t|t|D] }|j|=q.N)r`r?r1rrirr)rlr<Z pop_indexesrr7rr8 __delitem__Hs  zHeaders.__delitem__cCs$||j}|dd|jDkS)NcSsg|]\}}}|qSr7r7)rtrur<r7r7r8rv\sz(Headers.__contains__..)r`r?r1ri)rlr<rr7r7r8 __contains__ZszHeaders.__contains__ztyping.Iterator[typing.Any]cCs t|Srh)iterr|rxr7r7r8__iter__^szHeaders.__iter__intcCs t|jSrh)lenrirxr7r7r8__len__aszHeaders.__len__)otherr2cCsTz t|}Wntk r"YdSXdd|jD}dd|jD}t|t|kS)NFcSsg|]\}}}||fqSr7r7rzr7r7r8rvjsz"Headers.__eq__..cSsg|]\}}}||fqSr7r7rzr7r7r8rvks)r+rVrisorted)rlrZ other_headersZ self_listZ other_listr7r7r8__eq__ds zHeaders.__eq__cCsv|jj}d}|jdkr"d|j}tt|}t|}t|t|k}|rb|d||dS|d||dS)NrPr=z , encoding=()) __class____name__r1rjrdrdictr)rl class_nameZ encoding_strZas_listZas_dictZno_duplicate_keysr7r7r8__repr__ns  zHeaders.__repr__)NN)N)F)N)r __module__ __qualname____doc__rmpropertyr1setterrpr|r~r_rrrrrrrrrrrrrr7r7r7r8r+s4    c@seZdZdddddddddd ddddddd d d d d dd ddZdddddZeddddZddddZddddZddddZ d dd!d"Z d dd#d$d%Z dS)&r,N) r[rgcookiescontentdatafilesjsonstream extensionsr/z URL | strzQueryParamTypes | NonereCookieTypes | NonezRequestContent | NonezRequestData | NonezRequestFiles | Noneztyping.Any | None'SyncByteStream | AsyncByteStream | NonezRequestExtensions | Nonerf) methodrQr[rgrrrrrrrr2c Cs||_|dkrt|n t||d|_t||_| dkr>int| |_|rZt| || dkr|j d} t |||| t | r| |jjnddd\}} ||| |_t| tr|n| |_dS)N)r[rF)rE)rrrrboundary)upperrr'rQr+rgrrr*set_cookie_headerrr rr?r1_preparerr>rread) rlrrQr[rgrrrrrrrrEr7r7r8rms0      zRequest.__init__dict[str, str]default_headersr2cCs|D].\}}|dkr(d|jkr(q|j||qg}d|jk}d|jkpXd|jk}|sx|jjrx|d|jjf|s|jdkr|dt ||jj |_dS)Ntransfer-encodingzContent-LengthHostzTransfer-EncodingsHost)POSTPUTPATCH)sContent-Length0) r_r`rg setdefaultrQhostrWnetlocrr+rp)rlrr<rAZ auto_headersZhas_hostZhas_content_lengthr7r7r8rs   zRequest._preparer;rncCst|dst|jSN_content)hasattrrrrxr7r7r8rs zRequest.contentcCsHt|dsBt|jtjstd|j|_t|jtsBt|j|_|jS)6 Read and return the request content. r) rr>rtypingIterableAssertionErrorrrrrxr7r7r8rs    z Request.readcsXt|dsRt|jtjstddd|j2IdH|_t|jtsRt|j|_|jS)rrrcsg|z3dHW}|q6Srhr7rtpartr7r7r8rvsz!Request.aread..N) rr>rr AsyncIterablerrrrrxr7r7r8areads    z Request.areadcCs,|jj}t|j}d|d|jd|dS)N)rrr/rQr)rlrrQr7r7r8rs zRequest.__repr__dict[str, typing.Any]cCsdd|jDS)NcSsi|]\}}|dkr||qS))rrr7rtnamerAr7r7r8r{sz(Request.__getstate__..__dict__r_rxr7r7r8 __getstate__szRequest.__getstate__stater2cCs0|D]\}}t|||qi|_t|_dSrh)r_setattrrrrrlrrrAr7r7r8 __setstate__szRequest.__setstate__) rrrrmrrrrrrrrr7r7r7r8r,~s$(:c@seZdZddddddddddd ddddddd d d d d dd ddZdddddZeddddZejdddddZeddddZejdddd dZed!dd"d#Z ed!dd$d%Z ed&dd'd(Z ed)dd*d+Z ed!dd,d-Z eddd.d/Zejd!ddd0d/Zeddd1d2Zd3dd4d5Zed6dd7d8Zed6dd9d:Zed6dd;d<Zed6dd=d>Zed6dd?d@Zed6ddAdBZed6ddCdDZdddEdFZdddGdHdIZedJddKdLZedMddNdOZedddPdQZd!ddRdSZdTddUdVZdTddWdXdYZd)ddZd[Z dxd\d]d^d_d`Z!dyd\dad^dbdcZ"daddddeZ#dzd\d]d^dfdgZ$dddhdiZ%d)ddjdkZ&d{d\dld^dmdnZ'd|d\dod^dpdqZ(doddrdsZ)d}d\dld^dtduZ*dddvdwZ+dS)~r-Nro) rgrtexthtmlrrrequestrhistorydefault_encodingrrezResponseContent | Noner:rrzRequest | NonezResponseExtensions | Nonezlist[Response] | Nonez#str | typing.Callable[[bytes], str]rf) status_codergrrrrrrrrrr2c Cs||_t||_||_d|_| dkr(int| |_| dkr>gnt| |_d|_ d|_ | |_ |dkrt ||||\}}| |||_t|tr|n||_d|_dS)NFr)rr+rg_requestZ next_requestrrrjr is_closedis_stream_consumedrr rrr>rr_num_bytes_downloaded) rlrrgrrrrrrrrrr7r7r8rms"    zResponse.__init__rrcCs<|D].\}}|dkr(d|jkr(q|j||qdS)Nrzcontent-length)r_r`rgr)rlrr<rAr7r7r8r;szResponse._preparezdatetime.timedeltarncCst|dstd|jS)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 RuntimeErrorrrxr7r7r8elapsedBs  zResponse.elapsed)rr2cCs ||_dSrh)r)rlrr7r7r8rOsr,cCs|jdkrtd|jS)zR Returns the request instance associated to the current response. Nz7The request instance has not been set on this response.)rrrxr7r7r8rSs  zResponse.requestrNcCs ||_dSrh)rrsr7r7r8r^sr/cCs8z|jd}Wntk r$YdSX|jdddSdS)N http_versionzHTTP/1.1r=ignoreerrors)rrrq)rlrr7r7r8rbs zResponse.http_versioncCsBz|jd}Wn tk r.t|jYSX|jdddSdS)N reason_phraser=rr)rrrZget_reason_phraserrq)rlrr7r7r8rks zResponse.reason_phraser'cCs|jjS)zA Returns the URL for which the request was made. )rrQrxr7r7r8rQtsz Response.urlr;cCst|dst|jSr)rrrrxr7r7r8r{s zResponse.contentcCsNt|dsH|j}|sd|_n,t|jp&dd}d||j|g|_|jS)N_textrPror6)rrrrr1rrqflush)rlrdecoderr7r7r8rs z Response.textcCsZt|dsT|j}|dks t|sJt|jtr4|j}nt|drJ||j}|pPd|_|jS)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. rkNrro)rcharset_encodingr9r>rr/rrk)rlr1r7r7r8r1s     zResponse.encodingcCst|drtd||_dS)z Set the encoding to use for decoding the byte content into text. If the `text` attribute has been accessed, attempting to set the encoding will throw a ValueError. rz?Setting encoding after `text` has been accessed is not allowed.N)rrVrkrsr7r7r8r1s  cCs |jd}|dkrdSt|S)zO Return the encoding, as specified by the Content-Type header. z Content-TypeN)rgrrM)rlrEr7r7r8rs zResponse.charset_encodingr c Cst|dsg}|jjddd}|D]B}|}zt|}||Wq"tk rbYq"Yq"Xq"t|dkr~|d|_ n"t|dkrt |d|_ nt |_ |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. _decoderzcontent-encodingT)rrr)children) rrgrrSr`r rWrrrrr)rlZdecodersr~rAZ decoder_clsr7r7r8_get_content_decoders       zResponse._get_content_decoderr0cCs t|jS)zU A property which is `True` for 1xx status codes, `False` otherwise. )ris_informationalrrxr7r7r8rszResponse.is_informationalcCs t|jS)zU A property which is `True` for 2xx status codes, `False` otherwise. )r is_successrrxr7r7r8rszResponse.is_successcCs t|jS)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_redirectrrxr7r7r8rs zResponse.is_redirectcCs t|jS)zU A property which is `True` for 4xx status codes, `False` otherwise. )ris_client_errorrrxr7r7r8rszResponse.is_client_errorcCs t|jS)zU A property which is `True` for 5xx status codes, `False` otherwise. )ris_server_errorrrxr7r7r8rszResponse.is_server_errorcCs t|jS)z] A property which is `True` for 4xx and 5xx status codes, `False` otherwise. )ris_errorrrxr7r7r8rszResponse.is_errorcCs(|jtjtjtjtjtjfko&d|jkS)zs Returns True for 3xx responses with a properly formed URL redirection, `False` otherwise. ZLocation)rrMOVED_PERMANENTLYFOUND SEE_OTHERTEMPORARY_REDIRECTPERMANENT_REDIRECTrgrxr7r7r8has_redirect_locations zResponse.has_redirect_locationcCst|j}|dkrtd|jr |S|jr,d}nd}|jd}dddd d }||d }|j||d }t|||d dS)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://developer.mozilla.org/en-US/docs/Web/HTTP/Status/{0.status_code}z{error_type} '{0.status_code} {0.reason_phrase}' for url '{0.url}' For more information check: https://developer.mozilla.org/en-US/docs/Web/HTTP/Status/{0.status_code}dzInformational responsezRedirect responsez Client errorz Server error)rzInvalid status code) error_type)rresponse)rrrr rrformatr)rlrrIZ status_classZ error_typesrr7r7r8raise_for_statuss*  zResponse.raise_for_status)kwargsr2cKstj|jf|Srh)jsonlibloadsr)rlrr7r7r8r?sz Response.jsonr*cCs$t|dst|_|j||jS)N_cookies)rr*rextract_cookiesrxr7r7r8rBs  zResponse.cookiesz dict[str | None, dict[str, str]]cCs*|jd}|dkriSddt|DS)zI Returns the parsed header links of the response, if any r\NcSs"i|]}|dp|d|qS)relrQ)r)rtr\r7r7r8r{Rsz"Response.links..)rgrr^)rlheaderr7r7r8rXIs  zResponse.linkscCs|jSrh)rrxr7r7r8num_bytes_downloadedWszResponse.num_bytes_downloadedcCsd|jd|jdS)Nz )rrrxr7r7r8r[szResponse.__repr__rcCsdd|jDS)NcSsi|]\}}|dkr||qS))rrrrr7rr7r7r8r{_sz)Response.__getstate__..rrxr7r7r8r^szResponse.__getstate__rcCs6|D]\}}t|||qd|_i|_t|_dS)NT)r_rrrrrrr7r7r8res zResponse.__setstate__cCs t|dsd||_|jS)7 Read and return the response content. rr)rr iter_bytesrrxr7r7r8rls z Response.readz int | Noneztyping.Iterator[bytes]) chunk_sizer2c cst|drT|dkrt|jn|}tdt|jt|dD]}|j|||Vq8n|}t|d}t|jdf| D]$}| |}| |D] }|Vqq|| }| |D] }|Vq| D] }|VqW5QRXdSz A byte-iterator over the decoded response content. This allows us to handle gzip, deflate, brotli, and zstd encoded responses. rNrrrr) rrrrangemaxrr rriter_rawrqrrlrirchunkerZ raw_bytesdecodedchunkr7r7r8rts       zResponse.iter_bytestyping.Iterator[str]c cst|jp dd}t|d}t|jdf|D]$}||}||D] }|VqHq0|}||D] }|Vqh|D] }|Vq|W5QRXdSz A str-iterator over the decoded response content that handles both gzip, deflate, etc but also detects the content's string encoding. ror6r!r"N)rr1rrrrrqrrlrrr(Z byte_contentZ text_contentr*r7r7r8 iter_texts     zResponse.iter_textc csZt}t|jd>|D]}||D] }|Vq*q|D] }|Vq@W5QRXdSNr")rrrr.rqrrlrrliner7r7r8 iter_liness   zResponse.iter_linesc cs|jr t|jrtt|jts,tdd|_d|_t |d}t |j d:|jD],}|jt |7_| |D] }|VqvqVW5QRX|D] }|Vq|dS)@ A byte-iterator over the raw response content. z5Attempted to call a sync iterator on an async stream.Trr!r"N)rrrrr>rr&rrr rrrrqrcloserlrr(Zraw_stream_bytesr*r7r7r8r%s"    zResponse.iter_rawc CsFt|jtstd|jsBd|_t|jd|jW5QRXdS) 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>rr&rrrrr4rxr7r7r8r4s  zResponse.closecs0t|ds*ddd|2IdH|_|jS)rrrcsg|z3dHW}|q6Srhr7rr7r7r8rvsz"Response.aread..N)rr aiter_bytesrrxr7r7r8rs  zResponse.areadztyping.AsyncIterator[bytes]c Cst|drT|dkrt|jn|}tdt|jt|dD]}|j|||Vq8n|}t|d}t|jdp| 2z,3dHW}| |}| |D] }|Vqq|6| }| |D] }|Vq| D] }|VqW5QRXdSr ) rrrr#r$rr rr aiter_rawrqrr&r7r7r8r7s      zResponse.aiter_bytesztyping.AsyncIterator[str]c Cst|jp dd}t|d}t|jdp|2z,3dHW}||}||D] }|VqPq06|}||D] }|Vqr|D] }|VqW5QRXdSr,)rr1rrrr7rqrr-r7r7r8 aiter_texts    zResponse.aiter_textc Csdt}t|jdH|2z"3dHW}||D] }|Vq2q6|D] }|VqJW5QRXdSr/)rrrr9rqrr0r7r7r8 aiter_liness  zResponse.aiter_linesc Cs|jr t|jrtt|jts,tdd|_d|_t |d}t |j dD|j2z43dHW}|jt |7_| |D] }|Vq~qV6W5QRX|D] }|Vq|IdHdS)r3z6Attempted to call an async iterator on an sync stream.Trr!r"N)rrrrr>rrrrr rrrrqracloser5r7r7r8r8 s"   zResponse.aiter_rawc sLt|jtstd|jsHd|_t|jd|jIdHW5QRXdS)r6z3Attempted to call an async close on an sync stream.Tr"N)r>rrrrrrr;rxr7r7r8r;)s  zResponse.aclose)N)N)N)N)N)N),rrrrmrrrrrrrrQrrr1rrrrrrrrr rrrrXrrrrrrr.r2r%r4rr7r9r:r8r;r7r7r7r8r-s(7      %     c@s:eZdZdZd:dddddZddd d d Zd dd ddZd;ddddddddZdddddddZ d?dddd d!Z dddd"d#d$Z ddd%d&d'Z ddd%d(d)Zd*d+d,d-Zd.d+d/d0Zd1d+d2d3Zdd+d4d5ZGd6d7d7ejjZGd8d9d9ZdS)@r*z- HTTP Cookies, as a mutable mapping. Nrrf)rr2cCs|dkst|trDt|_t|tr|D]\}}|||q,n`t|trrt|_|D]\}}|||qZn2t|trt|_|jD]}|j|qn||_dSrh) r>rrjarr_setrjr* set_cookie)rlrr<rAcookier7r7r8rm<s     zCookies.__init__r-rr2cCs(||}||j}|j||dS)zO Loads any cookies based on the response `Set-Cookie` headers. N)_CookieCompatResponse_CookieCompatRequestrr<r)rlrZurllib_responseurllib_requestr7r7r8rMs  zCookies.extract_cookiesr,rr2cCs||}|j|dS)zM Sets an appropriate 'Cookie:' HTTP header on the `Request`. N)rBr<add_cookie_header)rlrrCr7r7r8rVs zCookies.set_cookie_headerrP/r/)rrAdomainpathr2cCsTd||dd|t||d|t|dddddddidd}tf|}|j|dS)zU Set a cookie value by name. May optionally include domain and path. rNF.THttpOnly)versionrrAportport_specifiedrGdomain_specifieddomain_initial_dotrHpath_specifiedsecureexpiresdiscardcomment comment_urlrestrfc2109)r0 startswithrr<r>)rlrrArGrHrr?r7r7r8r=]s( z Cookies.setr:)rrrGrHr2cCsnd}|jD]R}|j|kr |dks*|j|kr |dks<|j|kr |dk rVd|}t||j}q |dkrj|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=)r<rrGrHrrA)rlrrrGrHrAr?rIr7r7r8rws    z Cookies.get)rrGrHr2cs\dk r dk r |jSfdd|jD}|D]}|j|j|j|jqg|]6}|jkrdks$|jkrdks6|jkr|qSrh)rrGrHrtr?rGrrHr7r8rvs   z"Cookies.delete..)r<clearrGrHr)rlrrGrHremover?r7rZr8deletes zCookies.delete)rGrHr2cCsDg}|dk r|||dk r4|dk s*t|||jj|dS)z Delete all cookies. Optionally include a domain and path in order to only delete a subset of all the cookies. N)rWrr<r[)rlrGrHargsr7r7r8r[s   z Cookies.clearcCs$t|}|jD]}|j|qdSrh)r*r<r>)rlrr?r7r7r8rs zCookies.update)rrAr2cCs |||Srh)r=rlrrAr7r7r8rszCookies.__setitem__)rr2cCs||}|dkrt||Srh)rrr_r7r7r8rs zCookies.__getitem__cCs ||Srh)r])rlrr7r7r8rszCookies.__delitem__rrncCs t|jSrh)rr<rxr7r7r8rszCookies.__len__r+cCsdd|jDS)Ncss|] }|jVqdSrh)rrYr7r7r8 sz#Cookies.__iter__..r<rxr7r7r8rszCookies.__iter__r0cCs|jD]}dSdS)NTFra)rlrur7r7r8__bool__s zCookies.__bool__cCs"ddd|jD}d|dS)Nr}c Ss*g|]"}d|jd|jd|jdqS)z)rrArGrYr7r7r8rvsz$Cookies.__repr__..z eZdZdZdddfdd Zddddfd d ZZS) zCookies._CookieCompatRequest Wraps a `Request` instance up in a compatibility interface suitable for use with `CookieJar` operations. r,rfrDcs*tjt|jt|j|jd||_dS)N)rQrgr)superrmr/rQrrgrr)rlrrr7r8rms z%Cookies._CookieCompatRequest.__init__r/rcst||||jj|<dSrh)rdadd_unredirected_headerrrg)rlr<rArer7r8rfsz4Cookies._CookieCompatRequest.add_unredirected_header)rrrrrmrf __classcell__r7r7rer8rBsrBc@s.eZdZdZdddddZddd d Zd S) zCookies._CookieCompatResponsercr-rfr@cCs ||_dSrh)r)rlrr7r7r8rmsz&Cookies._CookieCompatResponse.__init__zemail.message.MessagerncCs,tj}|jjD]\}}|||<q|Srh)rHrIrJrrgr)rlinfor<rAr7r7r8rhs  z"Cookies._CookieCompatResponse.infoN)rrrrrmrhr7r7r7r8rAsrA)N)rPrF)NNN)NN)NN)N)rrrrrmrrr=rr]r[rrrrrrrbrurllibrr,rBrAr7r7r7r8r*7s.   )N)N)J __future__rr3datetime email.messagerHrrrTrurllib.requestricollections.abcrhttp.cookiejarrrrrrr r _decodersr r r rrrrr _exceptionsrrrrrrrZ _multipartrZ _status_codesr_typesrrrrr r!r"r#r$r%r&Z_urlsr'_utilsr(r)__all__rar9r@rDrMr^rdMutableMappingr/r+r,r-r*r7r7r7r8sF  ( $  4    % t8