Set correct session state when encountering EntityNotFound

aws-samples · Jan 31, 2025 · f8011ea · f8011ea
1 parent d8259f9
commit f8011ea
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dbt/adapters/glue/gluedbapi/connection.py b/dbt/adapters/glue/gluedbapi/connection.py
@@ -356,6 +356,12 @@ def state(self) -> str:
             session = response.get("Session", {})
             self._state = session.get("Status")
         except Exception as e:
+            if isinstance(e, botocore.exceptions.ClientError):
+                if e.response['Error']['Code'] == 'EntityNotFoundException':
+                    logger.debug(f"Session {self.session_id} not found")
+                    logger.debug(e)
+                    self._state = None
+                    return self._state
             logger.debug(f"Error while checking state of session {self.session_id}")
             logger.debug(e)
             self._state = GlueSessionState.STOPPED