在SQLAlchemy中,是否可以强制执行分配给映射列的值的最大字符串长度?我想要的只是在赋值的字符串值长于类型为STRING的相应表列的长度时引发异常。
谢谢
在SQLAlchemy中,是否可以强制执行分配给映射列的值的最大字符串长度?我想要的只是在赋值的字符串值长于类型为STRING的相应表列的长度时引发异常。
谢谢
最简单的方法就是重命名映射列并通过属性进行代理:
class Something(Base):
...
_foo = Column('foo', String(123))
@property
def foo(self):
return self._foo
@foo.setter
def foo(self, value):
if len(value) > _foo.type.length:
raise Exception("Value too long")
self._foo = value
你可以轻松将属性创建分解出来,甚至使用像formencode这样的通用验证框架。
如果你需要更针对SQLAlchemy的解决方案,并且不介意使用特定接口,那么SQLAlchemy具有捕获属性事件的扩展机制。 使用该机制的验证器应该看起来像这样:
from sqlalchemy.orm.interfaces import AttributeExtension, InstrumentationManager
from sqlalchemy.orm import ColumnProperty
class InstallValidatorListeners(InstrumentationManager):
def post_configure_attribute(self, class_, key, inst):
"""Add validators for any attributes that can be validated."""
prop = inst.prop
# Only interested in simple columns, not relations
if isinstance(prop, ColumnProperty) and len(prop.columns) == 1:
col = prop.columns[0]
# if we have string column with a length, install a length validator
if isinstance(col.type, String) and col.type.length:
inst.impl.extensions.insert(0, LengthValidator(col.type.length))
class ValidationError(Exception):
pass
class LengthValidator(AttributeExtension):
def __init__(self, max_length):
self.max_length = max_length
def set(self, state, value, oldvalue, initiator):
if len(value) > self.max_length:
raise ValidationError("Length %d exceeds allowed %d" %
(len(value), self.max_length))
return value
您可以通过在任何要验证的类上设置__sa_instrumentation_manager__ = InstallValidatorListeners
来使用此扩展。如果您希望它适用于所有派生自其的类,则还可以将其设置为基类。
这里是更新版本,适用于新版sqlalchemy的事件系统:
class InstallValidatorListeners(InstrumentationManager):
def post_configure_attribute(self, class_, key, inst):
"""Add validators for any attributes that can be validated."""
prop = inst.prop
# Only interested in simple columns, not relations
if isinstance(prop, ColumnProperty) and len(prop.columns) == 1:
col = prop.columns[0]
if isinstance(col.type, String) and col.type.length:
sqlalchemy.event.listen(
getattr(class_, key), 'set', LengthValidator(col.type.length), retval=True)
class ValidationError(Exception):
pass
class LengthValidator(AttributeExtension):
def __init__(self, max_length):
self.max_length = max_length
def __call__(self, state, value, oldvalue, initiator):
if len(value) > self.max_length:
raise ValidationError(
"Length %d exceeds allowed %d" % (len(value), self.max_length))
return value