如何简化SQLAlchemy中1-N关系的整数列表子属性操作？

可以用SQLAlchemy的association_proxy来实现这个需求，它能把一对多关系的关联对象转换成普通整数列表来读写，同时保留原始关系用于直接操作DrawNumber类。

步骤1：定义模型并添加关联代理 #

首先导入association_proxy，然后调整模型结构：

from sqlalchemy import Column, Integer, ForeignKey
from sqlalchemy.orm import relationship, sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.ext.associationproxy import association_proxy

Base = declarative_base()

class DrawNumber(Base):
    __tablename__ = 'draw_numbers'
    id = Column(Integer, primary_key=True)
    number = Column(Integer, nullable=False)
    lottery_draw_id = Column(Integer, ForeignKey('lottery_draws.id'))

class LotteryDraw(Base):
    __tablename__ = 'lottery_draws'
    id = Column(Integer, primary_key=True)
    # 保留原始关系，用于直接操作DrawNumber对象
    _numbers = relationship("DrawNumber", backref="lottery_draw")
    # 关联代理：将_numbers中的DrawNumber对象映射为整数列表
    numbers = association_proxy(
        '_numbers', 
        'number', 
        creator=lambda num: DrawNumber(number=num)
    )

步骤2：直接以整数列表方式操作numbers属性 #

现在你可以像操作普通列表一样读写LotteryDraw的numbers属性：

# 初始化会话
Session = sessionmaker()
session = Session()

# 创建抽奖记录并直接赋值整数列表
draw = LotteryDraw()
draw.numbers = [5, 12, 23, 37]

# 追加整数
draw.numbers.append(41)

# 遍历整数列表
for num in draw.numbers:
    print(num)  # 输出：5 12 23 37 41

# 保存到数据库
session.add(draw)
session.commit()

步骤3：保留直接操作DrawNumber的能力 #

如果需要直接操作draw_numbers表的DrawNumber对象，依然可以通过原始的_numbers关系进行：

# 添加新的DrawNumber对象
draw._numbers.append(DrawNumber(number=50))
session.commit()

# 此时numbers属性会自动包含新添加的整数
print(draw.numbers)  # 输出：[5, 12, 23, 37, 41, 50]

原理说明 #

• association_proxy会自动代理对numbers属性的所有列表操作（如赋值、追加、删除等），将其转换为对_numbers关系中DrawNumber对象的操作。
• creator参数指定了如何将整数转换为DrawNumber实例，当你添加新整数时，SQLAlchemy会自动调用这个lambda创建对应的对象。

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