我在Django中有一个简单的模型:
class ClientEmail(models.Model):
client = models.ForeignKey('Client', on_delete=models.CASCADE)
email = models.EmailField('E-mail')
default_email = models.BooleanField('Default e-mail')
如何使客户中只有一个外国人可以设置为死人?如何让其中一个外国人始终设置为死人,即使删除了这个已经设置为默认值的外国人?
我使用信号做到了,我不知道这是否是正确的方式,如果不是,请告诉我。
@receiver(
post_delete,
sender=ClientPhone,
dispatch_uid="keep_one_client_phone_as_deafult"
)
def keep_one_default_phone_after_delete(sender, instance, **kwargs):
phones = sender.objects.filter(client_id=instance.client_id)
if len(phones) == 0:
return
elif len(phones) == 1:
if not phones[0].default:
phones[0].default = True
phones[0].save()
return
else:
if instance.default:
sender.objects.get(client_id=instance.client_id).update(default=True)
@receiver(
post_save,
sender=ClientEmail,
dispatch_uid="keep_one_client_email_as_deafult"
)
def keep_one_default_email(sender, instance, **kwargs):
emails = sender.objects.filter(client_id=instance.client_id)
if len(emails) == 0:
return
elif len(emails) == 1:
if not emails[0].default:
emails[0].default = True
emails[0].save()
return
else:
if instance.default:
sender.objects.filter(
~Q(id=instance.id),
client_id=instance.client_id
).update(default=False)
else:
emails[0].default = True
emails[0].save()
像这样在 Django 中使用信号是好方法吗?还是应该扩展模型的默认保存和编辑功能?