a &jP@sPddlmZddlZddlZddlZddlmZddlmZ ddl m Z ddl m Z ddlmZmZmZmZmZmZddlmZmZd d lmZmZd d lmZerd d lmZd dlm Z dZ!GdddZ"GdddZ#Gddde j$eZ%ddZ&ddZ'd)ddZ(ddZ)ddZ*d d!Z+d*d#d$Z,d%d&Z-Gd'd(d(ee.e.fZ/dS)+) annotationsN)Message cookiejar)Cookie)Morsel) TYPE_CHECKINGAnyDictIteratorMutableMappingOptional)urlparse urlunparse)CookieConflictError CookieError) CookieTypes)Request)Response)Cookiesc@s"eZdZddddZddZdS) MockResponser)responsecCs ||_dSN)_original_response)selfrrAE:\soccer\ml_env\Lib\site-packages\tls_requests/models/cookies.py__init__szMockResponse.__init__cCs(t}|jjD]\}}|||<q|Sr)rrheadersitems)rinfokvrrrr"s zMockResponse.infoN)__name__ __module__ __qualname__rr"rrrrrsrc@seZdZddddZddZddZd d Zd d Zd dZddZ dddZ ddZ ddZ e ddZe ddZe ddZdS) MockRequestr)requestcCs*i|_||_|j|_tt|jj|_dSr) _new_headers_requestr _headersrstrurlschemetyperr)rrrr&szMockRequest.__init__cCs|jSr)r0rrrrget_type,szMockRequest.get_typecCstt|jjjSr)rr-r+r.netlocr2rrrget_host/szMockRequest.get_hostcCs|Srr5r2rrrget_origin_req_host2szMockRequest.get_origin_req_hostcCsP|jdst|jjS|jd}tt|jj}t|j||j|j |j |j gS)NHost) r,getr-r+r.rrr/pathparamsqueryfragment)rhostparsedrrr get_full_url5s   zMockRequest.get_full_urlcCsdSNTrr2rrris_unverifiableJszMockRequest.is_unverifiablecCs||jvp||jvSr)r,r*rnamerrr has_headerMszMockRequest.has_headerNcCs|j||j||Sr)r,r9r*)rrDdefaultrrr get_headerPszMockRequest.get_headercCs||j|<dSrr*rrDvaluerrradd_unredirected_headerSsz#MockRequest.add_unredirected_headercCs|jSrrHr2rrrget_new_headersVszMockRequest.get_new_headerscCs|Sr)rBr2rrr unverifiableYszMockRequest.unverifiablecCs|Sr)r7r2rrrorigin_req_host]szMockRequest.origin_req_hostcCs|Srr6r2rrrr>aszMockRequest.host)N)r%r&r'rr3r5r7r@rBrErGrKrLpropertyrMrNr>rrrrr(%s    r(cseZdZdZd6ddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZd7ddZfddZddZdd Zd!d"Zfd#d$Zd8d%d%d&d'fd(d) Zd9d*d+Zd:d,d-Zd.d/Zd0d1Zd2d3Zd4d5ZZS);RequestsCookieJaraCompatibility class; is a http.cookiejar.CookieJar, but exposes a dict interface. This is the CookieJar we create by default for requests and sessions that don't specify one, since some clients may expect response.cookies and session.cookies to support dict operations. Requests does not use the dict interface internally; it's just for compatibility with external client code. All requests code should work out of the box with externally provided instances of ``CookieJar``, e.g. ``LWPCookieJar`` and ``FileCookieJar``. Unlike a regular CookieJar, this class is pickleable. .. warning:: dictionary operations that are normally O(1) may be O(n). NcCs0z||||}|WSty*|YS0dS)zDict-like get() that also supports optional domain and path args in order to resolve naming collisions from using one cookie jar over multiple domains. .. warning:: operation is O(n), not O(1). N)_find_no_duplicatesKeyError)rrDrFdomainr:rJrrrr9xs  zRequestsCookieJar.getcKs\|dur(t|||d|dddSt|tr|j|kr |durTtd||j}q |durh|Std|d|d|dS)aBoth ``__get_item__`` and ``get`` call this function: it's never used elsewhere in Requests. :param name: a string containing name of cookie :param domain: (optional) string containing domain of cookie :param path: (optional) string containing path of cookie :raises KeyError: if cookie is not found :raises CookieConflictError: if there are multiple cookies that match name and optionally domain and path :return: cookie.value Nz&There are multiple cookies with name, rrr)r]rDrSr:rrJrR)rrDrSr:toReturnr_rrrrQ4s   z%RequestsCookieJar._find_no_duplicatescCs|j}|d|S)4Unlike a normal CookieJar, this class is pickleable. _cookies_lock)__dict__rpop)rstaterrr __getstate__Os  zRequestsCookieJar.__getstate__cCs,|j|d|jvr(ddl}||_dS)rrrN)rr threadingRLockr)rrrrrr __setstate__Vs  zRequestsCookieJar.__setstate__cCs"t}|||||S)z(Return a copy of this RequestsCookieJar.)rP set_policy get_policyr)rnew_cjrrrr^s zRequestsCookieJar.copycCs t|ddS)z&Return the CookiePolicy instance used._policyN)getattrr2rrrreszRequestsCookieJar.get_policy)NNN)NN)N)NN)NN)r%r&r'__doc__r9r\r`rbrcrdrer!rjrlrmrorqrtrvrwrYrrrQrrrr __classcell__rrrrrrPfs2          rPcCs t|}t|}|||dS)zExtract the cookies from the response into a CookieJar. :param cookiejar: http.cookiejar.CookieJar (not necessarily a RequestsCookieJar) :param request: our own requests.Request object :param response: tls_requests.Response object N)r(rextract_cookies)rrr)rrrextract_cookies_to_jarjsrcCs t|}|||dS)zj Produce an appropriate Cookie header string to be sent with `request`, or None. :rtype: str r)r(add_cookie_headerrLr9)jarr)rrrrget_cookie_headervs rcCszg}|D]N}|j|krq|dur,||jkr,q|dur@||jkr@q||j|j|jfq|D]\}}}||||q\dS)zjUnsets a cookie by key, by default over all domains and paths. Wraps CookieJar.clear(), is O(n). N)rDrSr:rgclear)rrDrSr: clearablesr_rrrrUs rUcCsN|dur dSt|dr|St|}||D]}|t|q4|S)Nr)r{rrrY)rnew_jarr_rrr_copy_cookie_jars  rcKsd||ddddddddddidd }t|t|}|rJtd t|||t|d |d <t|d |d <|d d|d<t|d|d<tjfi|S)zMake a cookie from underspecified parameters. By default, the pair of `key` and `value` will be set for the domain '' and sent on every request (this is sometimes called a "supercookie"). rNrz/FTHttpOnly) versionrDrJportrSr:secureexpiresdiscardcomment comment_urlrestrfc2109z2create_cookie() got unexpected keyword arguments: rport_specifiedrSdomain_specified.domain_initial_dotr:path_specified)r\rrarboolrxrr)rDrJrZresultbadargsrrrrXs. rXcCsd}|drRzttt|d}WqttyNtd|ddYqt0n"|drtd}tt|d|}t|dt|dd|d ||j |d dd |d idt|d |j |dpdd S)zBConvert a Morsel object into a Cookie containing the one k/v pair.Nzmax-agez max-age: z must be integerrz%a, %d-%b-%Y %H:%M:%S GMTrFrSr:rhttponlyrrr) rrrrSrrDr:rrrrrJr) inttime ValueErrorrcalendartimegmstrptimerXrkeyrJ)morselr time_templaterrrrWs0     rWTcCsR|durt}|durNdd|D}|D]$}|s8||vr(|t|||q(|S)aCReturns a CookieJar from a key/value dictionary. :param cookie_dict: Dict of key/values to insert into CookieJar. :param cookiejar: (optional) A cookiejar to add the cookies to. :param overwrite: (optional) If False, will not replace cookies already in the jar with new ones. :rtype: CookieJar NcSsg|] }|jqSrrD.0r_rrr z'cookiejar_from_dict..)rPrYrX) cookie_dictr overwritenames_from_jarrDrrrcookiejar_from_dicts  rcCstt|tjstdt|tr.t||dd}nBt|tjrpz||Wn&tyn|D]}||qZYn0|S)zAdd cookies to cookiejar and returns a merged CookieJar. :param cookiejar: CookieJar object to add the cookies to. :param cookies: Dictionary or CookieJar object to be added. :rtype: CookieJar z!You can only merge into CookieJarF)rr) rVrrrdictrrAttributeErrorrY)rcookies cookie_in_jarrrr merge_cookiess    rc@s"eZdZd9dddddZd:ddddd Zd d dd d dZd ddddZddddZd;ddddZddddd!d"Z d#dd$d%Z dddd&d'd(Z ddd)d*d+Zddd)d,d-Zd.dd/d0Zd1dd2d3Zd4dd5d6Zddd7d8ZdS)?rNzOptional[CookieTypes]r)rrcCs|||_dSr)_prepare_cookiejarrrrrrrrszCookies.__init__rPcCst||jr|jSt|ttttfrt}t|tr<|n|}|D]Z}t|ttfrDt |dkrD|d|d}}t|t t frt |}|t |t |qD|StS)Nrrr) rVrsrrtuplerar\rPr!lenfloatrr-)rrrZ cookie_itemsitemr#r$rrrrs zCookies._prepare_cookiejarrr)rr)rcCst|j||dSr)rr)rrr)rrrr%szCookies.extract_cookiesr-)r)rcCs t|j|Sr)rrr1rrrr(szCookies.get_cookie_headerzOptional[Cookie])rcKs`|dur*t|j||d|dddSt|tr>t|}nt||fi|}|j||S)NrSr:rT)rUrr9rVrrWrXrY)rrDrJrZr_rrrr\+s   z Cookies.setr cCs|j||||Srrr9)rrDrFrSr:rrrr98sz Cookies.getz Optional[str])rDrSr:rcCst|j|||dS)NrT)rUr)rrDrSr:rrrdelete;szCookies.deleterTcCs4g}|r|||r$|r$|||jj|dSr)rgrr)rrSr:r~rrrr>s   z Cookies.clearcCs|j||dSr)rrrrrrrrGszCookies.updatez 'Cookies'cCs|}t|j|_|Sr)rsrr)rretrrrrJs z Cookies.copy)rDrJrcCs|||dSrrurIrrrrvOszCookies.__setitem__)rDrcCs|j|pdS)NrzrrCrrrrtRszCookies.__getitem__cCs||dSr)rrCrrrrwUszCookies.__delitem__rcCs t|jSr)rrr2rrr__len__XszCookies.__len__z Iterator[str]cCsdd|jDS)Ncss|] }|jVqdSrrrrrr \rz#Cookies.__iter__..rr2rrr__iter__[szCookies.__iter__rcCs|jD]}dSdS)NTFr)r_rrr__bool__^s zCookies.__bool__cCs&ddd|jD}d|jj|fS)Nz, cSsg|] }t|qSr)reprrrrrrdrz$Cookies.__repr__..z<%s[%s]>)joinrrsr%)rZcookiejar_reprrrr__repr__cszCookies.__repr__)N)N)NNN)NN)NN)N)r%r&r'rrrrr\r9rrrrrvrtrwrrrrrrrrrs"  r)NN)NT)0 __future__rrrr email.messagerhttprrhttp.cookiejarr http.cookiesrtypingrr r r r r urllib.parserr exceptionsrrtypesrr)rrr__all__rr(rrPrrrUrrXrWrrr-rrrrrs:          A  $