Flask-SQLAlchemy如何操作多对多关联表的ordinality属性

当多对多关联需要存储额外字段（比如你的ordinalite）时，不能再用普通的db.Table关联表，必须改用关联对象模型来实现。以下是具体步骤：

1. 定义关联对象模型 #

创建一个独立的模型类来替代原有的关联表，专门存储关联关系和ordinalite字段：

from flask_sqlalchemy import SQLAlchemy
from sqlalchemy.ext.associationproxy import association_proxy

db = SQLAlchemy()

class PageVisualisation(db.Model):
    __tablename__ = 'page_visualisation'
    # 复合主键确保关联关系唯一
    page_id = db.Column(db.Integer, db.ForeignKey('page.id'), primary_key=True)
    visualisation_id = db.Column(db.Integer, db.ForeignKey('visualisation.id'), primary_key=True)
    # 自定义排序字段，设置为非空
    ordinalite = db.Column(db.Integer, nullable=False)
    
    # 关联到两个主模型
    page = db.relationship('Page', back_populates='page_visualisations')
    visualisation = db.relationship('Visualisation', back_populates='page_visualisations')

2. 更新主模型的关系定义 #

修改Page和Visualisation模型，将原有的多对多关系替换为与关联模型的关系，同时可通过association_proxy简化对可视化对象的直接访问：

class Page(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    localisation = db.Column(db.String)
    
    # 关联到中间模型，设置级联删除确保中间表数据同步清理
    page_visualisations = db.relationship(
        'PageVisualisation', 
        back_populates='page', 
        cascade='all, delete-orphan'
    )
    # 关联代理：直接通过page.visualisations访问关联的Visualisation对象（可选）
    visualisations = association_proxy('page_visualisations', 'visualisation')

class Visualisation(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    graphe_id = db.Column(db.String)
    
    page_visualisations = db.relationship(
        'PageVisualisation', 
        back_populates='visualisation'
    )

3. 插入带自定义ordinalite的数据 #

现在需要通过创建PageVisualisation实例来关联页面和可视化对象，并手动设置排序值：

v1 = Visualisation(graphe_id="graphe_id_1")
v2 = Visualisation(graphe_id="graphe_id_2")

page = Page(localisation="localisation")

# 创建关联对象并设置ordinalite
pv1 = PageVisualisation(visualisation=v1, ordinalite=1)
pv2 = PageVisualisation(visualisation=v2, ordinalite=2)

# 将关联对象添加到页面的关联集合中
page.page_visualisations.extend([pv1, pv2])

db.session.add(page)
db.session.commit()

如果想贴近你伪代码的写法，可以给Page模型添加一个辅助方法简化操作：

class Page(db.Model):
    # ... 已有字段和关系 ...
    
    def add_visualisation(self, visualisation, ordinalite):
        """辅助方法：添加可视化对象并设置排序值"""
        pv = PageVisualisation(visualisation=visualisation, ordinalite=ordinalite)
        self.page_visualisations.append(pv)

使用时可简化为：

v1 = Visualisation(graphe_id="graphe_id_1")
v2 = Visualisation(graphe_id="graphe_id_2")

page = Page(localisation="localisation")
page.add_visualisation(v1, 1)
page.add_visualisation(v2, 2)

db.session.add(page)
db.session.commit()

4. 后续编辑ordinalite字段 #

要修改已关联的排序值，直接找到对应的PageVisualisation实例更新即可：

# 方式1：通过查询定位关联对象
pv = PageVisualisation.query.filter_by(page_id=page.id, visualisation_id=v1.id).first()
if pv:
    pv.ordinalite = 3
    db.session.commit()

# 方式2：遍历页面的关联集合修改
for pv in page.page_visualisations:
    if pv.visualisation.graphe_id == "graphe_id_1":
        pv.ordinalite = 3
        break
db.session.commit()

内容的提问来源于stack exchange，提问作者gau000