make JSONFileCache a config value, defaulting to the functions CACHE_DIR

----------------------------- Co-authored-by: Leslie VanDeMark <[email protected]>
singer-io · Feb 5, 2025 · 0d5bdf5 · 0d5bdf5
1 parent 0dbdee4
commit 0d5bdf5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tap_s3_csv/s3.py b/tap_s3_csv/s3.py
@@ -121,6 +121,7 @@ def setup_aws_client_with_proxy(config):
     proxy_role_arn = "arn:aws:iam::{}:role/{}".format(config['proxy_account_id'].replace('-', ''),
                                                           config['proxy_role_name'])
     cust_role_arn = "arn:aws:iam::{}:role/{}".format(config['account_id'].replace('-', ''), config['role_name'])
+    credentials_cache_path = config.get("credentials_cache_path", JSONFileCache.CACHE_DIR)
 
     # Step 1: Assume Role in Account Proxy and set up refreshable session
     session_proxy = Session()
@@ -132,7 +133,7 @@ def setup_aws_client_with_proxy(config):
             'DurationSeconds': 3600,
             'RoleSessionName': 'ProxySession'
         },
-        cache=JSONFileCache("/tmp")
+        cache=JSONFileCache(credentials_cache_path)
     )
 
     # Refreshable credentials for Account Proxy
@@ -153,7 +154,7 @@ def setup_aws_client_with_proxy(config):
             'RoleSessionName': 'TapS3CSVCustSession',
             'ExternalId': config['external_id']
         },
-        cache=JSONFileCache("/tmp")
+        cache=JSONFileCache(credentials_cache_path)
     )
 
     # Set up refreshable session for Customer Account