From f8011ea95c8db33f4965b853dd08585be7747ee9 Mon Sep 17 00:00:00 2001 From: "Firth, Charlie" Date: Fri, 31 Jan 2025 17:12:05 +0000 Subject: [PATCH] Set correct session state when encountering EntityNotFound --- dbt/adapters/glue/gluedbapi/connection.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/dbt/adapters/glue/gluedbapi/connection.py b/dbt/adapters/glue/gluedbapi/connection.py index 08bbacc..3820112 100644 --- 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