Add process_response handler to unset the middleware request object.

jazzband · Mar 17, 2016 · 4ecf837 · 4ecf837
1 parent 5806e9b
commit 4ecf837
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/simple_history/middleware.py b/simple_history/middleware.py
@@ -11,3 +11,8 @@ class HistoryRequestMiddleware(object):
 
     def process_request(self, request):
         HistoricalRecords.thread.request = request
+
+    def process_response(self, request, response):
+        if hasattr(HistoricalRecords.thread, 'request'):
+            del HistoricalRecords.thread.request
+        return response