Added more NotFound error type (#94)

Some SQL queries will error with `Database(ucx_shydk,Some(hive_metastore)) does not exist`, `Function(hive_metastore.ucx_slgfw.ucx_tkrq6) does not exist`. We need to classify them as `NotFound` error messages --------- Co-authored-by: Serge Smertin <[email protected]>
databrickslabs · Apr 19, 2024 · 30c7d7a · 30c7d7a
1 parent 5782b23
commit 30c7d7a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/databricks/labs/lsql/core.py b/src/databricks/labs/lsql/core.py
@@ -448,6 +448,8 @@ def _raise_if_needed(status: StatementStatus):
             raise NotFound(error_message)
         if "DELTA_TABLE_NOT_FOUND" in error_message:
             raise NotFound(error_message)
+        if "does not exist" in error_message:
+            raise NotFound(error_message)
         if "DELTA_MISSING_TRANSACTION_LOG" in error_message:
             raise DataLoss(error_message)
         mapping = {