add cuser middleware

Author: juggler31

tracking_fields/middleware/cuser.py

@@ -0,0 +1,43 @@
+from __future__ import unicode_literals
+import threading
+from django.contrib.auth import get_user_model
+
+
+class CuserMiddleware:
+    _thread_local = threading.local()
+
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request):
+        try:
+            self.__class__.set_user(request.user)
+            response = self.get_response(request)
+            return response
+        finally:
+            self.__class__.del_user()
+
+    @classmethod
+    def get_user(cls, default=None):
+        """
+        Retrieve user info
+        """
+        return getattr(cls._thread_local, 'user', default)
+
+    @classmethod
+    def set_user(cls, user):
+        """
+        Store user info
+        """
+        if isinstance(user, str):
+            user_model = get_user_model()
+            user = user_model.objects.get(username=user)
+        cls._thread_local.user = user
+
+    @classmethod
+    def del_user(cls):
+        """
+        Delete user info
+        """
+        if hasattr(cls._thread_local, 'user'):
+            del cls._thread_local.user

tracking_fields/tests/settings.py

@@ -46,6 +46,7 @@
     "django.middleware.common.CommonMiddleware",
     "django.middleware.csrf.CsrfViewMiddleware",
     "django.contrib.auth.middleware.AuthenticationMiddleware",
+    "tracking_fields.middleware.cuser.CuserMiddleware",
     "django.contrib.messages.middleware.MessageMiddleware",
     "django.middleware.clickjacking.XFrameOptionsMiddleware",
 )

tracking_fields/tests/tests.py

@@ -4,7 +4,7 @@
 import datetime
 import json
 
-from cuser.middleware import CuserMiddleware
+from tracking_fields.middleware.cuser import CuserMiddleware
 from django.contrib.auth.models import User
 from django.contrib.contenttypes.models import ContentType
 from django.core.files import File
@@ -55,26 +55,6 @@ def test_create(self):
         assert human_event.user == self.user
         assert human_event.user_repr == self.user_repr
 
-    def test_update_without_cuser(self):
-        """
-        Test the CREATE event without the cuser module
-        """
-        from tracking_fields import tracking
-
-        tracking.CUSER = False
-        self.human.age = 43
-        self.human.save()
-        events = TrackingEvent.objects.order_by("date").all()
-        assert events.count() == 3
-        human_event = events.last()
-        assert human_event.date is not None
-        assert human_event.action == UPDATE
-        assert human_event.object == self.human
-        assert human_event.object_repr == self.human_repr
-        assert human_event.user is None
-        assert human_event.user_repr == "None"
-        tracking.CUSER = True
-
     def test_update(self):
         """
         Test the UPDATE event

tracking_fields/tracking.py

@@ -5,6 +5,7 @@
 import logging
 import uuid
 
+from tracking_fields.middleware.cuser import CuserMiddleware
 from django.contrib.contenttypes.models import ContentType
 from django.core.exceptions import ObjectDoesNotExist
 from django.db.models import ManyToManyField, Model
@@ -16,13 +17,6 @@
 except ImportError:
     StateWrapper = type("StateWrapper", (object,), dict())
 
-try:
-    from cuser.middleware import CuserMiddleware
-
-    CUSER = True
-except ImportError:
-    CUSER = False
-
 from tracking_fields.models import (
     CREATE,
     DELETE,
@@ -119,11 +113,10 @@ def _create_event(instance, action):
     """
     user = None
     user_repr = repr(user)
-    if CUSER:
-        user = CuserMiddleware.get_user()
-        user_repr = repr(user)
-        if user is not None and user.is_anonymous:
-            user = None
+    user = CuserMiddleware.get_user()
+    user_repr = repr(user)
+    if user is not None and user.is_anonymous:
+        user = None
     return TrackingEvent.objects.create(
         action=action,
         object_content_type=ContentType.objects.get_for_model(instance),