Обновление объекта SQLAlchemy orm с измененным значением python - PullRequest
1 голос
/ 13 апреля 2019

Проблемы с реализацией изменчивого dict, который будет реагировать на изменения в контенте. Мне удалось настроить SQLAlchemy для управления добавлением и удалением. Тем не менее, изменения в содержании сохраненного dict не «запускают» обновление базы данных SQLAlchemy.

Я нашел здесь несколько других вопросов о stackoverflow, которые предложили:

По умолчанию SQLAlchemy не отслеживает изменения внутри атрибутов dict. Чтобы отслеживать изменения, вы можете использовать изменяемое расширение:

Я последовал примеру здесь Как реализовать изменяемые типы PickleType, которые автоматически обновляются при изменении

Однако я не могу заставить его работать. В моем примере, когда я меняю Column (PickleType) на Column (MutableDict.as_mutable (PickleType)), сеанс SQLAlchemy больше не находит объект. Код ниже иллюстрирует то, что я пытаюсь сделать.

Первый код - это то, где я настраиваю базу данных, а второй блок, который я пытаюсь добавить транзакции человеку. Мне удается добавлять транзакции и удалять их, но не изменять их. Следовательно, почему я попытался с классом MutableDict, но я, кажется, не полностью понимаю это.

Настройка базы данных SQL: (sqlalchemy_declarative.py)

from sqlalchemy import Column, ForeignKey, Integer, String, PickleType
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.ext.mutable import Mutable
from sqlalchemy.orm import relationship
from sqlalchemy import create_engine

Base = declarative_base()


class MutableDict(Mutable, dict):

    @classmethod
    def coerce(cls, key, value):
        if not isinstance(value, MutableDict):
            if isinstance(value, dict):
                return MutableDict(value)
            return Mutable.coerce(key, value)
        else:
            return value

    def __delitem(self, key):
        dict.__delitem__(self, key)
        self.changed()

    def __setitem__(self, key, value):
        dict.__setitem__(self, key, value)
        self.changed()

    def __getstate__(self):
        return dict(self)

    def __setstate__(self, state):
        self.update(self)


class Person(Base):
    __tablename__ = 'person_object'
    # Here we define columns for the table person
    # Notice that each column is also a normal Python instance attribute.
    id = Column(Integer, primary_key=True)
    first_name = Column(String, nullable=False)
    last_name = Column(String, nullable=False)

    def __str__(self):  # prints when treated as string (for use interface)
        return f"Primary id: {self.id} \n" \
               f"First name: {self.first_name}"


class Transactions(Base):
    __tablename__ = 'transactions'
    # Here we define columns for the table address.
    # Notice that each column is also a normal Python instance attribute.
    id = Column(Integer, primary_key=True)
    transactions = Column(MutableDict.as_mutable(PickleType))
    # transactions = Column(PickleType)
    person_object_id = Column(Integer, ForeignKey('person_object.id'))
    person_object = relationship(Person)

    def update(self, tmp_dict):
        for key, value in tmp_dict.items():
            print(key, value)
            setattr(self, self.transactions[f'{key}'], value)


def create_db():
    # Create an engine that stores data in the local directory's
    # sqlalchemy_example.db file.
    engine = create_engine('sqlite:///person.db')

    # Create all tables in the engine. This is equivalent to "Create Table"
    # statements in raw SQL.
    Base.metadata.create_all(engine)

Пример кода:

from sqlalchemy_declarative import Person, Base, Transactions
from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine

import sqlalchemy_declarative as sqlalchemy_declarative


def main():
    sqlalchemy_declarative.create_db()
    engine = create_engine('sqlite:///person.db')
    Base.metadata.bind = engine
    db_session = sessionmaker()
    db_session.bind = engine
    session = db_session()

    transaction1 = {'item1': 'banana',
                    'item2': 'apple',
                    'item3': 'sugar',
                    'item4': 'coke',
                    'item5': 'candy'}
    transaction2 = {'item1': 'pizza',
                    'item2': 'water'}

    new_obj = Person(first_name='Bob', last_name='Smith')
    session.add(new_obj)

    new_transaction = Transactions(transactions=transaction1, person_object=new_obj)
    session.add(new_transaction)

    new_transaction = Transactions(transactions=transaction2, person_object=new_obj)
    session.add(new_transaction)
    session.commit()

    test = session.query(Transactions).all()

    for tmp in test:
        print(type(tmp.transactions))
        print(tmp.transactions == transaction1)

    test2 = session.query(Transactions).filter(Transactions.transactions == transaction1).all()
    print(test2)

    transaction1 = {'item1': 'banana',
                    'item2': 'apple',
                    'item3': 'sugar',
                    'item4': 'coke',
                    'item5': 'pineapple'}
    test2.update(transaction1)
    session.commit()

    all_transactions = session.query(Transactions).all()

    for tmp in all_transactions:
        print(tmp.transactions)


if __name__ == '__main__':
    main()

Однако фильтр test2 не находит транзакции, которая соответствует словарю транзакции1. Я подозреваю, что это связано с тем, что dict хранится как MutableDict, а не Dict. Но что мне с этим делать и как мне отредактировать и изменить мою транзакцию1 после ее добавления.

TLDR: я хочу изменить содержимое моего dict, которое хранится как ORM с использованием SQLAlchemy.

1 Ответ

0 голосов
/ 25 апреля 2019

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

    transaction1 = {'item1': 'banana',
                    'item2': 'apple',
                    'item3': 'sugar',
                    'item4': 'coke',
                    'item5': 'candy'}
    transaction2 = {'item1': 'pizza',
                    'item2': 'water'}

    new_obj = Person(first_name='Bob', last_name='Smith')
    session.add(new_obj)

    new_transaction = Transactions(transactions=transaction1, person_object=new_obj)
    session.add(new_transaction)

    new_transaction = Transactions(transactions=transaction2, person_object=new_obj)
    session.add(new_transaction)
    session.commit()

    test = session.query(Transactions).all()

    for tmp in test:
        print(tmp.transactions)

    new_transaction.transactions['item1'] = 'curry banana'
    session.commit()

    test = session.query(Transactions).all()

    for tmp in test:
        print(tmp.transactions)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...