sql-catalog: Fix Postgres initialization when tables do not exist (#356)

* fix postgres catalog initialization when tables do not exist * Format --------- Co-authored-by: Fokko Driesprong <[email protected]>
apache · Feb 4, 2024 · a3d529b · a3d529b
1 parent a276112
commit a3d529b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pyiceberg/catalog/sql.py b/pyiceberg/catalog/sql.py
@@ -32,7 +32,7 @@
     union,
     update,
 )
-from sqlalchemy.exc import IntegrityError, NoResultFound, OperationalError
+from sqlalchemy.exc import IntegrityError, NoResultFound, OperationalError, ProgrammingError
 from sqlalchemy.orm import (
     DeclarativeBase,
     Mapped,
@@ -112,7 +112,10 @@ def _ensure_tables_exist(self) -> None:
                 stmt = select(1).select_from(table)
                 try:
                     session.scalar(stmt)
-                except OperationalError:
+                except (
+                    OperationalError,
+                    ProgrammingError,
+                ):  # sqlalchemy returns OperationalError in case of sqlite and ProgrammingError with postgres.
                     self.create_tables()
                     return