wip

auditlog/conf.py

@@ -1,6 +1,5 @@
 from django.conf import settings
 
-
 # Register all models when set to True
 settings.AUDITLOG_INCLUDE_ALL_MODELS = getattr(
     settings, "AUDITLOG_INCLUDE_ALL_MODELS", False

auditlog/models.py

@@ -445,7 +445,7 @@ def changes_display_dict(self):
         auditlog = AuditlogRegistry.get_auditlog(model=model)
 
         model_fields = None
-        if auditlog.contains(model._meta.model):
+        if auditlog:
             model_fields = auditlog.get_model_fields(model._meta.model)
 
         changes_display_dict = {}

auditlog/registry.py

@@ -401,18 +401,20 @@ def get_auditlog(cls, model: ModelBase):
         Retrieve the auditlog object associated with a given model.
         """
         auditlogs = get_default_auditlogs()
-        return next(
-            (auditlog for auditlog in auditlogs if auditlog._registry.get(model)), None
-        )
+        for auditlog in auditlogs:
+            if auditlog.contains(model):
+                return auditlog
 
     @classmethod
     def get_model_fields(cls, model: ModelBase):
         """
         Wrapper of AuditlogModelRegistry.get_model_fields().
         """
         auditlog = cls.get_auditlog(model)
-        model_fields = auditlog.get_model_fields(model)
-        return model_fields
+        if auditlog:
+            model_fields = auditlog.get_model_fields(model)
+            return model_fields
+        return {}
 
 
 auditlog_registry = AuditlogRegistry()

auditlog_tests/models.py

@@ -330,22 +330,6 @@ class AutoManyRelatedModel(models.Model):
     related = models.ManyToManyField(SimpleModel)
 
 
-class CreateUpdateOnlyModel(models.Model):
-    """
-    A simple model to test create_only_auditlog and update_only_auditlog.
-    """
-
-    text = models.TextField(blank=True)
-    boolean = models.BooleanField(default=False)
-    integer = models.IntegerField(blank=True, null=True)
-    datetime = models.DateTimeField(auto_now=True)
-
-    history = AuditlogHistoryField(delete_related=True)
-
-    def __str__(self):
-        return self.text
-
-
 auditlog.register(AltPrimaryKeyModel)
 auditlog.register(UUIDPrimaryKeyModel)
 auditlog.register(ProxyModel)

auditlog_tests/tests.py

@@ -39,7 +39,6 @@
     AutoManyRelatedModel,
     CharfieldTextfieldModel,
     ChoicesFieldModel,
-    CreateUpdateOnlyModel,
     DateTimeFieldModel,
     JSONModel,
     ManyRelatedModel,
@@ -2650,15 +2649,18 @@ def test_get_changes_for_missing_model(self):
 class CustomAuditlogTest(TestCase):
     def setUp(self):
         super().setUp()
+        if auditlog.contains(SimpleModel):
+            auditlog.unregister(SimpleModel)
 
     def tearDown(self):
         for model in create_only_auditlog.get_models():
             create_only_auditlog.unregister(model)
         for model in update_only_auditlog.get_models():
             update_only_auditlog.unregister(model)
+        auditlog.register(SimpleModel)
 
     def make_object(self):
-        return CreateUpdateOnlyModel.objects.create(text="I am not difficult.")
+        return SimpleModel.objects.create(text="I am not difficult.")
 
     def update_object(self, obj):
         obj.text = "Changed"
@@ -2671,9 +2673,7 @@ def test_create_only_auditlog(self):
                 "auditlog_tests.test_registry.create_only_auditlog",
             ]
         ):
-            create_only_auditlog.register(
-                CreateUpdateOnlyModel, include_fields=["text"]
-            )
+            create_only_auditlog.register(SimpleModel, include_fields=["text"])
 
             # Get the object to work with
             obj = self.make_object()
@@ -2693,9 +2693,7 @@ def test_update_only_auditlog(self):
                 "auditlog_tests.test_registry.update_only_auditlog",
             ]
         ):
-            update_only_auditlog.register(
-                CreateUpdateOnlyModel, include_fields=["text"]
-            )
+            update_only_auditlog.register(SimpleModel, include_fields=["text"])
 
             # Get the object to work with
             obj = self.make_object()