velocity-python 0.0.204__tar.gz → 0.0.205__tar.gz

{velocity_python-0.0.204 → velocity_python-0.0.205}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: velocity-python
-Version: 0.0.204
+Version: 0.0.205
 Summary: A rapid application development library for interfacing with data storage
 Author-email: Velocity Team <info@codeclubs.org>
 License-Expression: MIT

{velocity_python-0.0.204 → velocity_python-0.0.205}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "velocity-python"
-version = "0.0.204"
+version = "0.0.205"
 authors = [
   { name="Velocity Team", email="info@codeclubs.org" },
 ]

{velocity_python-0.0.204 → velocity_python-0.0.205}/src/velocity/__init__.py

@@ -1,4 +1,4 @@
-__version__ = version = "0.0.204"
+__version__ = version = "0.0.205"
 
 from . import aws
 from . import db

{velocity_python-0.0.204 → velocity_python-0.0.205}/src/velocity/aws/handlers/base_handler.py

@@ -221,17 +221,61 @@ class BaseHandler:
             local_context: The context object
             exception: The exception that occurred
         """
+        # Always log the original exception so it isn't lost if the error-path
+        # (e.g. job-status updates) fails due to a transient DB disconnect.
+        logger.exception(
+            "Unhandled exception during action execution",
+            extra={
+                "handler": self.__class__.__name__,
+                "action": getattr(local_context, "action", lambda: None)(),
+                "tx_present": tx is not None,
+            },
+        )
+
         if hasattr(self, "onError"):
-            self.onError(
-                tx,
-                local_context,
-                exc=exception.__class__.__name__,
-                tb=traceback.format_exc(),
-            )
+            try:
+                self.onError(
+                    tx,
+                    local_context,
+                    exc=exception.__class__.__name__,
+                    tb=traceback.format_exc(),
+                )
+            except Exception as on_error_exc:
+                if self._is_transient_db_disconnect(on_error_exc):
+                    logger.warning(
+                        "onError failed due to transient DB disconnect; continuing",
+                        exc_info=True,
+                        extra={
+                            "handler": self.__class__.__name__,
+                            "action": getattr(local_context, "action", lambda: None)(),
+                            "original_exc": exception.__class__.__name__,
+                            "onerror_exc": on_error_exc.__class__.__name__,
+                        },
+                    )
+                    return
+                raise
         else:
             # Re-raise if no error handler is defined
             raise exception
 
+    @staticmethod
+    def _is_transient_db_disconnect(exc: Exception) -> bool:
+        """Return True if an exception looks like a transient DB disconnect.
+
+        This is intentionally message-based so it works even if the originating
+        exception type comes from psycopg2 / SQLAlchemy / wrapped Velocity errors.
+        """
+        try:
+            from velocity.db import exceptions as db_exceptions
+
+            if isinstance(exc, db_exceptions.DbConnectionError):
+                return True
+        except Exception:
+            # Avoid import-time coupling; fall back to message matching.
+            pass
+
+        return False
+
     def log(
         self, tx, message: str, function: Optional[str] = None, level: str = "info"
     ):

{velocity_python-0.0.204 → velocity_python-0.0.205}/src/velocity/db/core/engine.py

@@ -198,7 +198,9 @@ class Engine:
                         # Transient disconnects can happen during maintenance / restarts.
                         # Retrying the entire top-level function is the safest option.
                         msg = str(e).strip().lower()
-                        if not self._is_transient_connection_error_message(msg):
+                        if not getattr(
+                            self.sql, "is_transient_connection_error_message", lambda _m: False
+                        )(msg):
                             raise
 
                         connection_retry_count += 1
@@ -464,36 +466,8 @@ class Engine:
             raise exceptions.DbDatabaseMissingError(enhanced_message) from exception
         if re.findall(r"already exists", msg, re.M):
             raise exceptions.DbObjectExistsError(enhanced_message) from exception
-        if re.findall(r"server closed the connection unexpectedly", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"ssl syscall error: eof detected", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"ssl syscall error: connection reset by peer", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"eof detected", msg, re.M):
-            # Be conservative: EOFs typically indicate a transient disconnect.
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"no connection to the server", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"connection timed out", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"could not connect to server", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"cannot connect to server", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"terminating connection due to administrator command", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"connection reset by peer", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"broken pipe", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"could not receive data from server", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"could not send data to server", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"connection already closed", msg, re.M):
-            raise exceptions.DbConnectionError(enhanced_message) from exception
-        if re.findall(r"cursor already closed", msg, re.M):
+        # Dialect-specific connection error message classification (fallback when no/unknown code).
+        if getattr(self.sql, "is_connection_error_message", lambda _m: False)(msg):
             raise exceptions.DbConnectionError(enhanced_message) from exception
         if "no such table:" in msg:
             raise exceptions.DbTableMissingError(enhanced_message) from exception
@@ -512,26 +486,6 @@ class Engine:
         # If we can't classify it, re-raise with enhanced message
         raise type(exception)(enhanced_message) from exception
 
-    def _is_transient_connection_error_message(self, msg: str) -> bool:
-        """Return True if this looks like a retryable, transient connection drop.
-
-        Keep this intentionally conservative: authentication/config issues should not be retried.
-        """
-        needles = (
-            "ssl syscall error",
-            "eof detected",
-            "server closed the connection unexpectedly",
-            "connection reset by peer",
-            "broken pipe",
-            "could not receive data from server",
-            "could not send data to server",
-            "terminating connection due to administrator command",
-            "connection already closed",
-            "cursor already closed",
-            "no connection to the server",
-        )
-        return any(n in msg for n in needles)
-
     def _format_sql_with_params(self, sql, parameters):
         """
         Format SQL query with parameters merged for easy copy-paste debugging.

{velocity_python-0.0.204 → velocity_python-0.0.205}/src/velocity/db/servers/base/sql.py

@@ -52,6 +52,43 @@ class BaseSQLDialect(ABC):
         """
         pass
 
+    @classmethod
+    def is_connection_error_message(cls, msg: str) -> bool:
+        """Return True if an error message indicates a connection-level failure.
+
+        Dialects should override/extend this if they can do better than message matching.
+        Engine uses this only as a fallback when no/unknown error code is available.
+        """
+        if not msg:
+            return False
+        m = str(msg).strip().lower()
+        needles = (
+            "server closed the connection unexpectedly",
+            "no connection to the server",
+            "connection timed out",
+            "could not connect to server",
+            "cannot connect to server",
+            "connection already closed",
+            "cursor already closed",
+            "ssl syscall error",
+            "eof detected",
+            "connection reset by peer",
+            "broken pipe",
+            "terminating connection due to administrator command",
+            "could not receive data from server",
+            "could not send data to server",
+        )
+        return any(n in m for n in needles)
+
+    @classmethod
+    def is_transient_connection_error_message(cls, msg: str) -> bool:
+        """Return True if a connection error looks transient/retryable.
+
+        Default implementation treats most low-level disconnects as transient.
+        Dialects may override to be stricter.
+        """
+        return cls.is_connection_error_message(msg)
+
     # Core CRUD Operations
     @classmethod
     @abstractmethod

{velocity_python-0.0.204 → velocity_python-0.0.205}/src/velocity_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: velocity-python
-Version: 0.0.204
+Version: 0.0.205
 Summary: A rapid application development library for interfacing with data storage
 Author-email: Velocity Team <info@codeclubs.org>
 License-Expression: MIT