Tough method should get proper error message after connection was lost.

Author: Cito

DBUtils/SteadyDB.py

@@ -436,7 +436,7 @@ def close(self):
             self._closed = True
 
     def _get_tough_method(self, name):
-        """Return a "tough" version of the method."""
+        """Return a "tough" version of the given cursor method."""
         def tough_method(*args, **kwargs):
             execute = name.startswith('execute')
             try:
@@ -493,14 +493,21 @@ def tough_method(*args, **kwargs):
                             result = method2(*args, **kwargs)
                             if execute:
                                 self._clearsizes()
-                        except Exception:
-                            pass
+                        except error.__class__: # same execution error
+                            use2 = False
+                        except Exception, error: # other execution errors
+                            use2 = True
                         else:
+                            use2 = True
+                            error = None
+                        if use2:
                             self.close()
                             self._con._close()
                             self._con._store(con2)
                             self._cursor = cursor2
                             self._con._usage += 1
+                            if error:
+                                raise error # raise the other error
                             return result
                         try:
                             cursor2.close()

DBUtils/Tests/TestSteadyDB.py

@@ -470,6 +470,15 @@ def test8_SteadyDBConnectionFailures(self):
         db.close()
         db.cursor()
 
+    def test9_SteadyDBConnectionFailureError(self):
+        db = SteadyDBconnect(dbapi)
+        cursor = db.cursor()
+        db.close()
+        cursor.execute('select test')
+        cursor = db.cursor()
+        db.close()
+        self.assertRaises(ProgrammingError, cursor.execute, 'error')
+
 
 if __name__ == '__main__':
     unittest.main()