Fix path error

CityofToronto · Sep 29, 2023 · 7541ef6 · 7541ef6
1 parent f198148
commit 7541ef6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/dags/pull_here.py b/dags/pull_here.py
@@ -14,9 +14,12 @@
 from airflow.providers.postgres.hooks.postgres import PostgresHook
 from airflow.models import Variable 
 
-repo_path = os.path.abspath(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))
-sys.path.insert(0, repo_path)
-from dags.dag_functions import task_fail_slack_alert
+try:
+    repo_path = os.path.abspath(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))
+    sys.path.insert(0, repo_path)
+    from dags.dag_functions import task_fail_slack_alert
+except:
+    raise ImportError("Cannot import slack alert functions")
 
 dag_name = 'pull_here'