Как проверить типы данных столбцов в ORM SQLAlchemy? - программирование
Подтвердить что ты не робот

Как проверить типы данных столбцов в ORM SQLAlchemy?

Используя ORM SQLAlchemy, я хочу, чтобы значения были правильного типа для их столбцов.

Например, скажем, у меня есть столбец Integer. Я пытаюсь вставить значение "hello", которое не является допустимым целым числом. SQLAlchemy позволит мне это сделать. Только позже, когда я выполняю session.commit(), возникает ли это исключение: sqlalchemy.exc.DataError: (DataError) invalid input syntax integer: "hello"….

Я добавляю партии записей, и я не хочу фиксировать их после каждого add(…) по соображениям производительности.

Итак, как я могу:

  • Поднимите исключение, как только я сделаю session.add(…)
  • Или, убедитесь, что значение, которое я вставляю, можно преобразовать в целевой тип столбца столбца, прежде чем добавлять его в пакет?
  • Или любой другой способ предотвратить повреждение одной плохой записи всего commit().
4b9b3361

ОТВЕТЫ

Ответ 1

SQLAlchemy не создает это в том случае, если он отказывается от DBAPI/базы данных как лучший и наиболее эффективный источник проверки и принуждения значений.

Чтобы создать свою собственную проверку, обычно используется проверка типов TypeDecorator или ORM. TypeDecorator имеет то преимущество, что он работает в ядре и может быть довольно прозрачным, хотя это происходит только тогда, когда SQL действительно испускается.

Чтобы выполнить проверку и принуждение раньше, это на уровне ORM.

Валидация может быть ad-hoc на уровне ORM через @validates:

http://docs.sqlalchemy.org/en/latest/orm/mapped_attributes.html#simple-validators

Система событий, использующая @validates, также доступна напрямую. Вы можете написать обобщенное решение, которое связывает валидаторы по вашему выбору с отображаемыми типами:

from sqlalchemy import Column, Integer, String, DateTime
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import event
import datetime

Base= declarative_base()

def validate_int(value):
    if isinstance(value, basestring):
        value = int(value)
    else:
        assert isinstance(value, int)
    return value

def validate_string(value):
    assert isinstance(value, basestring)
    return value

def validate_datetime(value):
    assert isinstance(value, datetime.datetime)
    return value

validators = {
    Integer:validate_int,
    String:validate_string,
    DateTime:validate_datetime,
}

# this event is called whenever an attribute
# on a class is instrumented
@event.listens_for(Base, 'attribute_instrument')
def configure_listener(class_, key, inst):
    if not hasattr(inst.property, 'columns'):
        return
    # this event is called whenever a "set" 
    # occurs on that instrumented attribute
    @event.listens_for(inst, "set", retval=True)
    def set_(instance, value, oldvalue, initiator):
        validator = validators.get(inst.property.columns[0].type.__class__)
        if validator:
            return validator(value)
        else:
            return value


class MyObject(Base):
    __tablename__ = 'mytable'

    id = Column(Integer, primary_key=True)
    svalue = Column(String)
    ivalue = Column(Integer)
    dvalue = Column(DateTime)


m = MyObject()
m.svalue = "ASdf"

m.ivalue = "45"

m.dvalue = "not a date"

Проверка и принуждение также могут быть построены на уровне типа с использованием TypeDecorator, хотя это происходит только тогда, когда SQL испускается, например, этот пример, который принуждает строки utf-8 к unicode:

http://docs.sqlalchemy.org/en/latest/core/custom_types.html#coercing-encoded-strings-to-unicode