remove separate credential handler

cloudacademy · Nov 19, 2023 · f01bb43 · f01bb43
1 parent 7f1c9b5
commit f01bb43
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 25 deletions.
diff --git a/src/config.py b/src/config.py
@@ -1,7 +1,9 @@
 import os
 
 CONFIG = {
-    'client_id'     : os.environ.get('AWS_CLIENT_ID'),
-    'client_secret' : os.environ.get('AWS_CLIENT_SECRET'),
-    'region_name'   : os.environ.get('AWS_REGION', 'us-west-2')
-}
+    'credentials': {
+        'credential_id': os.environ.get('AWS_CLIENT_ID'),
+        'credential_key': os.environ.get('AWS_CLIENT_SECRET'),
+    },
+    'region_id': os.environ.get('AWS_REGION', 'us-west-2'),
+}
diff --git a/src/credentials.py b/src/credentials.py
diff --git a/src/vcf.py b/src/vcf.py
@@ -2,7 +2,6 @@
 import time
 
 from config import CONFIG
-from credentials import credential_helper
 
 import boto3
 import botocore
@@ -83,21 +82,17 @@ def create_new_session(event):
       	region_name=event['region_id']
     )
 
-def timed_handler():
+def timed_handler(event, context):
     start = time.time()
 
-    handler({}, None)
+    result = handler(event, context)
 
     end = time.time()
     print(end - start)
 
     return result
 
 
-def create_new_session(event):
-    return credential_helper.create_new_session()
-
-
 if __name__ == '__main__':
-    result = timed_handler()
+    result = timed_handler(CONFIG, None)
     print(result)