SQLAlchemy 使用合并更新 PostgreSQL 数组不起作用

Posted 2023-03-23

【中文标题】SQLAlchemy 使用合并更新 PostgreSQL 数组不起作用

【英文标题】：SQLAlchemy update PostgreSQL array using merge not work

【发布时间】：2013-12-19 03:58:45

【问题描述】：

我正在使用 SQLAlchemy 访问 PostgreSQL 数据库，并且我这样定义对象：

class SessionLog(Base):
    __tablename__ = 'session_log'

    id = Column(Integer, primary_key=True)
    recordFile = Column('record_file', String(128))
    appSrcPorts = Column('app_src_ports', ARRAY(Integer))
    info5 = Column('info5', String(100))

然后我像这样选择并更新 session_log 表：

session = Session()
sessionLog = session.query(SessionLog).filter_by(id=sessionLogId).first()
sessionLog.appSrcPorts.append(1)
session.merge(sessionLog)
session.commit()

但奇怪的是，在我调用 merge() 和 commit() 之后，“app_src_ports”列没有更新。 我找到了一个丑陋的方法让它工作，在 append() 行之前，添加这个：

sessionLog.appSrcPorts = list(sessionLog.appSrcPorts)

谁能告诉我为什么？

【问题讨论】：

你的解决方案（sessionLog.appSrcPorts = list(sessionLog.appSrcPorts) 使它成为一个列表作为一个魅力:)

【参考方案1】：

SQLAlchemy ORM 依赖于事件检测来判断某些数据何时发生更改，从而何时需要刷新数据。在这种情况下，您正在就地更改 Python 数组值，默认情况下不会产生任何事件。为了使其工作，您需要将mutable extension与ARRAY类型（以及发送这些事件的list子类）结合使用，以便将更改作为与父SessionLog相关的事件发送对象。

【讨论】：

你认为这是 SQLAlchemy 的错误吗？

好吧，我们应该在 sqlalchemy.ext.mutable 中准备好一个“MutableList”实现，我们现在还没有（只有 MutableDict）。而且，ORM 可以引入一个系统，通过该系统类型可以宣传其“可变”版本，从而减少样板的数量，以便将 Mutable 类型用于像 ARRAY 这样的标准案例。但这些都是增强功能，我们不称之为错误。

【参考方案2】：

找到this gist，这对我帮助很大。我添加了一个自定义 __setitem__ 以便能够更改列表中的项目，并添加了 __delitem__ 以删除一个：

from sqlalchemy.ext.mutable import Mutable
from sqlalchemy.dialects.postgresql import ARRAY

class MutableList(Mutable, list):

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

    def __delitem__(self, key):
        list.__delitem__(self, key)
        self.changed()

    def append(self, value):
        list.append(self, value)
        self.changed()

    def pop(self, index=0):
        value = list.pop(self, index)
        self.changed()
        return value

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

然后在你的模型中：

your_field = db.Column(
    MutableList.as_mutable(ARRAY(db.String())),
    server_default=""
)

【讨论】：

SQLA 提供MutableList 作为扩展，不再需要自定义实现（大多数情况下，不提供嵌套对象的跟踪）：docs.sqlalchemy.org/en/latest/orm/extensions/…