From d623c42a891f2fe8a26493354af0d9e299f3c526 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Tue, 27 Aug 2024 15:19:14 +0100 Subject: refactor: add parameter for sm_secret --- src/load_lambda.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'src/load_lambda.py') diff --git a/src/load_lambda.py b/src/load_lambda.py index f08e335..11d1d70 100644 --- a/src/load_lambda.py +++ b/src/load_lambda.py @@ -49,7 +49,6 @@ def retrieve_secrets(client=None, secret_name=None): if client == None: client = session.client(service_name="secretsmanager", region_name=region_name) - try: get_secret_value_response = client.get_secret_value(SecretId=secret_name) print(get_secret_value_response) @@ -66,9 +65,12 @@ def retrieve_secrets(client=None, secret_name=None): # connect to database, slightly different way of doing it, to allow manipulation through pandas -def connect_to_db_and_return_engine(): +def connect_to_db_and_return_engine(sm_secret=None): + if sm_secret is None: + sm_secret = retrieve_secrets() + try: - secrets = json.loads(retrieve_secrets()) + secrets = json.loads(sm_secret) host = secrets["host"] port = secrets["port"] user = secrets["user"] @@ -198,5 +200,6 @@ def upload_dfs_to_database(): db_engine.dispose() return upload_status + if __name__ == "__main__": lambda_handler(None, None) -- cgit v1.2.3