python - 使用 SQLAlchemy 定义 PostgreSQL 字符串列的最小长度

Question

我正在使用 SQLAlchemy 声明性基础定义一个 PostgreSQL 表，如下所示:

from sqlalchemy import Column, String, BigInteger
from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()

class MyTable(Base):
    __tablename__ = 'my_table'
    id = Column('id', BigInteger, primary_key=True)
    some_string = Column('some_string', String(256), nullable=False)

nullable 约束保证some_string 不能为null。但是，我还想给 some_string 一个最小长度，或者只是禁止它成为空字符串。我该怎么做？

Answer 1

理想情况下，我们希望通过约束在数据库层应用验证，并在我们的模型中也拒绝尝试将属性设置为太短的字符串，甚至在尝试保存它之前到数据库。我们可以用 CheckConstraint 做前者使用 char_length功能，后者通过添加 validator .下面是一个在 some_string 上强制执行最小长度为 3 个字符的示例:

from sqlalchemy import Column, String, BigInteger
from sqlalchemy.schema import CheckConstraint
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import validates
Base = declarative_base()

class MyTable(Base):
    __tablename__ = 'my_table'
    id = Column('id', BigInteger, primary_key=True)
    some_string = Column('some_string', String(256), nullable=False)

    __table_args__ = (
        CheckConstraint('char_length(some_string) > 2',
                        name='some_string_min_length'),
    )

    @validates('some_string')
    def validate_some_string(self, key, some_string) -> str:
        if len(some_string) <= 2:
            raise ValueError('some_string too short')
        return some_string