SQLAlchemy+Pydantic：如何让数据库自动设置id字段而非初始化必填

问题原因 #

你的Transcript模型中id字段被定义为必填的uuid.UUID类型，但你希望它由数据库自动生成，无需在初始化时传入。由于使用了MappedAsDataclass结合Pydantic的dataclass，未设置默认值的字段会被标记为必填参数，导致初始化时缺少id报错。

解决方案 #

修改id字段的定义，使其成为可选类型并设置默认值为None，同时保留数据库端的自动生成规则：

修正后的代码 #

import uuid
from typing import Optional
from sqlalchemy import UUID, Text, String
from sqlalchemy.orm import DeclarativeBase, MappedAsDataclass, Mapped, mapped_column
import sqlalchemy as sa
import pydantic

class Base(
    MappedAsDataclass,
    DeclarativeBase,
    dataclass_callable=pydantic.dataclasses.dataclass,
):
    class Config:
        arbitrary_types_allowed = True

class Transcript(Base):
    __tablename__ = "Transcript"

    id: Mapped[Optional[uuid.UUID]] = mapped_column(
        "id",
        UUID(as_uuid=True),
        primary_key=True,
        index=True,
        server_default=sa.text("gen_random_uuid()"),
        default=None,  # 设置Python端默认值为None
    )
    text: Mapped[str] = mapped_column("text", Text)
    display_name: Mapped[str] = mapped_column("display_name", String)

说明 #

1. 可选类型声明：将id的类型改为Optional[uuid.UUID]（Python 3.10+可简化为uuid.UUID | None），告诉Pydantic该字段是可选的。
2. Python端默认值：添加default=None到mapped_column，让dataclass在初始化时自动为id赋值None，无需手动传入。
3. 数据库自动生成：保留server_default=sa.text("gen_random_uuid()")，确保插入数据时由数据库生成UUID值。
4. 获取生成的id：插入数据库并提交会话后，SQLAlchemy会自动刷新对象，你可以直接访问db_transcript.id获取生成的UUID。

使用示例 #

# 初始化对象时无需传入id
transcript_data = {"text": "hello", "display_name": "world"}
db_transcript = Transcript(**transcript_data)

# 存入数据库
from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine

engine = create_engine("your_database_url")
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

with SessionLocal() as session:
    session.add(db_transcript)
    session.commit()
    session.refresh(db_transcript)  # 确保id被加载

print(db_transcript.id)  # 输出数据库生成的UUID

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