Mixin to use django-auditlog with DRF

drf_django_auditlog_mixin.py

from django.db.models.signals import pre_save
from django.utils.functional import curry
from auditlog.middleware import threadlocal, AuditlogMiddleware
from auditlog.models import LogEntry


class SetUserForAuditMixin:
    def should_active_signal(self, request):
        return request.user.is_authenticated
    
    def initial(self, request, *args, **kwargs):
        """
        Overwritten method to set the pre_save signal of django-auditlog
        if its activated
        """
        super().initial(request, *args, **kwargs)

        if hasattr(threadlocal, 'auditlog'):  # django-auditlog middleware is enabled
            signal_duid = threadlocal.auditlog['signal_duid']
            if self.should_active_signal(request):
                set_actor = curry(AuditlogMiddleware.set_actor, user=request.user,
                                  signal_duid=signal_duid)
                pre_save.connect(set_actor, sender=LogEntry,
                                 dispatch_uid=signal_duid, weak=False)

    def finalize_response(self, request, response, *args, **kwargs):
        """
        Overwritten method to disconnect pre_save signal of django-auditlog
        if it was connected before.
        """
        response = super().finalize_response(request, response, *args, **kwargs)

        if hasattr(threadlocal, 'auditlog'):
            pre_save.disconnect(sender=LogEntry, dispatch_uid=threadlocal.auditlog['signal_duid'])
        return response