python - 如何在 SQLAlchemy 中定义对继承列的约束？

Question

我有一个在 http://docs.sqlalchemy.org/en/latest/orm/inheritance.html#joined-table-inheritance 中列出的类继承方案我想定义一个使用父类和子类列的约束。

from sqlalchemy import (
    create_engine, Column, Integer, String, ForeignKey, CheckConstraint
)
from sqlalchemy.ext.declarative import declarative_base


Base = declarative_base()


class Parent(Base):
    __tablename__ = 'parent'

    id = Column(Integer, primary_key=True)
    type = Column(String)
    name = Column(String)

    __mapper_args__ = {'polymorphic_on': type}


class Child(Parent):
    __tablename__ = 'child'

    id = Column(Integer, ForeignKey('parent.id'), primary_key=True)
    child_name = Column(String)

    __mapper_args__ = {'polymorphic_identity': 'child'}
    __table_args__ = (CheckConstraint('name != child_name'),)


engine = create_engine(...)
Base.metadata.create_all(engine)

这不起作用，因为 name 不是 child 中的列；我得到错误

sqlalchemy.exc.ProgrammingError: (psycopg2.ProgrammingError) column "name" does not exist
 [SQL: '\nCREATE TABLE child (\n\tid INTEGER NOT NULL, \n\tPRIMARY KEY (id), \n\tCHECK (name="something"), \n\tFOREIGN KEY(id) REFERENCES parent (id)\n)\n\n']

那么我该如何定义这样的约束呢？

Answer 1

简单的回答:你不能使用 CHECK 约束来做到这一点。

你不能在普通的 RDBMS 中做到这一点，因此你不能在使用 SQLAlchemy 中做到这一点。
但是，如果所有数据修改都通过您的应用程序(而不是直接访问数据库)，您可以将验证例程添加到您的类中:

class Child(Parent):
    # ...

    @validates('child_name')
    def validate_child_name(self, key, child_name):
        assert child_name != name
        return child_name

阅读更多 Simple Validators .