Flask Postgresql 数组不会永久更新

Question

我正在使用 Flask 和 PostgreSQL 数据库以及 SQLAlchemy 进行一个项目。

我有一个Group对象，其中User包含一组成员的 ID。出于某种原因，当我尝试将 ID 添加到组时，它无法正确保存。

如果我尝试members.append(user_id)，它似乎根本不起作用。但是，如果我尝试members += [user_id]，该 id 将显示在列出所有组的视图中，但是如果我重新启动服务器，则添加的值不存在。然而，初始值是。

相关代码：

最初将组添加到数据库：

db = SQLAlchemy(app)
# ...
g = Group(request.form['name'], user_id)
db.session.add(g)
db.session.commit()

组类：

from flask.ext.sqlalchemy import SQLAlchemy
from sqlalchemy.dialects.postgresql import ARRAY

class Group(db.Model):
    __tablename__ = "groups"
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(128))
    leader = db.Column(db.Integer)

    # list of the members in the group based on user id
    members = db.Column(ARRAY(db.Integer))

    def __init__(self, name, leader):
        self.name = name
        self.leader = leader
        self.members = [leader]

    def __repr__(self):
        return "Name: {}, Leader: {}, Members: {}".format(self.name, self.leader, self.members)

    def add_user(self, user_id):
        self.members += [user_id]

我用于更新组的测试功能：

def add_2_to_group():
    g = Group.query.all()[0]
    g.add_user(2)
    db.session.commit()
    return redirect(url_for('show_groups'))

谢谢你的帮助！

Answer 1

正如您所提到的，ARRAYsqlalchemy 中的数据类型是不可变的。这意味着一旦初始化就无法将新数据添加到数组中。

要解决此问题，请创建 class MutableList。

from sqlalchemy.ext.mutable import Mutable

class MutableList(Mutable, list):
    def append(self, value):
        list.append(self, value)
        self.changed()

    @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

此代码段允许您扩展列表以为其添加可变性。所以，现在你可以使用上面的类来创建一个可变数组类型，如：

class Group(db.Model):
    ...
    members = db.Column(MutableList.as_mutable(ARRAY(db.Integer)))
    ...