SQLAlchemy如何从外键列获取关联模型或模型名称

SQLAlchemy 原生提供了完整的元数据访问接口，可以直接通过外键字段的属性拿到指向的关联表/模型，不需要额外依赖，以下是可直接使用的实现方式：

1. 核心实现逻辑 #

前置示例模型定义 #

from sqlalchemy import Column, Integer, String, ForeignKey
from sqlalchemy.orm import declarative_base, relationship

Base = declarative_base()

class Department(Base):
    __tablename__ = "department"
    id = Column(Integer, primary_key=True)
    name = Column(String)
    employees = relationship("Employee", back_populates="dept")

class Employee(Base):
    __tablename__ = "employee"
    id = Column(Integer, primary_key=True)
    name = Column(String)
    # 外键字段，指向department表的id列
    dept_id = Column(ForeignKey("department.id"))
    dept = relationship("Department", back_populates="employees")

获取关联模型的步骤 #

• 从模型类中取到外键字段的属性对象：fk_col = Employee.dept_id
• 读取该字段绑定的外键约束：fk_constraint = fk_col.property.columns[0].foreign_keys.pop()
• 拿到外键指向的关联表对象：related_table = fk_constraint.column.table
• 从注册表中映射得到关联模型类：related_model = Base.registry.mappers[related_table].class_
• 直接获取模型名称：related_model.__name__，此处输出为 Department

该方法不依赖relationship的定义，只要外键字段声明存在即可正常使用

注意：单个字段绑定多个外键约束属于极特殊的设计场景，常规业务开发中一个外键字段只会对应一个关联表，直接使用pop()获取即可；多外键场景遍历foreign_keys集合筛选即可。

2. 通用函数封装 #

可以直接封装为通用工具函数用于程序化调用：

def get_related_model_by_fk(current_model, fk_field_name: str):
    """
    从当前模型的指定外键字段，获取其指向的关联模型类
    :param current_model: 定义外键的当前模型类
    :param fk_field_name: 外键字段的字符串名称
    :return: 外键指向的关联模型类
    """
    fk_col = getattr(current_model, fk_field_name)
    fk = fk_col.property.columns[0].foreign_keys.pop()
    return current_model.registry.mappers[fk.column.table].class_

使用示例：

related_model = get_related_model_by_fk(Employee, "dept_id")
# 验证关联模型正确性
assert related_model is Department

3. 反射表场景（无对应模型类） #

如果是通过反射直接获取的表对象，没有绑定的模型类，可以直接获取关联表名：

# 假设已通过反射拿到employee表对象
employee_table = Base.metadata.tables["employee"]
fk_constraint = employee_table.c.dept_id.foreign_keys.pop()
related_table_name = fk_constraint.column.table.name
# 输出为 department

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