init

deployments/scripts/azure/output_dag_folder.py

+import os
 import shutil
 
 class OutputDag:
 
     def __init__(self):
         self.dag_name = f"Osdu_ingest"
+        self.dag_folder_path = os.environ.get('DAG_FOLDER_PATH', './src/osdu_dags')
         print(f"dag_name: {self.dag_name}")
 
     def create(self):
@@ -11,10 +13,10 @@ class OutputDag:
         self.create_workflow_request_body()
 
     def prepare_zip_folder(self):
-        shutil.make_archive('./output_dags/dags/manifest_ingestion_dags', 'zip', './src/osdu_dags')
+        shutil.make_archive('./output_dags/dags/manifest_ingestion_dags', 'zip', self.dag_folder_path)
 
     def create_workflow_request_body(self):
-        file_name = "workflow_request_body.json"
+        file_name = os.environ.get("WORKFLOW_REQUEST_BODY_FILE", "workflow_request_body.json")
         new_file= open("./output_dags/" + file_name, "w")
         data = '[{"workflowName":"DAG_NAME_PLACEHOLDER","description":"ManifestIngestion","registrationInstructions":{"dagName":"DAG_NAME_PLACEHOLDER"}}]'
         data = data.replace("DAG_NAME_PLACEHOLDER", self.dag_name)