add sql application name

hasadna · Oct 30, 2024 · 28aa4ef · 28aa4ef
1 parent db11064
commit 28aa4ef
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/open_bus_pipelines/operators/cli_bash_operator.py b/open_bus_pipelines/operators/cli_bash_operator.py
@@ -20,7 +20,8 @@ class CliBashOperator(BashOperator):
 
     def __init__(self, cmd, **kwargs):
         assert not kwargs.get('bash_command')
-        kwargs['bash_command'] = '{print_dag_run}{pip_install_deps}{STRIDE_VENV}/bin/{cmd}'.format(
+        kwargs['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(),