Django 模型 - 外国人之一设置为主要



我在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 中使用信号是好方法吗?还是应该扩展模型的默认保存和编辑功能?

最新更新