Skip to content

Commit b61debd

Browse files
committed
Modifying to recognized deployment & venue
1 parent 3ea2d6a commit b61debd

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

+3-1
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from datetime import datetime
1111

1212
import boto3
13+
from airflow.models import Variable
1314
from airflow.models.baseoperator import chain
1415
from airflow.models.param import Param
1516
from airflow.operators.python import PythonOperator
@@ -30,7 +31,8 @@
3031
# SSM keys for venue-dependent parameters
3132
UNITY_DAPA_CLIENT_ID = "/unity/shared-services/dapa/client-id"
3233
UNITY_DAPA_API_URL = "/unity/shared-services/dapa/api-url"
33-
UNITY_OUTPUT_DATA_BUCKET = "/unity/ds/data/bucket/primary-data-bucket"
34+
UNITY_IDENTIFIER = f"""{Variable.get("unity_deployment_name")}/{Variable.get("unity_venue")}-{Variable.get("unity_counter")}"""
35+
UNITY_OUTPUT_DATA_BUCKET = f"/unity/ds/data/bucket/{UNITY_IDENTIFIER}/primary-data-bucket"
3436

3537
# Resources needed by each Task
3638
# EC2 r6a.xlarge 4vCPU 32GiB

0 commit comments

Comments
 (0)