SQLAlchemy使用merge无法更新PostgreSQL数组

时间:2013-12-19 03:58:45

标签: python sql postgresql sqlalchemy

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

任何人都可以告诉我为什么?

2 个答案:

答案 0 :(得分:3)

SQLAlchemy ORM依赖于事件的检测,以便判断某些数据何时发生了变化,从而何时需要刷新数据。在这种情况下,您将就地更改Python数组值,默认情况下不会生成任何事件。为了实现这一点,您需要将mutable extension与ARRAY类型(以及发送这些事件的list子类)结合使用,以便将更改作为与之相关的事件发送。父SessionLog对象。

答案 1 :(得分:0)

发现了this gist,这对我有很大帮助。我添加了一个自定义embeddedServer(Netty, port = 8080, host = "localhost") 来更改列表中的一项,并添加了__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