0.2.7

chat v2

AuthApp/middleware.py

@@ -1,7 +1,8 @@
 import json
 import datetime
-from django.http import HttpResponse
+from django.http import HttpResponse, JsonResponse
 from django.conf import settings
+from django.utils.deprecation import MiddlewareMixin
 
 
 def set_cookie(response, key, value, days_expire=7):
@@ -25,32 +26,63 @@ def set_cookie(response, key, value, days_expire=7):
     return response
 
 class ResponseInterceptionMiddleware:
-    # def __init__(self, get_response):
-    #     self.get_response = get_response
-    #
-    # def __call__(self, request):
-    #     return self.get_response(request)
-    # def process_request(self, request):
     def __init__(self, get_response):
         self.get_response = get_response
-        # One-time configuration and initialization.
 
     def __call__(self, request):
-        # Code to be executed for each request before
-        # the view (and later middleware) are called.
-
-        if request.user.is_authenticated:
-            request = request.user.user_profile.save_user_alerts_to_session(request)
-
         response = self.get_response(request)
-        if request.user.is_authenticated:
-            for_save_to_session = request.user.user_profile.pop_node_by_name('for_save_to_session')
-            if for_save_to_session:
-                for key, val in for_save_to_session.items():
-                    set_cookie(response, key, json.dumps(val), 30)
-
-        # Code to be executed for each request/response after
-        # the view is called.
 
+        try:
+            if type(response) == JsonResponse:
+                for_save_to_session = request.user.user_profile.pop_node_by_name('for_save_to_session')
+                if for_save_to_session:
+                    data = json.loads(response.content)
+                    data.update(for_save_to_session)
+                    response.content = json.dumps(data)
+        except Exception as e:
+            msg = f'ResponseInterceptionMiddleware error = {str(e)}'
+            print(msg)
+        # if 'errors' in response.data:
+        #     response.data.update({'success': False})
+        # print(response.data)
         return response
 
+    # def process_view(self, request, view_func, *view_args, **view_kwargs):
+    #
+    #     if request.user.is_authenticated and request.body:
+    #         for_save_to_session = request.user.user_profile.pop_node_by_name('for_save_to_session')
+    #         if for_save_to_session:
+    #             data = json.loads(request.body)
+    #             data.update(for_save_to_session)
+    #
+    #     # request_data = getattr(request, '_body', request.body)
+    #     # request_data = json.loads(request_data)
+    #     # # here you can write the logic to append the payload to request data
+    #             request._body = json.dumps(data)
+    #     return None
+
+    # def __call__(self, request):
+    #     # Code to be executed for each request before
+    #     # the view (and later middleware) are called.
+    #
+    #     if request.user.is_authenticated and request.body:
+    #         for_save_to_session = request.user.user_profile.pop_node_by_name('for_save_to_session')
+    #         if for_save_to_session:
+    #             data = json.loads(request.body)
+    #             data.update(for_save_to_session)
+    #             request._body = json.dumps(data)
+    #             # return JsonResponse(data)
+    #         # request = request.user.user_profile.save_user_alerts_to_session(request)
+    #
+    #     response = self.get_response(request)
+    #     # if request.user.is_authenticated:
+    #     #     for_save_to_session = request.user.user_profile.pop_node_by_name('for_save_to_session')
+    #     #     if for_save_to_session:
+    #     #         for key, val in for_save_to_session.items():
+    #     #             set_cookie(response, key, json.dumps(val), 30)
+    #
+    #     # Code to be executed for each request/response after
+    #     # the view is called.
+    #
+    #     return response
+