django - django 检查约束布尔表达式 [postgresql]

Question

所以我有一个这样的模型

class Role(BaseModel):

    class Meta:
        verbose_name = 'role'
        verbose_name_plural = 'roles'
        ordering = ['position', 'cluster']
        required_db_features = {
            'supports_deferrable_unique_constraints',
        }
        constraints = [
            models.UniqueConstraint(
                fields=['position', 'cluster'],
                name='deferrable_unique_role_position',
                deferrable=models.Deferrable.DEFERRED
            ),
            models.CheckConstraint(
                name='default_role_check',
                check=models.Q(
                    is_default=True,
                    position=1,
                    color='#969696',
                    name='@everyone'
                )
            )
        ]

    permission_flags = [
        'READ_DATA', 'WRITE_DATA', 'MANAGE_RECORDS', 'MANAGE_ROLES',
        'MANAGE_CLUSTER', 'MANAGE_DATASHEETS', 'MANAGE_FIELDS', 'MANAGE_CONSTRAINTS',
        'KICK_MEMBERS', 'MANAGE_MEMBERS',
    ]

    default_perm_flags = ['READ_DATA', 'WRITE_DATA', 'MANAGE_RECORDS']

    def __str__(self):
        return self.name

    objects = managers.RolesManager()
    positions = managers.PositionalManager()
    permissions = BitField(flags=permission_flags, default=default_perm_flags, db_index=True)
    position = models.PositiveSmallIntegerField(null=True, blank=True, db_index=True, editable=True)
    name = models.CharField(max_length=100, validators=[MinLengthValidator(2)], db_index=True, default='new role')
    color = ColorField(db_index=True, default='#969696')
    is_default = models.BooleanField(default=False, db_index=True)
    cluster = models.ForeignKey('api_backend.Cluster', on_delete=models.CASCADE, editable=False)

基本上我想对该模型强制执行检查约束，以便当模型将is_default字段设置为 True 时，

• 名称必须始终是@everyone
• 颜色必须始终是#969696
• 位置必须始终为1。

我尝试实现相同的功能，但是我当前的实现不起作用。在使用序列化程序时，我可以编辑默认角色的名称，并且不会引发异常。

我在后端使用 postgresql。

有人可以帮帮我吗？提前致谢！

标签： djangopostgresqldjango-models

Answer 1

添加is_default=False案例让您更接近您的条件。

我用 Django==3.1.4 进行了测试，OR 方法导致了这个约束：

"default_role_check" CHECK (color::text = '#969696'::text AND is_default AND name::text = '@everyone'::text AND "position" = 1 OR NOT is_default)

约束示例：

class Role(BaseModel):

    class Meta:
        verbose_name = 'role'
        verbose_name_plural = 'roles'
        ordering = ['position', 'cluster']
        required_db_features = {
            'supports_deferrable_unique_constraints',
        }
       constraints = [
            models.UniqueConstraint(
                fields=['position', 'cluster'],
                name='deferrable_unique_role_position',
                deferrable=models.Deferrable.DEFERRED
            ),
            models.CheckConstraint(
                name='default_role_check',
                check=(models.Q(
                    is_default=True,
                    position=1,
                    color='#969696',
                    name='@everyone'
                ) |
                    models.Q(
                    is_default=False,
                )
                )
            )
        ]
    permission_flags = [
        'READ_DATA', 'WRITE_DATA', 'MANAGE_RECORDS', 'MANAGE_ROLES',
        'MANAGE_CLUSTER', 'MANAGE_DATASHEETS', 'MANAGE_FIELDS', 'MANAGE_CONSTRAINTS',
        'KICK_MEMBERS', 'MANAGE_MEMBERS',
    ]

    default_perm_flags = ['READ_DATA', 'WRITE_DATA', 'MANAGE_RECORDS']

    def __str__(self):
        return self.name

    objects = managers.RolesManager()
    positions = managers.PositionalManager()
    permissions = BitField(flags=permission_flags, default=default_perm_flags, db_index=True)
    position = models.PositiveSmallIntegerField(null=True, blank=True, db_index=True, editable=True)
    name = models.CharField(max_length=100, validators=[MinLengthValidator(2)], db_index=True, default='new role')
    color = ColorField(db_index=True, default='#969696')
    is_default = models.BooleanField(default=False, db_index=True)
    cluster = models.ForeignKey('api_backend.Cluster', on_delete=models.CASCADE, editable=False)