diff --git a/dagger/pipeline/pipeline.py b/dagger/pipeline/pipeline.py index 32c45a2..d0d28ee 100644 --- a/dagger/pipeline/pipeline.py +++ b/dagger/pipeline/pipeline.py @@ -1,4 +1,4 @@ -from datetime import datetime +from datetime import datetime, timedelta from os.path import join, relpath from dagger import conf @@ -71,6 +71,7 @@ def __init__(self, directory: str, config: dict): self._alerts = [] self._alert_factory = AlertFactory() self.process_alerts(config["alerts"] or []) + self.process_dag_parameters(self._parameters) @property def directory(self): @@ -129,3 +130,8 @@ def process_alerts(self, alert_configs): alert_type, join(self.directory, "pipeline.yaml"), alert_config ) ) + def process_dag_parameters(self, dag_parameters):#TODO: create long term fix for this + if dag_parameters is not None: + for key, value in dag_parameters.items(): + if key == 'dagrun_timeout': + self._parameters[key] = eval(value)