fix: avoid crashing when analyzing sqlalchemy errors (#411)

* fix: avoid crashing when analyzing sqlalchemy errors * whoops
codecov · Apr 24, 2024 · 5d85174 · 5d85174
1 parent 78b8c10
commit 5d85174
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tasks/base.py b/tasks/base.py
@@ -204,14 +204,18 @@ def _analyse_error(self, exception: SQLAlchemyError, *args, **kwargs):
         try:
             import psycopg2
 
-            if isinstance(exception.orig, psycopg2.errors.DeadlockDetected):
+            if hasattr(exception, "orig") and isinstance(
+                exception.orig, psycopg2.errors.DeadlockDetected
+            ):
                 log.exception(
                     "Deadlock while talking to database",
                     extra=dict(task_args=args, task_kwargs=kwargs),
                     exc_info=True,
                 )
                 return
-            elif isinstance(exception.orig, psycopg2.OperationalError):
+            elif hasattr(exception, "orig") and isinstance(
+                exception.orig, psycopg2.OperationalError
+            ):
                 log.warning(
                     "Database seems to be unavailable",
                     extra=dict(task_args=args, task_kwargs=kwargs),