5

我正在使用 ibm_db2 驱动程序和 sqlalchemy 处理 IBM DB2 数据库。我的模型是：

class User(Model):
    id          = Column('UID', Integer, primary_key=True)
    user        = Column('USER', String(20))
    password    = Column('PASSWORD', String(10))
    name        = Column('NAME', String(30))

数据库中的字符串字段（例如name）采用以下形式：

>>> "John                                "

，其中值由模式填充到字段的全长。

我需要将此行为更改为在 query.all()输出结果之前生成的 sqlalchemy类型 String （或其衍生物）（例如value.strip() ）：

>>> "John"

我怎样才能做到这一点？

@property装饰器不适用。我需要更改标准 sqlalchemy String 类的行为。

4

1 回答 1

7

我不想改变标准 String 的行为，而是创建一个新类型（然后你可以将它重命名为 String 每个模块或其他），但这样最干净：

from sqlalchemy import types

class StrippedString(types.TypeDecorator):
    """
    Returns CHAR values with spaces stripped
    """

    impl = types.String

    def process_bind_param(self, value, dialect):
        "No-op"
        return value

    def process_result_value(self, value, dialect):
        """
        Strip the trailing spaces on resulting values.
        If value is false, we return it as-is; it might be none
        for nullable columns
        """
        return value.rstrip() if value else value

    def copy(self):
        "Make a copy of this type"
        return StrippedString(self.impl.length)

现在您可以使用StrippedString而不是String

于 2013-08-24T01:31:32.893 回答