Fix #76 -- Ensure SealableModel(seal) respects MRO.

charettes · Sep 21, 2023 · 037a025 · 037a025
1 parent cda624a
commit 037a025
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/seal/models.py b/seal/models.py
@@ -61,6 +61,8 @@ class SealableModel(models.Model):
     """
 
     def __init_subclass__(cls, seal=None, **kwargs):
+        if seal is None:
+            seal = getattr(cls, "_seal_managers", seal)
         cls._seal_managers = seal
         return super().__init_subclass__(**kwargs)
 

diff --git a/tests/test_models.py b/tests/test_models.py
@@ -196,10 +196,17 @@ class SealedManagers(SealableModel):
         )
 
         class MixinInitSubclass:
-            def __init_subclass__(cls, foo, **kwargs):
+            def __init_subclass__(cls, foo=None, **kwargs):
+                cls.foo = foo
                 super().__init_subclass__(**kwargs)
 
-        class SealedModel(SealableModel, MixinInitSubclass, foo="bar", seal=True):
+        class SealedBaseModel(MixinInitSubclass, SealableModel, foo="bar", seal=True):
+            class Meta:
+                abstract = True
+
+        self.assertEqual(SealedBaseModel.foo, "bar")
+
+        class SealedModel(SealedBaseModel):
             manager = SealableManager(seal=False)
             as_manager = SealableQuerySet.as_manager(seal=False)