Search code examples
pythondjangomany-to-many

How do I access the child classes of an object in django without knowing the name of the child class?


In Django, when you have a parent class and multiple child classes that inherit from it you would normally access a child through parentclass.childclass1_set or parentclass.childclass2_set, but what if I don't know the name of the specific child class I want?

Is there a way to get the related objects in the parent->child direction without knowing the child class name?


Solution

  • (Update: For Django 1.2 and newer, which can follow select_related queries across reverse OneToOneField relations (and thus down inheritance hierarchies), there's a better technique available which doesn't require the added real_type field on the parent model. It's available as InheritanceManager in the django-model-utils project.)

    The usual way to do this is to add a ForeignKey to ContentType on the Parent model which stores the content type of the proper "leaf" class. Without this, you may have to do quite a number of queries on child tables to find the instance, depending how large your inheritance tree is. Here's how I did it in one project:

    from django.contrib.contenttypes.models import ContentType
    from django.db import models
    
    class InheritanceCastModel(models.Model):
        """
        An abstract base class that provides a ``real_type`` FK to ContentType.
    
        For use in trees of inherited models, to be able to downcast
        parent instances to their child types.
    
        """
        real_type = models.ForeignKey(ContentType, editable=False)
    
        def save(self, *args, **kwargs):
            if self._state.adding:
                self.real_type = self._get_real_type()
            super(InheritanceCastModel, self).save(*args, **kwargs)
        
        def _get_real_type(self):
            return ContentType.objects.get_for_model(type(self))
                
        def cast(self):
            return self.real_type.get_object_for_this_type(pk=self.pk)
        
        class Meta:
            abstract = True
    

    This is implemented as an abstract base class to make it reusable; you could also put these methods and the FK directly onto the parent class in your particular inheritance hierarchy.

    This solution won't work if you aren't able to modify the parent model. In that case you're pretty much stuck checking all the subclasses manually.