如何正确实现SQLAlchemy共享ORM类或反射已声明ORM类？

一、替代访问_class_registry的优雅方案 #

你之前通过访问DeclarativeBase注册表的保护属性_class_registry获取ORM类，这种方式依赖内部实现，不够稳健。可以改用SQLAlchemy提供的公开API——Registry.classes属性，它是一个包含所有已注册ORM类的集合，直接遍历匹配类名即可：

def _get_class_from_registry(self, class_name: str) -> None | Type[DeclarativeBase]:
    # 遍历公开的classes集合，匹配类名
    return next(
        (cls for cls in self.db_base_cls.registry.classes if cls.__name__ == class_name),
        None
    )

说明 #

Base.registry.classes是SQLAlchemy官方暴露的公开属性，包含所有继承自该DeclarativeBase的ORM类，完全不需要依赖内部保护属性，代码兼容性和可维护性更好。

二、修复automap_base的使用问题 #

你的测试代码中手动定义的User类未出现在Base.classes里，主要有两个原因：

1. 执行Base.prepare的时机错误：在创建表之前就尝试反射，数据库中还不存在user表，automap无法生成对应类；
2. Base.classes仅存储automap自动生成的ORM类，手动定义的类不会自动加入其中。

修正后的代码 #

import os

from sqlalchemy import Column, String, create_engine
from sqlalchemy.ext.automap import automap_base
from sqlalchemy.orm import Session

# 清理旧数据库文件
if os.path.exists("my.db"):
    os.remove("my.db")

# 初始化automap基类
Base = automap_base()

# 手动定义ORM类，继承自automap的Base
class User(Base):
    __tablename__ = "user"
    name = Column(String, primary_key=True)

# 创建数据库引擎
engine = create_engine("sqlite+pysqlite:///my.db", echo=True)

# 先创建所有表（包括手动定义的user表）
Base.metadata.create_all(engine)

# 执行automap准备：反射数据库表，复用已手动定义的类，生成未定义的类
Base.prepare(autoload_with=engine, reflect=True)

# 统一获取ORM类：从registry的classes集合中查找（包含手动定义和automap生成的类）
user_cls = next(cls for cls in Base.registry.classes if cls.__name__ == "User")

# 测试数据插入
with Session(engine) as session:
    session.add(User(name="test1"))
    session.commit()

# 测试手动定义类的查询
with Session(engine) as session:
    u1 = session.query(User).first()
    print(u1.name)

# 测试从registry获取类的查询
with Session(engine) as session:
    u2 = session.query(user_cls).first()
    print(u2.name)

关键修正点 #

• 调整执行顺序：先调用Base.metadata.create_all(engine)创建表，再执行Base.prepare，确保automap能反射到已存在的表；
• 统一访问入口：用Base.registry.classes替代Base.classes，它能获取所有已注册的ORM类（包括手动定义和automap自动生成的）。

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