SQLAlchemy表关联模式解析：基于表继承的通用关联实现

最新推荐文章于 2025-06-11 09:14:03 发布

翟万实Robust

最新推荐文章于 2025-06-11 09:14:03 发布

阅读量296

点赞数 5

本文链接：https://blog.csdn.net/gitblog_00164/article/details/148415597

版权

SQLAlchemy表关联模式解析：基于表继承的通用关联实现

sqlalchemy The Database Toolkit for Python 项目地址: https://gitcode.com/gh_mirrors/sq/sqlalchemy

概述

在SQLAlchemy中，处理不同实体间的关联关系有多种模式。本文将深入分析一种高效且灵活的关联模式——基于表继承的通用关联实现（Table per Related）。这种模式特别适合需要为不同父类维护独立关联表的场景。

核心设计理念

这种关联模式的核心思想是：为每个需要关联的父类动态生成专属的关联表。相比将所有关联数据存储在单一表中的方案，这种设计具有以下优势：

数据隔离性：每个父类的关联数据完全独立存储，不会相互影响
查询效率高：关联查询直接针对特定表，无需额外过滤条件
结构清晰：关联关系直观且双向导航简单
扩展性好：新增父类类型时自动创建对应的关联表

实现解析

基础架构

首先定义一个基础Base类，使用@as_declarative()装饰器将其转换为声明性基类。这个基类自动提供表名生成和主键列：

@as_declarative()
class Base:
    @declared_attr
    def __tablename__(cls):
        return cls.__name__.lower()
    
    id = Column(Integer, primary_key=True)

地址模型设计

定义Address混合类，包含地址的通用字段。注意这是一个mixin类，不是完整的模型：

class Address:
    street = Column(String)
    city = Column(String)
    zip = Column(String)
    
    def __repr__(self):
        return f"{self.__class__.__name__}(street={self.street!r}, ...)"

动态关联表生成

HasAddresses混合类是实现动态关联表的核心。它使用@declared_attr装饰器动态创建每个父类专属的地址类：

class HasAddresses:
    @declared_attr
    def addresses(cls):
        cls.Address = type(
            f"{cls.__name__}Address",
            (Address, Base),
            dict(
                __tablename__=f"{cls.__tablename__}_address",
                parent_id=Column(Integer, ForeignKey(f"{cls.__tablename__}.id")),
                parent=relationship(cls)
            )
        )
        return relationship(cls.Address)

这段代码的关键点在于：

使用Python的type()函数动态创建新类
新类继承自Address和Base，获得地址字段和ORM功能
自动设置表名和与父表的外键关系
建立双向关系(parent属性)

业务模型定义

定义具体的业务模型，只需继承HasAddresses即可自动获得地址关联能力：

class Customer(HasAddresses, Base):
    name = Column(String)

class Supplier(HasAddresses, Base):
    company_name = Column(String)

实际应用示例

创建并填充数据的示例展示了这种模式的实际使用：

# 创建表结构
Base.metadata.create_all(engine)

# 添加数据
session.add_all([
    Customer(
        name="customer 1",
        addresses=[
            Customer.Address(street="123 anywhere street", ...),
            Customer.Address(street="40 main street", ...)
        ]
    ),
    Supplier(
        company_name="Ace Hammers",
        addresses=[Supplier.Address(street="2569 west elm", ...)]
    )
])

# 查询示例
for customer in session.query(Customer):
    for address in customer.addresses:
        print(address, address.parent)