Adding SSM decryption

unity-sds · Apr 23, 2024 · 5d158bc · 5d158bc
1 parent afbade9
commit 5d158bc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/airflow/dags/sbg_L1_to_L2_e2e_cwl_step_by_step_dag.py b/airflow/dags/sbg_L1_to_L2_e2e_cwl_step_by_step_dag.py
@@ -121,7 +121,11 @@ def setup(ti=None, **context):
     # retrieve the venue-dependent parameters from SSM
     # FIXME: must coordinate SSM keys with U-CS, U-DS
     ssm_client = boto3.client('ssm', region_name='us-west-2')
-    ssm_response = ssm_client.get_parameters(Names=[UNITY_DAPA_CLIENT_ID, UNITY_DAPA_API_URL, UNITY_OUTPUT_DATA_BUCKET])
+    ssm_response = ssm_client.get_parameters(Names=[UNITY_DAPA_CLIENT_ID,
+                                                    UNITY_DAPA_API_URL,
+                                                    UNITY_OUTPUT_DATA_BUCKET],
+                                             WithDecryption=True)
+    print(ssm_response)
 
     # dictionary containing venue dependent parameters common to all Tasks
     # this dictionary is merged into each Task specific dictionary