@W-9746696 refresh token logic for falcon added

argusclient/client.py

@@ -695,6 +695,9 @@ def with_auth_token(*args, **kwargs):
             except ArgusAuthException:
                 if argus.password:
                     logging.debug("Token refresh failed, will attempt a fresh login", exc_info=True)
+                if argus.falconEnv:
+                    logging.debug("Access Token and Refresh Token expired, update token files and provide relevant path")
+                    argus.refreshToken, argus.accessToken = getTokensFromFiles()
                 else:
                     raise
         if not argus.accessToken and argus.password:
@@ -712,6 +715,18 @@ def with_auth_token(*args, **kwargs):
 
     return with_auth_token
 
+# Updates Access token and refresh token from files for mac to falcon usecase
+def getTokensFromFiles():
+    # Prompt for access token
+    filepath = raw_input("Provide path to access token file")
+    with open(filepath, 'r') as f:
+        access_token = f.read()
+
+    # Prompt for refresh token
+    filepath = raw_input("Provide path to refresh token file")
+    with open(filepath, 'r') as f:
+        refresh_token = f.read()
+    return access_token, refresh_token
 
 class ArgusServiceClient(object):
     """
@@ -763,7 +778,7 @@ class ArgusServiceClient(object):
 
     """
 
-    def __init__(self, user, password, endpoint, timeout=(10, 120), refreshToken=None, accessToken=None):
+    def __init__(self, user, password, endpoint, timeout=(10, 120), refreshToken=None, accessToken=None, falconEnv=False):
         """
         Creates a new client object to interface with the Argus RESTful API.
 
@@ -779,6 +794,8 @@ def __init__(self, user, password, endpoint, timeout=(10, 120), refreshToken=Non
         :type refreshToken: str
         :param accessToken: A token that can be used to authenticate with Argus. If a ``refreshToken`` or ``password`` is specified, the ``accessToken`` will be refreshed as and when it is needed.
         :type refreshToken: str
+        :param falconEnv: is falcon environment
+        :type falconEnv: bool
         """
         if not user:
             raise ValueError("A valid user must be specified")
@@ -788,6 +805,7 @@ def __init__(self, user, password, endpoint, timeout=(10, 120), refreshToken=Non
             raise ValueError("Need a valid Argus endpoint URL")
 
         self.user = user
+        self.falconEnv = falconEnv
         self.password = password
         self.endpoint = endpoint
         self.timeout = timeout