django_model_with_djchoices_enum_field_definition.py

1from django.db import models
2from djchoices import DjangoChoices, ChoiceItem
3
4class Student(models.Model):
5    class YearInSchool(DjangoChoices):
6        freshman = ChoiceItem("FR", "Freshman")
7        sophomore = ChoiceItem("SO", "Sophomore")
8        junior = ChoiceItem("JR", "Junior")
9        senior = ChoiceItem("SR", "Senior")
10
11    year_in_school = models.CharField(
12        max_length=2,
13        choices=YearInSchool.choices,
14        default=YearInSchool.freshman,
15    )
16
17    def is_upperclassman(self):
18        return self.year_in_school in (
19            self.YearInSchool.junior,
20            self.YearInSchool.senior,
21        )