From 3a590da1dd6fa57333e6ec3d4873d9547a984bc9 Mon Sep 17 00:00:00 2001
From: "github-actions[bot]"
 <41898282+github-actions[bot]@users.noreply.github.com>
Date: Wed, 2 Oct 2024 16:47:19 -0400
Subject: [PATCH] Catch google `NotFound` exception as a `DbtDatabaseError`
 (#1360) (#1361)

* catch google NotFound exception as a DbtDatabaseError

(cherry picked from commit 8833c5450a5f1f59b688a5c488f27864dc02f93c)

Co-authored-by: Mike Alfare <13974384+mikealfare@users.noreply.github.com>
---
 .changes/unreleased/Fixes-20241001-193207.yaml | 7 +++++++
 dbt/adapters/bigquery/connections.py           | 4 ++++
 2 files changed, 11 insertions(+)
 create mode 100644 .changes/unreleased/Fixes-20241001-193207.yaml

diff --git a/.changes/unreleased/Fixes-20241001-193207.yaml b/.changes/unreleased/Fixes-20241001-193207.yaml
new file mode 100644
index 000000000..584445a5b
--- /dev/null
+++ b/.changes/unreleased/Fixes-20241001-193207.yaml
@@ -0,0 +1,7 @@
+kind: Fixes
+body: Catch additional database error exception, NotFound, as a DbtDatabaseError instead
+  of defaulting to a DbtRuntimeError
+time: 2024-10-01T19:32:07.304353-04:00
+custom:
+  Author: mikealfare
+  Issue: "1360"
diff --git a/dbt/adapters/bigquery/connections.py b/dbt/adapters/bigquery/connections.py
index f96bc1381..afe609a5a 100644
--- a/dbt/adapters/bigquery/connections.py
+++ b/dbt/adapters/bigquery/connections.py
@@ -254,6 +254,10 @@ def exception_handler(self, sql):
             message = "Access denied while running query"
             self.handle_error(e, message)
 
+        except google.cloud.exceptions.NotFound as e:
+            message = "Not found while running query"
+            self.handle_error(e, message)
+
         except google.auth.exceptions.RefreshError as e:
             message = (
                 "Unable to generate access token, if you're using "