通常为所有String字段剥离空格 - SQLAlchemy

时间:2018-06-14 09:56:46

标签: python orm sqlalchemy flask-sqlalchemy string-parsing

我通过Flask-SQLAlchemy使用SQLAlchemy作为Web应用程序的ORM。

我想在分配给任何字符串字段时自动引导和跟踪条带空格(例如str.strip)。

执行此操作的一种方法如下,但需要为每个字符串字段指定:

class User(db.Model):
    _email = db.Column('email', db.String(100), primary_key=True)
    @hybrid_property
    def email(self): return self._email
    @email.setter
    def email(self, data): self._email = data.strip()

我想为每个String字段更一般地执行此操作(无需为每个字段编写上述内容)。

1 个答案:

答案 0 :(得分:2)

一种方法是创建一个处理此类处理的自定义augmented string type

from sqlalchemy.types import TypeDecorator

class StrippedString(TypeDecorator):

    impl = db.String

    def process_bind_param(self, value, dialect):
        # In case you have nullable string fields and pass None
        return value.strip() if value else value

    def copy(self, **kw):
        return StrippedString(self.impl.length)

然后,您可以在模型中使用此代替普通String

class User(db.Model):
    email = db.Column(StrippedString(100), primary_key=True)

这与您自己的实现完全不同,因为处理发生在将值作为参数绑定到查询时,或者换句话说稍后:

In [12]: u = User(email='     so.much@white.space     ')

In [13]: u.email
Out[13]: '     so.much@white.space     '

In [14]: session.add(u)

In [15]: session.commit()

In [16]: u.email
Out[16]: 'so.much@white.space'