Add handling of OperationalError for better mysql support

UW-Macrostrat · Jul 29, 2024 · 09cdc8e · 09cdc8e
1 parent 20c14b5
commit 09cdc8e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/database/macrostrat/database/utils.py b/database/macrostrat/database/utils.py
@@ -17,6 +17,7 @@
     InternalError,
     InvalidRequestError,
     ProgrammingError,
+    OperationalError
 )
 from sqlalchemy.orm import sessionmaker
 from sqlalchemy.schema import Table
@@ -309,7 +310,7 @@ def _run_sql(connectable, sql, params=None, **kwargs):
             elif hasattr(connectable, "commit"):
                 connectable.commit()
             pretty_print(sql_text, dim=True)
-        except (ProgrammingError, IntegrityError, InternalError) as err:
+        except (ProgrammingError, IntegrityError, InternalError, OperationalError) as err:
             _err = str(err.orig).strip()
             dim = "already exists" in _err
             if trans is not None:

diff --git a/database/pyproject.toml b/database/pyproject.toml
@@ -3,7 +3,7 @@ authors = ["Daven Quinn <[email protected]>"]
 description = "A SQLAlchemy-based database toolkit."
 name = "macrostrat.database"
 packages = [{ include = "macrostrat" }]
-version = "3.3.0"
+version = "3.3.1"
 
 [tool.poetry.dependencies]
 GeoAlchemy2 = "^0.14.0"