Динамическое сопоставление sqlalchemy

У меня есть следующая проблема:

У меня есть класс:


class Word(object):

    def __init__(self):
        self.id = None
        self.columns = {}

    def __str__(self):
        return "(%s, %s)" % (str(self.id), str(self.columns))

self.columns - это dict, который будет содержать значения (columnName: columnValue). Имя столбцов известно во время выполнения, и они загружаются в список wordColumns, например


wordColumns = ['english', 'korean', 'romanian']


wordTable = Table('word', metadata,
                  Column('id', Integer, primary_key = True)
                  )
for columnName in wordColumns:
    wordTable.append_column(Column(columnName, String(255), nullable = False))

Я даже создал явные свойства mapper, чтобы "принудительно" столбцы таблицы отображались в word.columns [columnName], вместо word.columnName, я не получаю никаких ошибок при сопоставлении, но, похоже, т работы.


mapperProperties = {}
for column in wordColumns:
    mapperProperties['columns[\'%']' % column] = wordTable.columns[column]

mapper(Word, wordTable, mapperProperties)

Когда я загружаю объект word, SQLAlchemy создает объект, который имеет свойства word.columns ['english'], word.columns ['korean'] и т.д. вместо того, чтобы загружать их в word.columns dict. Поэтому для каждого столбца создается новое свойство. Кроме того, словарь word.columns даже не существует.

Точно так же, когда я пытаюсь сохранить слово, SQLAlchemy ожидает найти значения столбца в свойствах, названных как word.columns ['english'] (строковый тип) вместо словаря word.columns.

Я должен сказать, что мой опыт работы с Python и SQLAlchemy довольно ограничен, возможно, невозможно сделать то, что я пытаюсь сделать.

Любая помощь оценивается,

Спасибо заранее.

Ответ 1

Кажется, что вы можете просто использовать атрибуты напрямую, а не использовать columns dict.

Рассмотрим следующую настройку:

from sqlalchemy import Table, Column, Integer, Unicode, MetaData, create_engine
from sqlalchemy.orm import mapper, create_session

class Word(object):
    pass

wordColumns = ['english', 'korean', 'romanian']
e = create_engine('sqlite://')
metadata = MetaData(bind=e)

t = Table('words', metadata, Column('id', Integer, primary_key=True),
    *(Column(wordCol, Unicode(255)) for wordCol in wordColumns))
metadata.create_all()
mapper(Word, t)
session = create_session(bind=e, autocommit=False, autoflush=True)

С этим пустым классом вы можете сделать:

w = Word()
w.english = u'name'
w.korean = u'이름'
w.romanian = u'nume'

session.add(w)
session.commit()

И когда вы хотите получить доступ к данным:

w = session.query(Word).filter_by(english=u'name').one()
print w.romanian

Чтобы целая sqlalchemy точка ORM вместо использования tuple или dict для доступа к данным, вы используете доступ к атрибутам в своем собственном классе.

Поэтому мне было интересно, почему вы хотите использовать dict. Возможно, это потому, что у вас есть строки с именами языков. Ну, для этого вы могли бы использовать python getattr и setattr вместо этого, как и для любого объекта python:

language = 'korean'
print getattr(w, language)

Это должно решить все ваши проблемы.

Тем не менее, если вы все еще хотите использовать dict -подобный доступ к столбцам, это также возможно. Вам просто нужно реализовать объект dict -like. Теперь я предоставлю код для этого, хотя я считаю, что это абсолютно ненужный беспорядок, поскольку доступ к атрибутам настолько чист. Если ваша проблема уже решена с использованием вышеприведенного метода, не используйте код ниже этой точки.

Вы можете сделать это в классе Word:

class Word(object):
    def __getitem__(self, item): 
        return getattr(self, item)
    def __setitem__(self, item, value):
        return setattr(self, item, value)

Остальная часть настройки работает, как указано выше. Затем вы можете использовать его следующим образом:

w = Word()
w['english'] = u'name'

Если вам нужен атрибут columns, тогда вам понадобится dict -like

class AttributeDict(DictMixin):
    def __init__(self, obj):
        self._obj = obj
    def __getitem__(self, item):
        return getattr(self._obj, item)
    def __setitem__(self, item, value):
        return setattr(self._obj, item, value)

class Word(object):
    def __init__(self):
        self.columns = AttributeDict(self)

Затем вы можете использовать, как вы планировали:

w = Word()
w.columns['english'] = u'name'

Я думаю, вы согласитесь, что все это излишне сложно, без дополнительной выгоды.

Ответ 2

Я использовал решение nosklo (спасибо!), но у меня уже был первичный ключ (переданный как pk_col) в строке столбца (первая строка csv). Поэтому я решил поделиться своей модификацией. Я использовал троицу.

table = Table(tablename, metadata,
    *((Column(pk_col, Integer, primary_key=True)) if rowname == pk_col else (Column(rowname, String())) for rowname in row.keys()))
table.create()