如何将Pydantic模型导入SQLModel？解决手动添加表的KeyError问题

核心问题分析 #

你之前遇到的KeyError是因为未继承SQLModel的类不会自动注册到SQLModel.metadata中，元数据里根本不存在对应的表项，手动调用create自然报错。要解决这个问题，必须动态生成继承自SQLModel的类，让SQLModel自动完成元数据注册。

动态生成SQLModel的可行代码 #

直接从字典生成可用于数据库操作的SQLModel类，代码如下：

from sqlmodel import SQLModel, Field
from sqlalchemy import create_engine

# 你的字段字典
model_dict = {"feature_a": int, "feature_b": str}

# 动态构建SQLModel类
Sopro = type(
    "Sopro",  # 类名
    (SQLModel,),  # 继承SQLModel
    {
        "__tablename__": "sopro",  # 显式指定数据库表名，避免自动生成的不一致
        # 遍历字典生成带Field的数据库字段
        **{field: (type_, Field()) for field, type_ in model_dict.items()},
        # 添加主键字段（根据需求可选，建议保留）
        "id": (int, Field(primary_key=True)),
    }
)

# 初始化数据库引擎
engine = create_engine("sqlite:///your_database.db")

# 现在可以正常创建所有注册的表
SQLModel.metadata.create_all(engine)

关键注意事项 #

• 必须继承SQLModel：只有继承该类，SQLModel才会自动将表结构注册到metadata中
• 显式指定__tablename__：避免SQLModel根据类名自动生成表名时出现偏差，确保和你期望的表名一致
• 用Field包装字段：普通Pydantic字段不会被识别为数据库字段，必须通过Field()标记，才能生成对应的表列

现有Pydantic模型转SQLModel的方法 #

如果已经有现成的Pydantic模型，无需重新定义，直接多继承即可：

from pydantic import BaseModel
from sqlmodel import SQLModel, Field

# 已有的Pydantic模型
class SoproPydantic(BaseModel):
    feature_a: int
    feature_b: str

# 转换为SQLModel
class SoproSQL(SQLModel, SoproPydantic):
    __tablename__ = "sopro"
    id: int = Field(primary_key=True)

# 正常创建表
SQLModel.metadata.create_all(engine)

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