Add ReplicationMiddleware.process_exception

django_replicated/middleware.py

@@ -96,6 +96,10 @@ def process_response(self, request, response):
         routers.reset()
         return response
 
+    def process_exception(self, request, exception):
+        if settings.REPLICATED_RESET_ON_EXCEPTION:
+            routers.reset()
+
     def check_state_override(self, request, state):
         '''
         Used to check if a web request should use a master or slave

django_replicated/settings.py

@@ -31,5 +31,8 @@
 # Status codes on which set cookie for read-after-write workaround
 REPLICATED_FORCE_MASTER_COOKIE_STATUS_CODES = (302, 303)
 
+# Reset state on exception in request
+REPLICATED_RESET_ON_EXCEPTION = False
+
 
 REPLICATED_MANAGE_ATOMIC_REQUESTS = False