o Zj4@sdgZddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZmZddlmZmZdd lmZd d lmZd d lmZmZede eeefeeeeffZeGdddZ e!dej"Z#e!dej"Z$GdddeeefZ%dS)CookiesN)suppress) dataclass)Cookie CookieJar)_unquote)OptionalUnion)IteratorMutableMapping)urlparse)CurlCffiWarning)CookieConflict RequestsErrorc@seZdZUeed<eed<dZeed<dZeed<dZeed<dZ eed <d Z e ed <dZ eed <e d dZe ddZedefddZddZedefddZdefddZdS) CurlMorselnamevaluehostnameF subdomains/pathsecurerexpires http_onlycCs|dkS)NTRUEsrrU/home/dinkstrade/pdmp/venv/lib/python3.10/site-packages/curl_cffi/requests/cookies.py parse_bool$szCurlMorsel.parse_boolcCs |rdSdS)NrFALSErrrrr! dump_bool(s zCurlMorsel.dump_boolset_cookie_linec Csr|d\}}}}}}}|r|ddkrd} |dd} nd} |} || |||||t||t|| dS)N r#T F)rrrrrrrr)decodesplitr"intr) clsr%rrrrrrrrdomainrrr!from_curl_format,s0 zCurlMorsel.from_curl_formatc CsT|jstd|jd|jd|j||j|j||jt |j |j|jgS)NzDomain not found for cookie =r&) rrrrjoinr$rrrstrrselfrrr!to_curl_formatIs  zCurlMorsel.to_curl_formatcookiec Cs2||j|jpd|j|j|j|jt|jpdddS)NrrF)rrrrrrrr)rrr-domain_specifiedrrr+r)r,r5rrr!from_cookiejar_cookieXs z CurlMorsel.from_cookiejar_cookiereturncCstdiddd|jd|jddddd|jd |jd t|jd d |jd t|jd|jd|j dkr;dn|j d|j dkdddddt |j dddSd|j dkdddddt |j dddS)Nversionrrrportport_specifiedFr-r6domain_initial_dot.rpath_specifiedrrdiscardcomment comment_urlrest)rrfc2109r) rrrrrbool startswithrrrdictrr2rrr!to_cookiejar_cookiees^       zCurlMorsel.to_cookiejar_cookieN)__name__ __module__ __qualname__r1__annotations__rrrDrrrr+r staticmethodr"r$ classmethodbytesr.r4rr7rGrrrr!rs&          rz:\d+$z\.\d+$c @seZdZdZd;deeddfddZdefddZd eddfd d Z de fd d Z de e fddZde e fddZde ddfddZde eddfddZ ddee dee defd%d&Z  d>de dee dee ddfd'd(Zd>dee dee ddfd)d*Zd;deeddfd+d,Zde de ddfd-d.Zde de fd/d0Zde ddfd1d2Zdefd3d4Zdee fd5d6Zde fd7d8Z!de fd9d:Z"dS)?rz- HTTP Cookies, as a mutable mapping. Ncookiesr8cCs|dus t|tr%t|_t|tr!|D] \}}|||qdSdSt|tr=t|_|D] \}}|||q0dSt|trTt|_|jD]}|j|qIdS||_dSN) isinstancerFrjaritemssetlistr set_cookie)r3rOkeyrr5rrr!__init__s$      zCookies.__init__cCsB|j}|d}|j}|ddt||d<||d<|S)NrR _cookies_lock _jar_type _jar_state)__dict__copypoptype)r3staterR jar_staterrr! __getstate__s     zCookies.__getstate__r`cCsJ|d}|d}t|}|j|t|_|j|||_dS)NrZr[) r^object__new__r\update threadingRLockrYrR)r3r`jar_typerarRrrr! __setstate__s       zCookies.__setstate__cCs^t|jd}|dkr|jdd}td|d}|}|ddkr-t |s-|d7}|S)z Almost equivalent to the eff_request_host function in: https://github.com/python/cpython/blob/3.11/Lib/http/cookiejar.py#L636 rrHostr=z.local) r urlheadersget cut_port_resublowerfindIPV4_REsearch)r3requesthostrrr!_eff_request_hostszCookies._eff_request_hostcCs|jjg}z-tt|jj_|_|jD]}t|}|j s(| ||_ | |qW|jj n|jj w|j |S)zWthe process is similar to ``cookiejar.add_cookie_header``, but load all cookies)rRrYacquirer+time_policy_nowrr7rrwappendreleaseclear_expired_cookies)r3rumorselsr5morselrrr!get_cookies_for_curls      zCookies.get_cookies_for_curlrcCs,|D] }|}|j|q|jdSrP)rGrRrVr~)r3rrr5rrr!update_cookies_from_curlsz Cookies.update_cookies_from_curlrc Csr|jg}|jr|jd}|d|g}|D] }tt|j||j|jWdn1s1wYqdS)Nr=) rrlstriprKeyErrorrRclearrr)r3rdomainsr-rrr!_delete_curl_morsels  zCookies._delete_curl_morselchangescCsz|D]3}|d\}}}|r|dvrtd|t|}|dkr0|||j|q||q|jdS)N )SETsDELETEzInvalid curl cookie change: r) partition ValueErrorrr.rrRrVrGr~)r3rchangeaction separator curl_formatrrrr! update_cookies_from_curl_changess    z(Cookies.update_cookies_from_curl_changesrrFrrr-rcCs|dr|durtjdtddd}n|dr1|dus#|s#|dkr1tjd tddd}d }d}id d d |d|ddddd|dt|d|dd|dt|d|dddddddddddidd}td i|}|j|dS)!zU Set a cookie value by name. May optionally include domain and path. z __Secure-Fz9`secure` changed to True for `__Secure-` prefixed cookiesr ) stacklevelTz__Host-rz^`host` changed to True, `domain` removed, `path` changed to `/` for `__Host-` prefixed cookiesrr9rrrr:Nr;r-r6r<r=rr>rrr?r@rArBHttpOnlyrCr)rEwarningswarnrrDrrRrV)r3rrr-rrkwargsr5rrr!rTsj      z Cookies.setdefaultc Csd}d}|jD]I}|j|krP|dus|j|krP|dus |j|krP|durH||jsHt|j|sH||jkrHd|d|d|jd}t||j}|jpOd}q|durW|S|S)z Get a cookie by name. May optionally include domain and path in order to specify exactly which cookie to retrieve. Nrz!Multiple cookies exist with name=z on z and z., add domain parameter to suppress this error.)rRrr-rendswithr1rr) r3rrr-rrmatched_domainr5messagerrr!rns2      z Cookies.getcCsDi}|jD]}|dus|j|kr|dus|j|kr|j||j<q|S)z Cookies with the same name on different domains may overwrite each other, do NOT use this function as a method of serialization. N)rRr-rrr)r3r-rretr5rrr!get_dictAs  zCookies.get_dictcs\durdur|jSfdd|jD}|D] }|j|j|j|jqdS)z Delete a cookie by name. May optionally include domain and path in order to specify exactly which cookie to delete. Ncs>g|]}|jkrdus|jkrdus|jkr|qSrP)rr-r.0r5r-rrrr! ]s z"Cookies.delete..)rRrr-rr)r3rr-rremover5rrr!deletePs zCookies.deletecCsDg}|dur |||dur|dusJ|||jj|dS)z Delete all cookies. Optionally include a domain and path in order to only delete a subset of all the cookies. N)r|rRr)r3r-rargsrrr!rhs   z Cookies.clearcCs$t|}|jD]}|j|qdSrP)rrRrV)r3rOr5rrr!reus zCookies.updatecCs |||SrP)rTr3rrrrr! __setitem__zs zCookies.__setitem__cCs||}|dur t||SrP)rnrrrrr! __getitem__}s zCookies.__getitem__cCs ||SrP)r)r3rrrr! __delitem__ zCookies.__delitem__cCs t|jSrP)lenrRr2rrr!__len__rzCookies.__len__cCsdd|jDS)Ncss|]}|jVqdSrP)rrrrr! sz#Cookies.__iter__..rRr2rrr!__iter__szCookies.__iter__cCs|jD]}dSdS)NTFr)r3_rrr!__bool__s zCookies.__bool__cCs"ddd|jD}d|dS)Nz, c Ss*g|]}d|jd|jd|jdqS)z)rrr-rrrr!rsz$Cookies.__repr__..z )r0rR)r3 cookies_reprrrr!__repr__s  zCookies.__repr__rP)rrF)NNN)NN)#rHrIrJ__doc__r CookieTypesrXrFrbrir1rwrUrrrrrNrrTrnrrrrerrrr+rr rrDrrrrrr!rs    1 (   )&__all__rerfryr contextlibr dataclassesrhttp.cookiejarrr http.cookiesrtypingrr collections.abcr r urllib.parser utilsrerrorsrrrFr1rUtuplerrcompileASCIIrorsrrrrr!s(     $e