o .jM6@sFUdZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z m Z ddlmZddlmZdd lmZdd lmZejrPdd lmZd Zd ed<dZd ed<d'ddZGdddZGdddZGdddeZGdddeZ Gdd d e Z!e Gd!d"d"Z"Gd#d$d$eZ#Gd%d&d&e#eZ$dS)(z] requests.auth ~~~~~~~~~~~~~ This module contains the authentication handlers for Requests. ) annotationsN) b64encode) dataclassfield)urlparse)iscoroutinefunction)extract_cookies_to_jar)parse_dict_header)PreparedRequestz!application/x-www-form-urlencodedstrCONTENT_TYPE_FORM_URLENCODEDzmultipart/form-dataCONTENT_TYPE_MULTI_PARTusername str | bytespasswordreturncCsJt|tr |d}t|tr|d}dtd||f}|S)zReturns a Basic Auth string.utf-8zBasic :) isinstancer encoderjoinstripdecode)rrauthstrrB/home/thesage/.local/lib/python3.10/site-packages/niquests/auth.py_basic_auth_strs    rc@eZdZdZdddZdS) AsyncAuthBasezABase class that all asynchronous auth implementations derive fromrr rcs tdNzAuth hooks must be callable.NotImplementedErrorselfr rrr__call__0szAsyncAuthBase.__call__Nr r rr __name__ __module__ __qualname____doc__r&rrrrr-rc@r) AuthBasez@Base class that all synchronous auth implementations derive fromr r rcCstdr!r"r$rrrr&7szAuthBase.__call__Nr'r(rrrrr.4r-r.c@s6eZdZdZdddZddd Zdd d Zd d ZdS)BearerTokenAuthz.Simple token injection in Authorization headertokenr cCs ||_dSN)r0)r%r0rrr__init__> zBearerTokenAuth.__init__rboolcCs|jt|ddkS)Nr0)r0getattrr%otherrrr__eq__AszBearerTokenAuth.__eq__cC ||k Sr1rr6rrr__ne__Dr3zBearerTokenAuth.__ne__cCsB|jjddd}t|dkrd|j|jd<|S|j|jd<|S)N r)maxsplitzBearer Authorization)r0splitlenheaders)r%r detect_token_typerrrr&Gs   zBearerTokenAuth.__call__N)r0r rr4r)r*r+r,r2r8r:r&rrrrr/;s     r/c@s6eZdZdZdddZdd d Zdd d Zd dZdS) HTTPBasicAuthz?Attaches HTTP Basic Authentication to the given Request object.rrrcCs||_||_dSr1)rrr%rrrrrr2Us zHTTPBasicAuth.__init__rr4cC(t|jt|ddk|jt|ddkgSNrrallrr5rr6rrrr8Y zHTTPBasicAuth.__eq__cCr9r1rr6rrrr:ar3zHTTPBasicAuth.__ne__cCt|j|j|jd<|S)Nr=rrrr@r$rrrr&dzHTTPBasicAuth.__call__N)rrrrrBrCrrrrrDRs     rDc@eZdZdZddZdS) HTTPProxyAuthz=Attaches HTTP Proxy Authentication to a given Request object.cCrK)NzProxy-AuthorizationrLr$rrrr&lrMzHTTPProxyAuth.__call__Nr(rrrrrOis rOc@s`eZdZUdZdZded<dZded<dZd ed <ee d Z d ed <dZ ded<dZ ded<dS)DigestAuthStatez/Container for digest auth state per task/threadFr4initr last_noncerint nonce_count)default_factoryztyping.Mapping[str, str | None]chalNz int | Nonepos num_401_calls) r)r*r+r,rQ__annotations__rSrUrdictrWrXrYrrrrrPqs     rPc@sdeZdZdZdddZdd d Zd ddZdddZddZddZ ddZ d!ddZ d!ddZ dS)"HTTPDigestAuthz@Attaches HTTP Digest Authentication to the given Request object.rr rcCs||_||_td|_dS)Ndigest_auth_state)rr contextvars ContextVar _thread_localrErrrr2szHTTPDigestAuth.__init__rNonecCs4|jd}|dus |js|jtdddSdS)NT)rQ)r`getrQsetrP)r%staterrrinit_per_thread_states z$HTTPDigestAuth.init_per_thread_statemethodurl str | Nonec s|jd}|dus J|jd}|jd}|jd}|jd}|jd}d|dur1d} n|} | dks=| dkrEd2d d } | n,| dkrQd2dd} | n | dkr]d2dd} | n| dkrid2dd} | ntd| dfdd}dur}dSd}t|}|jpd}|jr|d|j7}|jd|d|j }|d|}|}|}||j kr|j d7_ nd|_ |j d}t |j  d }|dusJ|| d 7}|t d 7}|td!7}t|dd"}| dkr|d|d|}|s|||d|}n"|d#ksd#|d$vr2|d|d|d%|}|||}ndS||_ d&|jd'|d(|d)|d*|d+ }|rU|d,|d+7}|r`|d-|d+7}|rk|d.|d+7}|ry|d/|d0|d+7}d1|S)3Nrealmnonceqop algorithmopaqueMD5zMD5-SESSxrrr cS"t|tr |d}t|SNr)rr rhashlibmd5 hexdigestrorrrmd5_utf8  z4HTTPDigestAuth.build_digest_header..md5_utf8SHAcSrprq)rr rrrsha1rtrurrrsha_utf8rwz4HTTPDigestAuth.build_digest_header..sha_utf8zSHA-256cSrprq)rr rrrsha256rtrurrr sha256_utf8rwz7HTTPDigestAuth.build_digest_header..sha256_utf8zSHA-512cSrprq)rr rrrsha512rtrurrr sha512_utf8rwz7HTTPDigestAuth.build_digest_header..sha512_utf8'z$' hashing algorithm is not supportedcs|d|S)N:r)sd hash_utf8rrsz4HTTPDigestAuth.build_digest_header../?rr08xrauth,z:auth:z username="z ", realm="z ", nonce="z", uri="z ", response=""z , opaque="z , algorithm="z , digest="z, qop="auth", nc=z , cnonce="zDigest )rorrr )r`rbrWupper ValueErrorrpathqueryrrrSrUr rtimectimeosurandomrrryrtr>)r%rfrgrdrirjrkrlrm _algorithmrvrzr|r~KDentdigp_parsedrA1A2HA1HA2ncvaluercnoncerespdignoncebitbaserrrbuild_digest_headers                  & z"HTTPDigestAuth.build_digest_headercKs,|jd}|dus J|jrd|_dSdS)z)Reset num_401_calls counter on redirects.Nr)r`rb is_redirectrY)r%r kwargsrdrrrhandle_redirects   zHTTPDigestAuth.handle_redirectcs@|jd}|dus Jd|jkrdksnd|_|S|jdur+|jj|j|jdd}d| vr|jdkr|jd7_t j d t j d }t |jd|dd |_|jIdH|IdH|j}t|j|j|j||j||j|j|jd <|jj|fi|IdH}|j|||_|Sd|_|S) z Takes the given response and tries digest-auth, if needed (async version). :rtype: requests.Response Nrwww-authenticaterRdigestdigest flagscountr=r`rb status_coderYrXrequestbodyseekr@lowerrecompile IGNORECASEr subrWcontentclosecopyr _cookiesrawprepare_cookiesrrfrg connectionsendhistoryappendr%r rrds_authpatprep_rrrrasync_handle_401s2       zHTTPDigestAuth.async_handle_401cKs,|jd}|dus Jd|jkrdksnd|_|S|jdur*|jj|j|jdd}d| vr|jdkr|jd7_t j d t j d }t |jd|dd |_|j||j}t|j|j|j||j||j|j|jd <|jj|fi|}|j|||_|Sd|_|S) zo Takes the given response and tries digest-auth, if needed. :rtype: requests.Response NrrrrrRrrrrrr=rrrrr handle_4014s0      zHTTPDigestAuth.handle_401cCs||jd}|dusJ|jr||j|j|jd<z|j |_ Wn t y2d|_ Ynw| d|j | d|jd|_|SNr=responser)rer`rbrSrrfrgr@rtellrXAttributeError register_hookrrrYr%r rdrrrr&as    zHTTPDigestAuth.__call__r4cCrFrGrHr6rrrr8xrJzHTTPDigestAuth.__eq__cCr9r1rr6rrrr:r3zHTTPDigestAuth.__ne__N)rr rr )rra)rfr rgr rrhrB) r)r*r+r,r2rerrrrr&r8r:rrrrr\}s    q -- r\c@rN)AsyncHTTPDigestAuthaAsync version of HTTPDigestAuth for use with AsyncSession. Attaches HTTP Digest Authentication to the given Request object and handles 401 responses asynchronously. Example usage:: >>> import niquests >>> auth = niquests.auth.AsyncHTTPDigestAuth('user', 'pass') >>> async with niquests.AsyncSession() as session: ... r = await session.get('https://httpbin.org/digest-auth/auth/user/pass', auth=auth) ... print(r.status_code) 200 cs||jd}|dusJ|jr||j|j|jd<zt|j j r0|j IdH|_ n|j |_ Wn t yCd|_ Ynw| d|j| d|jd|_|Sr)rer`rbrSrrfrgr@rrrrXrrrrrYrrrrr&s$      zAsyncHTTPDigestAuth.__call__Nr(rrrrrs r)rrrrrr )%r, __future__rr^rrrrrtypingbase64r dataclassesrr urllib.parser_compatrcookiesr utilsr TYPE_CHECKINGmodelsr r rZrrrr.r/rDrOrPr\rrrrrs<