Merge pull request #32 from ghrcdaac/mlh0079-5352-private-api-lambda

Mlh0079 5352 private api lambda

Author: camposeddie

pylot/plugins/cumulus_api/main.py

@@ -73,10 +73,10 @@ def generate_parser(subparsers, action_target_dict):
     action_subparsers = cumulus_api_parser.add_subparsers(title='actions', dest='action', required=True)
     for action_k, target_v in action_target_dict.items():
         action_parser = action_subparsers.add_parser(
-                action_k, help=f'{action_k} action for the cumulus API',
-                usage=f'{action_k} <target> ',
-                description=f'{action_k} action for the cumulus API'
-            )
+            action_k, help=f'{action_k} action for the cumulus API',
+            usage=f'{action_k} <target> ',
+            description=f'{action_k} action for the cumulus API'
+        )
         target_subparsers = action_parser.add_subparsers(title='target', dest='target', required=True)
         for target_k, argument_v in target_v.items():
             plural = ''

pylot/plugins/helpers/pylot_helpers.py

@@ -31,15 +31,21 @@ def get_hash_token_file():
     @classmethod
     def get_cumulus_api_instance(cls):
         """ Get Cumulus instance with cached token"""
-        get_hashed_file_name = cls.get_hash_token_file()
-        token: str
-        tempfile = f'{mkdtemp()}/{get_hashed_file_name}'
-        if not os.path.isfile(tempfile):
-            with open(tempfile, 'w', encoding='utf-8') as _file:
-                cml = CumulusApi()
-                _file.write(cml.TOKEN)
+        if 'PRIVATE_API_LAMBDA_ARN' in os.environ:
+            print('getting non-token instance')
+            cml = CumulusApi()
+            pass
         else:
-            with open(tempfile, 'r', encoding='utf-8') as _file:
-                token = _file.readline()
-                cml = CumulusApi(token=token)
+            print('getting token instance')
+            get_hashed_file_name = cls.get_hash_token_file()
+            token: str
+            tempfile = f'{mkdtemp()}/{get_hashed_file_name}'
+            if not os.path.isfile(tempfile):
+                with open(tempfile, 'w', encoding='utf-8') as _file:
+                    cml = CumulusApi()
+                    _file.write(cml.TOKEN)
+            else:
+                with open(tempfile, 'r', encoding='utf-8') as _file:
+                    token = _file.readline()
+                    cml = CumulusApi(token=token)
         return cml

requirements.txt

@@ -1,2 +1,3 @@
-cumulus-api @ https://github.com/ghrcdaac/cumulus-api/archive/refs/tags/v2.0.0.zip
+
+cumulus-api @ https://github.com/ghrcdaac/cumulus-api/archive/refs/tags/v2.2.0.zip
 boto3==1.24.28