Skip to content

Commit 8f1a6be

Browse files
committed
Modifying to recognized deployment & venue
1 parent 5b1acea commit 8f1a6be

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

airflow/dags/sbg_L1_to_L2_e2e_cwl_step_by_step_dag.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from datetime import datetime
1212

1313
import boto3
14+
from airflow.models import Variable
1415
from airflow.models.baseoperator import chain
1516
from airflow.models.param import Param
1617
from airflow.operators.python import PythonOperator
@@ -35,7 +36,8 @@
3536
# SSM keys for venue-dependent parameters
3637
UNITY_DAPA_CLIENT_ID = "/unity/shared-services/dapa/client-id"
3738
UNITY_DAPA_API_URL = "/unity/shared-services/dapa/api-url"
38-
UNITY_OUTPUT_DATA_BUCKET = "/unity/ds/data/bucket/primary-data-bucket"
39+
UNITY_IDENTIFIER = f"""{Variable.get("unity_deployment_name")}/{Variable.get("unity_venue")}-{Variable.get("unity_counter")}"""
40+
UNITY_OUTPUT_DATA_BUCKET = f"/unity/ds/data/bucket/{UNITY_IDENTIFIER}/primary-data-bucket"
3941

4042
# Resources needed by each Task
4143
# EC2 r6a.xlarge 4vCPU 32GiB

0 commit comments

Comments
 (0)