oauth2_helper.py

# SPDX-License-Identifier: EPL-1.0
##############################################################################
# Copyright (c) 2019, 2023 The Linux Foundation and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Eclipse Public License v1.0
# which accompanies this distribution, and is available at
# http://www.eclipse.org/legal/epl-v10.html
##############################################################################
"""Helper script to get access_token for lfid api."""
from __future__ import annotations

import logging
from typing import Tuple

import httplib2
from oauth2client import client

from lftools import config


def oauth_helper() -> Tuple[str, str]:
    """Helper script to get access_token for lfid api."""
    logging.getLogger("oauth2client").setLevel(logging.ERROR)
    client_id: str = str(config.get_setting("lfid", "clientid"))
    client_secret: str = str(config.get_setting("lfid", "client_secret"))
    refresh_token: str = str(config.get_setting("lfid", "refresh_token"))
    token_uri: str = str(config.get_setting("lfid", "token_uri"))
    url: str = str(config.get_setting("lfid", "url"))

    credentials: client.Oauth2Credentials = client.OAuth2Credentials(
        access_token=None,  # set access_token to None since we use a refresh token
        client_id=client_id,
        client_secret=client_secret,
        refresh_token=refresh_token,
        token_expiry=None,
        token_uri=token_uri,
        user_agent=None,
    )
    credentials.refresh(httplib2.Http())
    access_token: str = credentials.access_token
    return access_token, url