Modify db clean to also catch the ProgrammingError exception (#23699)

(cherry picked from commit a80b2fc)
apache · May 21, 2022 · dd22e54 · dd22e54
1 parent d7092b7
commit dd22e54
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/airflow/utils/db_cleanup.py b/airflow/utils/db_cleanup.py
@@ -27,7 +27,7 @@
 
 from pendulum import DateTime
 from sqlalchemy import and_, false, func
-from sqlalchemy.exc import OperationalError
+from sqlalchemy.exc import OperationalError, ProgrammingError
 
 from airflow.cli.simple_table import AirflowConsole
 from airflow.jobs.base_job import BaseJob
@@ -260,7 +260,7 @@ def __enter__(self):
         return self
 
     def __exit__(self, exctype, excinst, exctb):
-        caught_error = exctype is not None and issubclass(exctype, OperationalError)
+        caught_error = exctype is not None and issubclass(exctype, (OperationalError, ProgrammingError))
         if caught_error:
             logger.warning("Table %r not found.  Skipping.", self.table)
         return caught_error