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

Question

我正在使用 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)

谁能告诉我为什么？

Answer 1

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

Answer 2

发现这个要点对我帮助很大。我添加了一个自定义项__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="{}"
)