SQLAlchemy - установить значение в модели, используя имя столбца базы данных, а не имя свойства

Question

У меня есть таблица с колонкой TS_TEST_ID. У меня есть модель SQLAlchemy со следующим свойством:

id = Column(u'TS_TEST_ID', INTEGER(), primary_key=True, nullable=False)

Есть ли способ установить id для экземпляра моего модельного класса, когда известен только TS_TEST_ID? То есть я знаю только имя столбца в базе данных, и я хочу каким-то образом сопоставить его с id и установить свойство id в моей модели. Я надеялся, что просто MyModel(**{'TS_TEST_ID':1}) сработает, но я получаю следующую ошибку:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<string>", line 4, in __init__
  File "C:\Python26\lib\site-packages\sqlalchemy\orm\state.py", line 111,
   in initialize_instance
    return manager.events.original_init(*mixed[1:], **kwargs)
  File "C:\Python26\lib\site-packages\sqlalchemy\ext\declarative.py", line 1377,
   in _declarative_constructor
    (k, cls_.__name__))
TypeError: 'TS_TEST_ID' is an invalid keyword argument for MyModel

Я бы предпочел не определять TS_TEST_ID = Column(u'TS_TEST_ID', INTEGER(), primary_key=True, nullable=False) для каждого столбца в моей модели, или даже метод с именем TS_TEST_ID, который возвращает свойство id.

Answer 1

Это кажется сложнее, чем нужно, но работает.Я надеюсь, что есть более прямой способ сделать это ...

@staticmethod
def fromFieldHash(h):
    row = MyModel()
    cols = list(row.__table__._columns)
    for k, v in h.iteritems():
        col = find(lambda c: c.key == k, cols)
        # See http://www.sqlalchemy.org/trac/wiki/06Migration#AnImportantExpressionLanguageGotcha
        if col is not None:
            prop = row.__mapper__._columntoproperty[col].key
            setattr(row, prop, v)
    return row

Я использую метод find из Чистейшая функция поиска в списке Python .

Answer 2

Если вы хотите, чтобы __init__ обрабатывал как атрибуты, так и столбцы, хотя выделенная фабрика classmethod кажется более чистым подходом, в качестве абстрактной базы можно использовать следующую реализацию:

from sqlalchemy import inspect
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()


def _create_col_lookup(mapper):
    return {c.name: k for k, c in mapper.columns.items()}


class ColBase(Base):

    __abstract__ = True

    def __init__(self, *args, **kwargs):
        # Create a lookup of column name -> attr key
        ctoa = _create_col_lookup(inspect(self).mapper)

        # Process intersecting keys
        for k in ctoa.keys() & kwargs.keys():
            setattr(self, ctoa[k], kwargs.pop(k))

        super().__init__(*args, **kwargs)

Используя вышеприведенный пропуск следующих утверждений:

class Foo(ColBase):
    __tablename__ = 'foo'
    id = Column(Integer, primary_key=True)
    bar = Column('Baz 1')

f = Foo(**{'Baz 1': 1})
assert f.bar == 1

f = Foo(**{'bar': 1})
assert f.bar == 1