Commits

pfw committed 7cd52e2

flaskext/durus.py - only abort if durus_db exists on ctx

Comments (0)

Files changed (1)

flaskext/durus.py

     def teardown_request(self, exception):
         #print "teardown request, put connection back in the pool"
         ctx = _request_ctx_stack.top
-        ctx.durus_db.abort()
-        self.connections.append(ctx.durus_db)
-        #print "back in pool, size [%d]" % len(self.connections)
-        del(ctx.durus_db)
+        if hasattr(ctx, 'durus_db'):
+            ctx.durus_db.abort()
+            self.connections.append(ctx.durus_db)
+            #print "back in pool, size [%d]" % len(self.connections)
+            del(ctx.durus_db)
 
     def abort(self):
         ctx = _request_ctx_stack.top