我通过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字段更一般地执行此操作(无需为每个字段编写上述内容)。
答案 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'