o b@sddlmZddlmZddlmZddlmZmZddl m Z m Z m Z m Z ddgZGd ddeeZGd d d eeZGd ddeeZd S))Session)AuthBase) OAuth2Client) ClientAuth TokenAuth) OAuthErrorInvalidTokenErrorMissingTokenErrorUnsupportedTokenTypeError OAuth2Session OAuth2Authc@s eZdZdZddZddZdS)r zFSign requests for OAuth 2.0, currently only bearer token is supported.cCs|jrI|jrK|jd}|j}|jd}|r$|r$|j||ddS|jddkrF|jd}|j|dd}|jrD|j||ddSdStdSdS) N refresh_tokenZtoken_endpoint)r grant_typeZclient_credentials access_token)r)r) clienttoken is_expiredgetmetadatar fetch_token update_tokenr )selfrrurlrrrr/var/www/secure340b-portal/env/lib/python3.10/site-packages/authlib/integrations/requests_client/oauth2_session.pyensure_active_tokens   zOAuth2Auth.ensure_active_tokenc Cs`|z||j|j|j\|_|_|_W|Sty/}z dt|}t|dd}~ww)NzUnsupported token_type: {}) description) rpreparerheadersbodyKeyErrorformatstrr )rreqerrorrrrr__call__!s  zOAuth2Auth.__call__N)__name__ __module__ __qualname____doc__rr&rrrrr s c@seZdZdZddZdS)OAuth2ClientAuthzFAttaches OAuth Client Authentication to the given Request object. cCs(||j|j|j|j\|_|_|_|SN)rmethodrrr )rr$rrrr&/szOAuth2ClientAuth.__call__N)r'r(r)r*r&rrrrr+,s r+csZeZdZdZeZeZdZ      dddZ dddZ dfd d Z e d d Z ZS)r aConstruct a new OAuth 2 client requests session. :param client_id: Client ID, which you get from client registration. :param client_secret: Client Secret, which you get from registration. :param authorization_endpoint: URL of the authorization server's authorization endpoint. :param token_endpoint: URL of the authorization server's token endpoint. :param token_endpoint_auth_method: client authentication method for token endpoint. :param revocation_endpoint: URL of the authorization server's OAuth 2.0 revocation endpoint. :param revocation_endpoint_auth_method: client authentication method for revocation endpoint. :param scope: Scope that you needed to access user resources. :param redirect_uri: Redirect URI you registered as callback. :param token: A dict of token attributes such as ``access_token``, ``token_type`` and ``expires_at``. :param token_placement: The place to put token in HTTP request. Available values: "header", "body", "uri". :param update_token: A function for you to update token. It accept a :class:`OAuth2Token` as parameter. ) allow_redirectstimeoutcookiesfilesproxieshooksstreamverifycertjsonNheaderc Ks6t|tj|f|||||||||| d | dS)N) session client_id client_secrettoken_endpoint_auth_methodrevocation_endpoint_auth_methodscope redirect_urirtoken_placementr)r__init__r) rr:r;r<r=r>r?rr@rkwargsrrrrATs  zOAuth2Session.__init__cKs|j|fi|S)zAlias for fetch_token.)r)rrrBrrrfetch_access_tokenfsz OAuth2Session.fetch_access_tokenFc s<|s|dur|js t|j}tt|j||fd|i|S)zs