Python中ODM模型类动态继承与类名映射优化技术问询

一、优化Database类的动态实现 #

无需维护静态映射字典，直接利用Python内置的getattr()函数动态获取实例属性，这是更符合Python风格的实现方式：

from typing import Any

class User_B1(UserModel, BunnetDocument):
    class Settings:
        name = "user"

class User_B2(UserModel, BeanieDocument):
    class Settings:
        name = "user"

class Database:
    def __init__(self, odm_name: str):
        if odm_name == "bunnet":
            self.User = User_B1
        elif odm_name == "beanie":
            self.User = User_B2
        else:
            raise ValueError("无效的ODM名称")

    def __call__(self, class_name: str) -> Any:
        try:
            return getattr(self, class_name)
        except AttributeError:
            raise ValueError(f"不存在名为{class_name}的模型类")

# 使用示例
if __name__ == "__main__":
    db = Database("bunnet")
    user_cls = db.User
    user_cls = db("User")  # 两种调用方式均生效
    obj = user_cls(
        username="ali2",
        password="Addsasad12365",
        name="ali",
        family="ali",
        email="asdasdoh34@yahoo.com",
    )
    obj.insert()

只要在__init__方法中将对应ODM的模型类绑定为实例属性，__call__就能通过属性名动态获取目标类，彻底摆脱静态字典的维护成本。

二、动态切换User类父类的可行方案 #

完全可以不用维护两个独立的User类，通过动态创建类的方式，根据所选ODM自动切换父类，具体实现如下：

1. 类工厂函数实现 #

from typing import Type
from pydantic import BaseModel

class UserModel(BaseModel):
    username: str
    password: str
    name: str
    family: str
    email: str

def create_user_model(odm_name: str) -> Type:
    # 根据ODM选择对应基类
    if odm_name == "bunnet":
        base_cls = BunnetDocument
    elif odm_name == "beanie":
        base_cls = BeanieDocument
    else:
        raise ValueError("无效的ODM名称")
    
    # 动态生成User类
    User = type(
        "User",
        (UserModel, base_cls),
        {"Settings": type("Settings", (), {"name": "user"})}
    )
    return User

class Database:
    def __init__(self, odm_name: str):
        self.User = create_user_model(odm_name)

    def __call__(self, class_name: str) -> Any:
        try:
            return getattr(self, class_name)
        except AttributeError:
            raise ValueError(f"不存在名为{class_name}的模型类")

# 使用示例
if __name__ == "__main__":
    db = Database("beanie")
    user_cls = db.User
    obj = user_cls(
        username="ali2",
        password="Addsasad12365",
        name="ali",
        family="ali",
        email="asdasdoh34@yahoo.com",
    )
    obj.insert()

2. 关键注意事项 #

• 元类兼容性：确保Beanie和Bunnet的基类元类无冲突，目前主流Pydantic兼容的ODM都能正常适配动态类创建逻辑。
• 模型注册步骤：部分ODM（如Beanie）需要将模型注册到数据库连接，动态生成类后需完成对应注册操作（例如Beanie的beanie.init_beanie需包含动态生成的User类）。
• IDE提示优化：动态生成的类可能缺失IDE自动补全，若需要完善类型提示，可结合typing.TypeVar和Generic进行扩展。

3. 方案对比 #

• 动态类方案：大幅减少重复代码，适合不同ODM下模型结构完全一致的场景，但调试和IDE支持略弱。
• 独立类方案：结构更直观，调试便捷，适合不同ODM下模型存在细微差异的场景。

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