python - SQLAlchemy:检查两列之一的任何约束不为空？

Question

这可能是完全愚蠢的问题，但我在我的模型中有这样的要求，其中至少 category 或 parent_category 是 not null

我的模型看起来像

class BudgetCategories(db.Model):
    __tablename__ = 'budget_categories'
    uuid = Column('uuid', GUID(), default=uuid.uuid4, primary_key=True,
                  unique=True)
    budget_id = Column(GUID(), ForeignKey('budgets.uuid'), nullable=False)
    budget = relationship('Budget', backref='budgetCategories')
    category = Column('category', sa.types.String, nullable=True)
    parent_category = Column('parent_category', sa.types.String, nullable=True)
    amount = Column('amount', Numeric(10, 2), nullable=False)
    recurring = Column('recurring', sa.types.Boolean,
                       nullable=False)
    created_on = Column('created_on', sa.types.DateTime(timezone=True),
                        nullable=False)

我该如何指定。我什至不知道该尝试什么

感谢任何指点

我正在使用 PostgreSQL 作为后端数据库

Answer 1

我不是 100% 确定 PostgreSQL 语法，但是在添加到您的 BudgetCategories 模型之后应该使用 CheckConstraint 来解决问题:

class BudgetCategories(Base):
    __tablename__ = 'budget_categories'
    # ...

    # @note: new
    __table_args__ = (
            CheckConstraint('NOT(category IS NULL AND parent_category IS NULL)'),
            )