使用更改的python dict更新SQLAlchemy orm对象

问题描述 投票:1回答:1

实现可变dict的问题将对内容的变化作出反应。我设法设置SQLAlchemy来管理添加和删除。但是,对存储的dict内容的更改不会“触发”SQLAlchemy db的更新。

我在stackoverflow上找到了一些其他问题,建议:

默认情况下,SQLAlchemy不会跟踪dict属性中的更改。要使其跟踪更改,您可以使用可变扩展:

我按照How to implement mutable PickleTypes that automatically update on change的例子

但是,我无法让它发挥作用。在我的示例中,当我从Column(PickleType)更改为Column(MutableDict.as_mutable(PickleType))时,SQLAlchemy会话不再找到该对象。下面的代码说明了我正在尝试做的事情。

第一个代码是我设置数据库的地方,第二个代码我正在尝试向一个人添加交易。我设法添加事务并删除它们但不更改它们。因此,为什么我尝试使用MutableDict类,但我似乎并不完全理解它。

设置SQL db:(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过滤器找不到与transaction1字典匹配的任何事务。我怀疑这与dict存储为MutableDict而不是Dict的事实有关。但是我该如何处理,以及如何在添加之后编辑和更改我的transaction1。

TLDR:我想改变我的dict中使用SQLAlchemy存储为ORM的内容。

python sqlalchemy
1个回答
0
投票

我意识到我只是不使用.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)
© www.soinside.com 2019 - 2024. All rights reserved.