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 数组值，默认情况下不会产生任何事件。为了使其工作，您需要将mutable extension与ARRAY类型（以及发送这些事件的list子类）结合使用，以便将更改作为与父SessionLog相关的事件发送对象。

Answer 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="{}"
)