- [bug] Added __reduce__ to StatementError,

DBAPIError so that exceptions are pickleable,
as when using multiprocessing.  However, not
all DBAPIs support this yet, such as
psycopg2. [ticket:2371]

1 files changed, 8 insertions, 0 deletions

diff --git a/lib/sqlalchemy/exc.py b/lib/sqlalchemy/exc.py
index 59a15f079..55205b274 100644
--- a/lib/sqlalchemy/exc.py
+++ b/lib/sqlalchemy/exc.py
@@ -164,6 +164,10 @@ class StatementError(SQLAlchemyError):
         self.params = params
         self.orig = orig
 
+    def __reduce__(self):
+        return self.__class__, (self.message, self.statement, 
+                                self.params, self.orig)
+
     def __str__(self):
         from sqlalchemy.sql import util
         params_repr = util._repr_params(self.params, 10)
@@ -219,6 +223,10 @@ class DBAPIError(StatementError):
 
         return cls(statement, params, orig, connection_invalidated)
 
+    def __reduce__(self):
+        return self.__class__, (self.statement, self.params, 
+                    self.orig, self.connection_invalidated)
+
     def __init__(self, statement, params, orig, connection_invalidated=False):
         try:
             text = str(orig)