o 'ȋh+@sdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZGd d d eZ     d d dZddZd!ddZd"ddZd#ddZddZddZGdddZGdddeZdS)$z oauthlib.oauth2.rfc6749.tokens ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ This module contains methods for adding two types of access tokens to requests. - Bearer https://tools.ietf.org/html/rfc6750 - MAC https://tools.ietf.org/html/draft-ietf-oauth-v2-http-mac-01 N) b2a_base64)urlparse)common)add_params_to_qsadd_params_to_uri)utilscsreZdZdfdd ZeddZeddZedd Zed d Zed d Z eddZ eddZ Z S) OAuth2TokenNcsnt|d|_|drtt|d|_|dur1tt||_|jdur/|j|_dSdS|j|_dS)Nscope)super__init__ _new_scopegetsetr scope_to_list _old_scope)selfparams old_scope __class__a/var/www/html/aiguide_backend/venv/lib/python3.10/site-packages/oauthlib/oauth2/rfc6749/tokens.pyr s     zOAuth2Token.__init__cCs |j|jkSN)r rrrrr scope_changed& zOAuth2Token.scope_changedcC t|jSr)r list_to_scoperrrrrr*rzOAuth2Token.old_scopecC t|jSr)listrrrrr old_scopes. zOAuth2Token.old_scopescCrr)rrr rrrrr 2rzOAuth2Token.scopecCrr)r r rrrrscopes6r"zOAuth2Token.scopescCt|j|jSr)r rr rrrrmissing_scopes:zOAuth2Token.missing_scopescCr$r)r r rrrrradditional_scopes>r&zOAuth2Token.additional_scopesr) __name__ __module__ __qualname__r propertyrrr!r r#r%r' __classcell__rrrrr s       r hmac-sha-1c Cs.|}t|\} } |dkrtj} n|dkrtj} ntd| dkr5|p3dt | t }nt }t }t |\}}}}}}|rO|d|n|}|durn| dkrn|d}t| |dd d}nd }g}| dkr|||n |||||||||| || | dkr||||pd d |d }t|tr|d}t||d| }t|dd d}g}|d || dkr|d ||d||r|d||r|d||d||p i}d||d<|S)a_Add an `MAC Access Authentication`_ signature to headers. Unlike OAuth 1, this HMAC signature does not require inclusion of the request payload/body, neither does it use a combination of client_secret and token_secret but rather a mac_key provided together with the access token. Currently two algorithms are supported, "hmac-sha-1" and "hmac-sha-256", `extension algorithms`_ are not supported. Example MAC Authorization header, linebreaks added for clarity Authorization: MAC id="h480djs93hd8", nonce="1336363200:dj83hs9s", mac="bhCQXTVyfj5cmA9uKkPFx1zeOXM=" .. _`MAC Access Authentication`: https://tools.ietf.org/html/draft-ietf-oauth-v2-http-mac-01 .. _`extension algorithms`: https://tools.ietf.org/html/draft-ietf-oauth-v2-http-mac-01#section-7.1 :param token: :param uri: Request URI. :param key: MAC given provided by token endpoint. :param http_method: HTTP Request method. :param nonce: :param headers: Request headers as a dictionary. :param body: :param ext: :param hash_algorithm: HMAC algorithm provided by token endpoint. :param issue_time: Time when the MAC credentials were issued (datetime). :param draft: MAC authentication specification version. :return: headers dictionary with the authorization field added. r.z hmac-sha-256zunknown hash algorithmrz{}:{}?Nzutf-8r- z MAC id="%s"zts="%s"z nonce="%s"z bodyhash="%s"zext="%s"zmac="%s"z, Authorization)upperr host_from_urilowerhashlibsha1sha256 ValueErrorformat generate_agergenerate_noncegenerate_timestamprencoderdigestdecodeappendjoin isinstancestrhmacnew)tokenurikey http_methodnonceheadersbodyexthash_algorithm issue_timedrafthostporthtsschnetpathparqueryfra request_uribodyhashbase base_stringsignheaderrrrprepare_mac_headerCsb(              rbcCt|d|fgS)aAdd a `Bearer Token`_ to the request URI. Not recommended, use only if client can't use authorization header or body. http://www.example.com/path?access_token=h480djs93hd8 .. _`Bearer Token`: https://tools.ietf.org/html/rfc6750 :param token: :param uri: access_token)r)rGrHrrrprepare_bearer_uris recCs|pi}d||d<|S)zAdd a `Bearer Token`_ to the request URI. Recommended method of passing bearer tokens. Authorization: Bearer h480djs93hd8 .. _`Bearer Token`: https://tools.ietf.org/html/rfc6750 :param token: :param headers: z Bearer %sr2r)rGrLrrrprepare_bearer_headerss rfcCrc)zAdd a `Bearer Token`_ to the request body. access_token=h480djs93hd8 .. _`Bearer Token`: https://tools.ietf.org/html/rfc6750 :param token: :param body: rd)r)rGrMrrrprepare_bearer_bodys rgFcCstS)zp :param request: OAuthlib request. :type request: oauthlib.common.Request :param refresh_token: )rgenerate_token)request refresh_tokenrrrrandom_token_generatorsrkc sfdd}|S)z :param private_pem: cs|_t|Sr)claimsrgenerate_signed_token)rikwargs private_pemrrsigned_token_generators z6signed_token_generator..signed_token_generatorr)rprorqrrnrrqsrqcCsPd}d|jvr#|jd}t|dkr!|ddkr!|d}|S|j}|S)z Helper function to extract a token from the request header. :param request: OAuthlib request. :type request: oauthlib.common.Request :return: Return the token or None if the Authorization header is malformed. Nr2rbearerr)rLrsplitlenr5rd)rirG split_headerrrrget_token_from_headers rwc@s*eZdZdZd ddZddZddZd S) TokenBaserFcCtd)N&Subclasses must implement this method.NotImplementedError)rrirjrrr__call__szTokenBase.__call__cCryb :param request: OAuthlib request. :type request: oauthlib.common.Request rzr{rrirrrvalidate_requestzTokenBase.validate_requestcCryr~r{rrrr estimate_type rzTokenBase.estimate_typeNF)r(r)r* __slots__r}rrrrrrrxs   rxc@s8eZdZdZ  d ddZd ddZdd Zd d ZdS) BearerToken)request_validatortoken_generatorrefresh_token_generator expires_inNcCs*||_|pt|_|p |j|_|pd|_dS)Ni)rrkrrr)rrrrrrrrr s  zBearerToken.__init__FcKsd|vr tdtt|jr||n|j}||_|||dd}|jdur0d|j|d<|rH|jrA|j |sA|j|d<n| ||d<| |j pNit|S) z Create a BearerToken, by default without refresh token. :param request: OAuthlib request. :type request: oauthlib.common.Request :param refresh_token: save_tokenzx`save_token` has been deprecated, it was not called internally.If you do, call `request_validator.save_token()` instead.Bearer)rdr token_typeN r rj)warningswarnDeprecationWarningcallablerrr#rBrjrrotate_refresh_tokenrupdateextra_credentialsr )rrirjrorrGrrr create_token$s(   zBearerToken.create_tokencCst|}|j||j|S)r)rwrvalidate_bearer_tokenr#)rrirGrrrrKszBearerToken.validate_requestcCs6|jdddddkrdS|jdurdSdS) rr2r-rrrs N)rLrrtr5rdrrrrrTs  zBearerToken.estimate_type)NNNNr)r(r)r*rr rrrrrrrrs ' r)NNNr-r.Nrr)r-r)__doc__r6rErbinasciir urllib.parseroauthlibroauthlib.commonrrr-rdictr rbrerfrgrkrqrwrxrrrrrs4     . j