a w=icM@sdZddlmZddlZddlZddlZzddlmZWneyVddl mZYn0ddl m Z m Z ddl Z ddlZddlZddlZddlZddlZeeZGdddeZGdd d eZGd d d ejjZGd d d eZdS)akOAuth 2.0 Authorization Flow This module provides integration with `requests-oauthlib`_ for running the `OAuth 2.0 Authorization Flow`_ and acquiring user credentials. Here's an example of using :class:`Flow` with the installed application authorization flow:: from google_auth_oauthlib.flow import Flow # Create the flow using the client secrets file from the Google API # Console. flow = Flow.from_client_secrets_file( 'path/to/client_secrets.json', scopes=['profile', 'email'], redirect_uri='urn:ietf:wg:oauth:2.0:oob') # Tell the user to go to the authorization URL. auth_url, _ = flow.authorization_url(prompt='consent') print('Please go to this URL: {}'.format(auth_url)) # The user will get an authorization code. This code is used to get the # access token. code = input('Enter the authorization code: ') flow.fetch_token(code=code) # You can use flow.credentials, or you can just get a requests session # using flow.authorized_session. session = flow.authorized_session() print(session.get('https://www.googleapis.com/userinfo/v2/me').json()) This particular flow can be handled entirely by using :class:`InstalledAppFlow`. .. _requests-oauthlib: http://requests-oauthlib.readthedocs.io/en/stable/ .. _OAuth 2.0 Authorization Flow: https://tools.ietf.org/html/rfc6749#section-1.2 )urlsafe_b64encodeN) SystemRandom) ascii_lettersdigitsc@speZdZdZdddZeddZedd Zed d Z e j d d Z d dZ ddZ eddZ ddZdS)FlowaOAuth 2.0 Authorization Flow This class uses a :class:`requests_oauthlib.OAuth2Session` instance at :attr:`oauth2session` to perform all of the OAuth 2.0 logic. This class just provides convenience methods and sane defaults for doing Google's particular flavors of OAuth 2.0. Typically you'll construct an instance of this flow using :meth:`from_client_secrets_file` and a `client secrets file`_ obtained from the `Google API Console`_. .. _client secrets file: https://developers.google.com/identity/protocols/oauth2/web-server #creatingcred .. _Google API Console: https://console.developers.google.com/apis/credentials NFcCs,||_|||_||_||_||_||_dS)a Args: oauth2session (requests_oauthlib.OAuth2Session): The OAuth 2.0 session from ``requests-oauthlib``. client_type (str): The client type, either ``web`` or ``installed``. client_config (Mapping[str, Any]): The client configuration in the Google `client secrets`_ format. redirect_uri (str): The OAuth 2.0 redirect URI if known at flow creation time. Otherwise, it will need to be set using :attr:`redirect_uri`. code_verifier (str): random string of 43-128 chars used to verify the key exchange.using PKCE. autogenerate_code_verifier (bool): If true, auto-generate a code_verifier. .. _client secrets: https://github.com/googleapis/google-api-python-client/blob /main/docs/client-secrets.md N) client_type client_config oauth2session redirect_uri code_verifierautogenerate_code_verifier)selfr rrr r r rj/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/google_auth_oauthlib/flow.py__init__`s  z Flow.__init__c Kstd|vrd}nd|vrd}ntd|dd}|dd}tjj||fi|\}}|dd}|||||||S)a(Creates a :class:`requests_oauthlib.OAuth2Session` from client configuration loaded from a Google-format client secrets file. Args: client_config (Mapping[str, Any]): The client configuration in the Google `client secrets`_ format. scopes (Sequence[str]): The list of scopes to request during the flow. kwargs: Any additional parameters passed to :class:`requests_oauthlib.OAuth2Session` Returns: Flow: The constructed Flow instance. Raises: ValueError: If the client configuration is not in the correct format. .. _client secrets: https://github.com/googleapis/google-api-python-client/blob/main/docs/client-secrets.md Zweb installedz2Client secrets must be for a web or installed app.r Nr r ) ValueErrorpopgoogle_auth_oauthlibhelpersZsession_from_client_configget) clsrscopeskwargsrr r sessionr rrrfrom_client_configs0   zFlow.from_client_configcKsJt|d}t|}Wdn1s*0Y|j|fd|i|S)aCreates a :class:`Flow` instance from a Google client secrets file. Args: client_secrets_file (str): The path to the client secrets .json file. scopes (Sequence[str]): The list of scopes to request during the flow. kwargs: Any additional parameters passed to :class:`requests_oauthlib.OAuth2Session` Returns: Flow: The constructed Flow instance. rNr)openjsonloadr)rZclient_secrets_filerrZ json_filerrrrfrom_client_secrets_files (zFlow.from_client_secrets_filecCs|jjS)zXThe OAuth 2.0 redirect URI. Pass-through to ``self.oauth2session.redirect_uri``.r r r rrrr szFlow.redirect_uricCs ||j_dSNr!)r valuerrrr sc  s|dd|jrJttdtfddtddD}d||_|jrt }| t |j| }t|}|d d}|d ||d d |jj|jd fi|\}}||fS)ahGenerates an authorization URL. This is the first step in the OAuth 2.0 Authorization Flow. The user's browser should be redirected to the returned URL. This method calls :meth:`requests_oauthlib.OAuth2Session.authorization_url` and specifies the client configuration's authorization URI (usually Google's authorization server) and specifies that "offline" access is desired. This is required in order to obtain a refresh token. Args: kwargs: Additional arguments passed through to :meth:`requests_oauthlib.OAuth2Session.authorization_url` Returns: Tuple[str, str]: The generated authorization URL and state. The user must visit the URL to complete the flow. The state is used when completing the flow to verify that the request originated from your application. If your application is using a different :class:`Flow` instance to obtain the token, you will need to specify the ``state`` when constructing the :class:`Flow`. Z access_typeZofflinez-._~csg|]}qSr)choice).0_charsZrndrr z*Flow.authorization_url..r=code_challengeZcode_challenge_methodZS256Zauth_uri) setdefaultr rrrrangejoinr hashlibsha256updatestrencodedigestrdecodesplitr authorization_urlr) r rZrandom_verifierZ code_hashZunencoded_challengeZ b64_challenger/urlstaterr(rr;s(      zFlow.authorization_urlcKs:|d|jd|d|j|jj|jdfi|S)aCompletes the Authorization Flow and obtains an access token. This is the final step in the OAuth 2.0 Authorization Flow. This is called after the user consents. This method calls :meth:`requests_oauthlib.OAuth2Session.fetch_token` and specifies the client configuration's token URI (usually Google's token server). Args: kwargs: Arguments passed through to :meth:`requests_oauthlib.OAuth2Session.fetch_token`. At least one of ``code`` or ``authorization_response`` must be specified. Returns: Mapping[str, str]: The obtained tokens. Typically, you will not use return value of this function and instead and use :meth:`credentials` to obtain a :class:`~google.auth.credentials.Credentials` instance. Z client_secretr Z token_uri)r0rr r fetch_token)r rrrrr>szFlow.fetch_tokencCstj|j|jS)aReturns credentials from the OAuth 2.0 session. :meth:`fetch_token` must be called before accessing this. This method constructs a :class:`google.oauth2.credentials.Credentials` class using the session's token and the client config. Returns: google.oauth2.credentials.Credentials: The constructed credentials. Raises: ValueError: If there is no access token in the session. )rrZcredentials_from_sessionr rr"rrr credentials szFlow.credentialscCstjjj|jS)aReturns a :class:`requests.Session` authorized with credentials. :meth:`fetch_token` must be called before this method. This method constructs a :class:`google.auth.transport.requests.AuthorizedSession` class using this flow's :attr:`credentials`. Returns: google.auth.transport.requests.AuthorizedSession: The constructed session. )googleauth transportrequestsZAuthorizedSessionr?r"rrrauthorized_session2s zFlow.authorized_session)NNF)__name__ __module__ __qualname____doc__r classmethodrr propertyr setterr;r>r?rDrrrrrMs$ & 3   - rc@sDeZdZdZdZdZdZdZeefddZdd eed d fd d Z d S)InstalledAppFlowaAuthorization flow helper for installed applications. This :class:`Flow` subclass makes it easier to perform the `Installed Application Authorization Flow`_. This flow is useful for local development or applications that are installed on a desktop operating system. This flow has two strategies: The console strategy provided by :meth:`run_console` and the local server strategy provided by :meth:`run_local_server`. Example:: from google_auth_oauthlib.flow import InstalledAppFlow flow = InstalledAppFlow.from_client_secrets_file( 'client_secrets.json', scopes=['profile', 'email']) flow.run_local_server() session = flow.authorized_session() profile_info = session.get( 'https://www.googleapis.com/userinfo/v2/me').json() print(profile_info) # {'name': '...', 'email': '...', ...} Note that these aren't the only two ways to accomplish the installed application flow, they are just the most common ways. You can use the :class:`Flow` class to perform the same flow with different methods of presenting the authorization URL to the user or obtaining the authorization response, such as using an embedded web view. .. _Installed Application Authorization Flow: https://github.com/googleapis/google-api-python-client/blob/main/docs/oauth-installed.md zurn:ietf:wg:oauth:2.0:oobz:Please visit this URL to authorize this application: {url}zEnter the authorization code: zAThe authentication flow has completed. You may close this window.cKsR|dd|j|_|jfi|\}}t|j|dt|}|j|d|jS)azRun the flow using the console strategy. The console strategy instructs the user to open the authorization URL in their browser. Once the authorization is complete the authorization server will give the user a code. The user then must copy & paste this code into the application. The code is then exchanged for a token. Args: authorization_prompt_message (str): The message to display to tell the user to navigate to the authorization URL. authorization_code_message (str): The message to display when prompting the user for the authorization code. kwargs: Additional keyword arguments passed through to :meth:`authorization_url`. Returns: google.oauth2.credentials.Credentials: The OAuth 2.0 credentials for the user. promptZconsentr<)code) r0_OOB_REDIRECT_URIr r;printformatinputr>r?)r authorization_prompt_messageZauthorization_code_messagerauth_urlr'rOrrr run_consolexs  zInstalledAppFlow.run_console localhostiTcKst|}dtjj_tjj|||td} |r.dnd} | || j|_ |j fi|\} } |rjt j | dddt |j| d| |jd d } |j| d | |jS) aRun the flow using the server strategy. The server strategy instructs the user to open the authorization URL in their browser and will attempt to automatically open the URL for them. It will start a local web server to listen for the authorization response. Once authorization is complete the authorization server will redirect the user's browser to the local web server. The web server will get the authorization code from the response and shutdown. The code is then exchanged for a token. Args: host (str): The hostname for the local redirect server. This will be served over http, not https. port (int): The port for the local redirect server. authorization_prompt_message (str): The message to display to tell the user to navigate to the authorization URL. success_message (str): The message to display in the web browser the authorization flow is complete. open_browser (bool): Whether or not to open the authorization URL in the user's browser. redirect_uri_trailing_slash (bool): whether or not to add trailing slash when constructing the redirect_uri. Default value is True. kwargs: Additional keyword arguments passed through to :meth:`authorization_url`. Returns: google.oauth2.credentials.Credentials: The OAuth 2.0 credentials for the user. F)Z handler_classz http://{}:{}/z http://{}:{}T)newZ autoraiserNhttphttps)authorization_response)_RedirectWSGIAppwsgiref simple_serverZ WSGIServerallow_reuse_addressZ make_server_WSGIRequestHandlerrRZ server_portr r; webbrowserrrQhandle_requestlast_request_urireplacer> server_closer?)r hostportrTsuccess_messageZ open_browserZredirect_uri_trailing_slashrZwsgi_appZ local_serverZredirect_uri_formatrUr'r\rrrrun_local_servers"'   z!InstalledAppFlow.run_local_serverN) rErFrGrHrPZ_DEFAULT_AUTH_PROMPT_MESSAGEZ_DEFAULT_AUTH_CODE_MESSAGEZ_DEFAULT_WEB_SUCCESS_MESSAGErVrjrrrrrL@s"( )rLc@seZdZdZddZdS)razWCustom WSGIRequestHandler. Uses a named logger instead of printing to stderr. cGstj|g|RdSr#)_LOGGERinfo)r rRargsrrr log_messagesz_WSGIRequestHandler.log_messageN)rErFrGrHrnrrrrrasrac@s eZdZdZddZddZdS)r]zwWSGI app to handle the authorization redirect. Stores the request URI and displays the given success message. cCsd|_||_dS)z Args: success_message (str): The message to display in the web browser the authorization flow is complete. N)rd_success_message)r rirrrrsz_RedirectWSGIApp.__init__cCs(|ddgtj||_|jdgS)a WSGI Callable. Args: environ (Mapping[str, Any]): The WSGI environment. start_response (Callable[str, list]): The WSGI start_response callable. Returns: Iterable[bytes]: The response body. z200 OK)z Content-typeztext/plain; charset=utf-8zutf-8)r^util request_urirdror7)r environZstart_responserrr__call__s z_RedirectWSGIApp.__call__N)rErFrGrHrrsrrrrr]s r]) rHbase64rr3rloggingZsecretsr ImportErrorrandomstringrrrbZwsgiref.simple_serverr^Z wsgiref.utilZgoogle.auth.transport.requestsr@Zgoogle.oauth2.credentialsZgoogle_auth_oauthlib.helpersr getLoggerrErkobjectrrLr_ZWSGIRequestHandlerrar]rrrrs,'   t&