可以通过 Django ORM 中的信号机制来在保存 m2m 关系模型前进行检查。具体步骤如下:
- 在 models.py 中定义 m2m 关系模型,并添加附加字段(additional_field):
from django.db import models
class M2MModel(models.Model):
name = models.CharField(max_length=50)
additional_field = models.IntegerField(default=0)
def __str__(self):
return self.name
class MainModel(models.Model):
name = models.CharField(max_length=50)
m2m_relation = models.ManyToManyField(M2MModel, related_name='main_models')
def __str__(self):
return self.name
- 在 apps.py 中定义检查函数(check_sum_equals_100),并将其绑定到 m2m_relation 的 m2m_changed 信号上:
from django.apps import AppConfig
from django.db.models.signals import m2m_changed
from django.dispatch import receiver
class MyAppConfig(AppConfig):
name = 'my_app'
def ready(self):
from my_app.models import MainModel
@receiver(m2m_changed, sender=MainModel.m2m_relation.through)
def check_sum_equals_100(sender, instance, **kwargs):
if kwargs['action'] == 'pre_add':
m2m_ids = kwargs['pk_set']
additional_field_sum = M2MModel.objects.filter(id__in=m2m_ids).aggregate(models.Sum('additional_field'))['additional_field__sum']
if additional_field_sum is not None and additional_field_sum != 100:
raise ValueError('The sum of additional_field must be 100.')
- 在 settings.py 中将应用程序配置类(MyAppConfig)添加到 INSTALLED_APPS 中:
INSTALLED_APPS = [
'my_app.apps.MyAppConfig',
...
]
现在,当通过 m2m_relation 添加 M2MModel 对象时,系统会自动检查 additional_field 字段的总和是否等于 100,如果不等于,则会抛出值错误(ValueError)。