fix operator for sql application name

hasadna · Oct 30, 2024 · 0cec5bb · 0cec5bb
1 parent c5faa3b
commit 0cec5bb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/open_bus_pipelines/operators/cli_bash_operator.py b/open_bus_pipelines/operators/cli_bash_operator.py
@@ -20,13 +20,13 @@ class CliBashOperator(BashOperator):
 
     def __init__(self, cmd, **kwargs):
         assert not kwargs.get('bash_command')
-        kwargs['bash_command'] = '{print_dag_run}{pip_install_deps}{ENV}{STRIDE_VENV}/bin/{cmd}'.format(
+        super(CliBashOperator, self).__init__(**kwargs)
+        self.bash_command = '{print_dag_run}{pip_install_deps}{ENV}{STRIDE_VENV}/bin/{cmd}'.format(
             ENV=f'SQLALCHEMY_APPLICATION_NAME="{self.task_id}" SQLALCHEMY_APPLICATION_VERSION="$(cat {STRIDE_VENV}/open_bus_pipelines_commit.txt)" ',
             STRIDE_VENV=STRIDE_VENV,
             cmd=cmd,
             print_dag_run=get_print_dag_run(),
             pip_install_deps=get_pip_install_deps()
         )
         if OPEN_BUS_PIPELINES_ALERT_EMAILS:
-            kwargs['email'] = OPEN_BUS_PIPELINES_ALERT_EMAILS
-        super(CliBashOperator, self).__init__(**kwargs)
+            self.email = OPEN_BUS_PIPELINES_ALERT_EMAILS