Refactor: Add annoations to lftools.oauth2_helper

Issue: RELENG-4933 Signed-off-by: Andrew Grimberg <[email protected]> Change-Id: I5d80203fa91196de29e77c55721150668545aeee
lfit · Oct 9, 2023 · 92af662 · 92af662
1 parent 1ae25fc
commit 92af662
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/lftools/oauth2_helper.py b/lftools/oauth2_helper.py
@@ -1,32 +1,34 @@
 # SPDX-License-Identifier: EPL-1.0
 ##############################################################################
-# Copyright (c) 2019 The Linux Foundation and others.
+# 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():
+def oauth_helper() -> Tuple[str, str]:
     """Helper script to get access_token for lfid api."""
     logging.getLogger("oauth2client").setLevel(logging.ERROR)
-    client_id = config.get_setting("lfid", "clientid")
-    client_secret = config.get_setting("lfid", "client_secret")
-    refresh_token = config.get_setting("lfid", "refresh_token")
-    token_uri = config.get_setting("lfid", "token_uri")
-    url = config.get_setting("lfid", "url")
+    client_id: str = config.get_setting("lfid", "clientid")
+    client_secret: str = config.get_setting("lfid", "client_secret")
+    refresh_token: str = config.get_setting("lfid", "refresh_token")
+    token_uri: str = config.get_setting("lfid", "token_uri")
+    url: str = config.get_setting("lfid", "url")
 
-    credentials = client.OAuth2Credentials(
+    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,
@@ -36,5 +38,5 @@ def oauth_helper():
         user_agent=None,
     )
     credentials.refresh(httplib2.Http())
-    access_token = credentials.access_token
+    access_token: str = credentials.access_token
     return access_token, url