Django创建Enrollment对象时自动递增关联模型整数字段实现

你在Django项目中定义了Teaching Semester和Enrollments两个模型，需要实现特定课程的选课人数自动统计：在TeachingSemester模型新增registered students字段，每当创建一条Enrollments对象时，关联的学期课程记录的该字段值自动递增1。
你当前定义的模型代码如下：
TeachingSemester模型：

class TeachingSemester(models.Model):
    department = models.ForeignKey(Department,on_delete=models.CASCADE)
    program    = models.ForeignKey(Program,on_delete=models.CASCADE)
    faculty = models.ForeignKey(FacultyData,on_delete=models.CASCADE)
    semester_choices = (
        ('Spring' , 'Spring'),
        ('Fall'   , 'Fall'),
        ('Summer'  , 'Summer')
    )
    sem_choice = models.CharField(max_length=30,choices=semester_choices,default='Spring')
    course_date      =   models.DateField(default=datetime.date.today())
    section          = models.CharField(max_length=5,default = 1)
    faculty_course     = models.ForeignKey(Course,on_delete=models.CASCADE)

Enrollment模型：

class Enrollment(models.Model):
    student = models.ForeignKey(StudentData,on_delete=models.CASCADE)
    faculty = models.ForeignKey(TeachingSemester,on_delete=models.CASCADE)

第一步：给TeachingSemester模型新增计数字段 #

首先在原有TeachingSemester模型里新增选课人数字段，用非负整数类型，初始值设为0：

class TeachingSemester(models.Model):
    department = models.ForeignKey(Department,on_delete=models.CASCADE)
    program    = models.ForeignKey(Program,on_delete=models.CASCADE)
    faculty = models.ForeignKey(FacultyData,on_delete=models.CASCADE)
    semester_choices = (
        ('Spring' , 'Spring'),
        ('Fall'   , 'Fall'),
        ('Summer'  , 'Summer')
    )
    sem_choice = models.CharField(max_length=30,choices=semester_choices,default='Spring')
    course_date = models.DateField(default=datetime.date.today())
    section = models.CharField(max_length=5,default = 1)
    faculty_course = models.ForeignKey(Course,on_delete=models.CASCADE)
    # 新增选课计数字段
    registered_students = models.PositiveIntegerField(default=0, verbose_name='已选课学生数')

修改完模型后执行两条命令同步数据库结构：

python manage.py makemigrations
python manage.py migrate

第二步：实现计数自动更新逻辑 #

不要直接用「读取当前值+1再保存」的写法，高并发场景下会出现计数错乱，用Django内置的F()表达式做数据库层面的原子更新，搭配信号实现逻辑解耦，新增/删除选课记录时自动更新计数。

1. 在对应app目录下新建signals.py文件，写入以下逻辑：

from django.db.models.signals import post_save, post_delete
from django.db.models import F
from django.dispatch import receiver
from .models import Enrollment

@receiver(post_save, sender=Enrollment)
def add_enrollment_count(sender, instance, created, **kwargs):
    # 仅在新建选课记录时触发计数+1，修改已有记录不触发
    if created:
        instance.faculty.registered_students = F('registered_students') + 1
        instance.faculty.save(update_fields=['registered_students'])

@receiver(post_delete, sender=Enrollment)
def reduce_enrollment_count(sender, instance, **kwargs):
    # 删除选课记录时计数-1
    instance.faculty.registered_students = F('registered_students') - 1
    instance.faculty.save(update_fields=['registered_students'])

2. 打开同目录下的apps.py，在App配置类的ready()方法中导入信号模块，确保信号能正常加载：

from django.apps import AppConfig

# 类名和name替换成你自己app对应的配置
class CourseConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'course'

    def ready(self):
        from . import signals

补充注意事项 #

• 如果项目中已经存在历史选课数据，新增字段后需要先初始化一次计数，避免初始值为0和实际数据不符。可以在Django shell中执行以下代码完成初始化：

from django.db.models import Count
from .models import TeachingSemester

# 关联统计每个学期的实际选课人数，同步到新增字段
for semester in TeachingSemester.objects.annotate(real_count=Count('enrollment')):
    semester.registered_students = semester.real_count
    semester.save()

• 如果业务中有用bulk_create批量创建选课记录、bulk_delete批量删除记录的场景，Django的信号不会被触发，需要在批量操作的逻辑里手动更新对应学期的选课计数。
• 目前Enrollment模型里关联TeachingSemester的外键字段命名为faculty，和模型里指向教师数据的faculty字段重名，容易造成后续维护混淆，建议改成teaching_semester，不调整也不影响本次功能实现。

内容的提问来源于stack exchange，提问作者Amiable.Azam