如何通过关联类构造器处理SQLAlchemy关联数据？附代码示例

嘿，我来帮你搞明白怎么用关联类（Association Object）处理SQLAlchemy里的关联数据～ 关联类主要用来应对多对多关系中需要额外存储关联信息的场景（比如关联时间、权限等级这类字段），也能让你更灵活地控制关联逻辑。结合你给出的代码基础，咱们一步步实操：

1. 定义基础模型与关联类 #

基于你已有的Model基类，先创建两个需要关联的主模型（比如User和Group），再定义关联类UserGroup来存储它们的关联关系和额外字段：

# coding: utf-8
from sqlalchemy import Column, VARCHAR, ForeignKey, create_engine, DateTime
from sqlalchemy.dialects.mysql import INTEGER, TINYINT
from sqlalchemy.orm import relationship, sessionmaker
from sqlalchemy.inspection import inspect
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.ext.associationproxy import association_proxy
from datetime import datetime

engine = create_engine('mysql://root:toor@localhost/test', echo=True)
Model = declarative_base()
Session = sessionmaker(bind=engine)
session = Session()

# 关联类：处理User和Group的多对多关联，包含额外字段join_date
class UserGroup(Model):
    __tablename__ = 'user_group'
    user_id = Column(INTEGER, ForeignKey('user.id'), primary_key=True)
    group_id = Column(INTEGER, ForeignKey('group.id'), primary_key=True)
    join_date = Column(DateTime, default=datetime.utcnow)  # 额外关联字段：加入时间

    # 与主模型建立双向关系绑定
    user = relationship("User", back_populates="group_associations")
    group = relationship("Group", back_populates="user_associations")

# 主模型1：用户
class User(Model):
    __tablename__ = 'user'
    id = Column(INTEGER, primary_key=True, autoincrement=True)
    name = Column(VARCHAR(50), nullable=False)

    # 指向关联类的关系
    group_associations = relationship("UserGroup", back_populates="user")
    # 用association_proxy简化访问：直接关联到Group，跳过关联类
    groups = association_proxy("group_associations", "group")

# 主模型2：群组
class Group(Model):
    __tablename__ = 'group'
    id = Column(INTEGER, primary_key=True, autoincrement=True)
    name = Column(VARCHAR(50), nullable=False)

    # 指向关联类的关系
    user_associations = relationship("UserGroup", back_populates="group")
    # 用association_proxy简化访问：直接关联到User，跳过关联类
    users = association_proxy("user_associations", "user")

# 创建所有表结构
Model.metadata.create_all(engine)

2. 用关联类构造器创建关联数据 #

现在可以通过两种方式创建带额外字段的关联关系：

方式一：直接实例化关联类（适合需要设置额外字段的场景） #

# 创建用户和群组实例
user1 = User(name="Alice")
group1 = Group(name="Developers")

# 实例化关联类，手动设置额外字段join_date
user_group = UserGroup(user=user1, group=group1, join_date=datetime(2024, 1, 1))

# 添加到会话并提交
session.add_all([user1, group1, user_group])
session.commit()

方式二：通过主模型的关联属性添加（适合额外字段有默认值的场景） #

借助association_proxy，可以跳过关联类直接操作主模型的关联关系：

user2 = User(name="Bob")
group2 = Group(name="Designers")

# 直接将群组添加到用户的groups属性，SQLAlchemy会自动创建关联类实例
user2.groups.append(group2)

session.add_all([user2, group2])
session.commit()

3. 查询关联数据（包含关联类的额外字段） #

查询时可以轻松获取关联类里的额外信息：

查询用户的所有关联群组及加入时间 #

user = session.query(User).filter_by(name="Alice").first()
for assoc in user.group_associations:
    print(f"用户 {user.name} 加入群组 {assoc.group.name} 的时间：{assoc.join_date}")

查询群组的所有关联用户及加入时间 #

group = session.query(Group).filter_by(name="Developers").first()
for assoc in group.user_associations:
    print(f"群组 {group.name} 的成员 {assoc.user.name} 加入时间：{assoc.join_date}")

关键要点总结 #

• 关联类本质是普通模型类，通过外键关联两个主模型，同时支持添加任意额外字段。
• 用relationship在关联类和主模型之间建立双向绑定，确保数据操作的一致性。
• association_proxy是简化工具，能让你跳过关联类直接访问关联的主模型；但如果需要操作关联类的额外字段，还是得直接实例化关联类。

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