Merge pull request #114 from rsenkbeil/inheritance-manager-self-refer…

…ence Fix InheritanceManager when the model has a self reference.
jazzband · Feb 19, 2014 · f26b33a · f26b33a
2 parents fe0ecea + 617ec2a
commit f26b33a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/model_utils/managers.py b/model_utils/managers.py
@@ -106,6 +106,7 @@ def _get_subclasses_recurse(self, model, levels=None):
             rel for rel in model._meta.get_all_related_objects()
             if isinstance(rel.field, OneToOneField)
             and issubclass(rel.field.model, model)
+            and model is not rel.field.model
             ]
         subclasses = []
         if levels:

diff --git a/model_utils/tests/models.py b/model_utils/tests/models.py
@@ -25,6 +25,7 @@ class InheritanceManagerTestParent(models.Model):
     related = models.ForeignKey(
         InheritanceManagerTestRelated, related_name="imtests", null=True)
     normal_field = models.TextField()
+    related_self = models.OneToOneField("self", related_name="imtests_self", null=True)
     objects = InheritanceManager()
 
     def __unicode__(self):