diff --git a/src/auditlog/middleware.py b/src/auditlog/middleware.py index b8bd34b..d6c7d68 100644 --- a/src/auditlog/middleware.py +++ b/src/auditlog/middleware.py @@ -6,7 +6,7 @@ import time from django.conf import settings from django.db.models.signals import pre_save from django.utils.functional import curry -from django.db.models.loading import get_model +from django.apps import apps from auditlog.models import LogEntry @@ -65,9 +65,9 @@ class AuditlogMiddleware(object): """ try: app_label, model_name = settings.AUTH_USER_MODEL.split('.') - auth_user_model = get_model(app_label, model_name) + auth_user_model = apps.get_model(app_label, model_name) except ValueError: - auth_user_model = get_model('auth', 'user') + auth_user_model = apps.get_model('auth', 'user') if sender == LogEntry and isinstance(user, auth_user_model) and instance.actor is None: instance.actor = user if hasattr(threadlocal, 'auditlog'):