django_polymorphic_model_hierarchy_with_automatic_subclass_queries.py

1from django.db import models
2from polymorphic.models import PolymorphicModel
3
4class Project(PolymorphicModel):
5    topic = models.CharField(max_length=30)
6
7class ArtProject(Project):
8    artist = models.CharField(max_length=30)
9
10class ResearchProject(Project):
11    supervisor = models.CharField(max_length=30)
12
13# Create objects
14Project.objects.create(topic="Agriculture")
15ArtProject.objects.create(topic="Painting", artist="Mona Lisa")
16ResearchProject.objects.create(topic="Physics", supervisor="Einstein")
17
18# Polymorphic queries
19# This returns a queryset containing Project, ArtProject and ResearchProject objects
20# with their specific fields automatically populated.
21all_projects = Project.objects.all()
22# [<Project: Project object>, <ArtProject: ArtProject object>, <ResearchProject: ResearchProject object>]