SQLAlchemy - максимальная длина столбца

Question

возможно ли в SQLAlchemy обеспечить максимальную длину строки значения, назначенного сопоставляемому столбцу? Все, что я хочу, это вызвать исключение, если назначенное строковое значение длиннее, чем длина соответствующего столбца таблицы типа STRING.

Спасибо

Answer 1

Проще всего переименовать сопоставленный столбец и передать его через свойство:

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 в любом классе, который вы хотите проверить. Вы также можете просто установить его для базового класса, если хотите, чтобы он применялся ко всем классам, производным от него.

Answer 2

Вот обновленная версия, которая соответствует системе событий более новых версий 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