Trying to reconnect to db if cookie auth fails

piotras · Aug 22, 2010 · d22ac9c · d22ac9c
1 parent e2b730f
commit d22ac9c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/midgardmvc/lib/midgard/auth/cookie.py b/midgardmvc/lib/midgard/auth/cookie.py
@@ -167,6 +167,12 @@ def authenticate(self, environ, identity):
 
         user = h.midgard.db.user.get({"login": identity.get("login"), "authtype": self.authtype})
 
+        if not user:
+            # Try to reopen db connection and then try again
+            from midgardmvc.lib.midgard.connection import instance as connection_instance
+            if connection_instance.reconnect():
+                user = h.midgard.db.user.get({"login": identity.get("login"), "authtype": self.authtype})
+
         log.debug("user: ")
         log.debug(user)
 

diff --git a/midgardmvc/lib/midgard/connection.py b/midgardmvc/lib/midgard/connection.py
@@ -55,6 +55,16 @@ def connect(self):
             raise Exception('Could not open database connection, reason: %s' % midgard._connection.get_error_string())
 
         return True
+
+    def reconnect(self):
+        try:
+            self._connected = self._connection.reopen()
+        except Exception, e:
+            print "Exception occurred while trying to reconnect: "
+            print e
+            self._connected = False
+
+        return self._connected
 
 instance = ConnectionWrapper()