如何为SQLAlchemy模型正确添加类型以适配Pydantic？

你的问题核心是SQLAlchemy模型字段的类型注解不规范，导致mypy无法识别查询后实例字段的实际Python类型，误将Column对象的类型当成了字段值类型。

1. 修正SQLAlchemy模型的类型注解 #

在SQLAlchemy 2.0+版本中，必须用Mapped标注字段对应的Python类型，配合Column定义数据库列属性。修改后的UserDB模型如下：

from sqlalchemy import Boolean, Column, DateTime, Integer, String
from sqlalchemy.orm import Mapped, relationship
from sqlalchemy.sql.functions import current_timestamp
from sqlalchemy.ext.declarative import Base
from datetime import datetime

class UserDB(Base):
    __tablename__ = "users"
    id: Mapped[int] = Column(Integer, primary_key=True, index=True, autoincrement=True)
    name: Mapped[str] = Column(String, index=True)
    email: Mapped[str] = Column(String, index=True, nullable=False, unique=True)
    hashed_password: Mapped[str] = Column(String(length=255), nullable=False)
    is_active: Mapped[bool] = Column(Boolean, default=True, nullable=False)
    is_admin: Mapped[bool] = Column(Boolean, default=False, nullable=False)
    created_at: Mapped[datetime] = Column(DateTime, default=current_timestamp(), nullable=False)
    instagram_dms: Mapped[list["InstagramDmDB"]] = relationship("InstagramDmDB", back_populates="user")

    # 若仍需保留to_pydantic方法，此时mypy会正确识别字段类型
    def to_pydantic(self) -> UserPydantic:
        return UserPydantic(
            name=self.name,
            email=self.email,
            is_active=self.is_active,
            is_admin=self.is_admin,
            created_at=self.created_at
        )

2. 更优实现：直接用Pydantic的ORM转换 #

由于你的UserPydantic已开启from_attributes=True（ORM模式），完全不需要手动编写to_pydantic方法，直接通过model_validate即可完成类型安全的转换：

# 查询获取UserDB实例
user_db = db.query(UserDB).first()
# 直接转换为Pydantic模型
user_pydantic = UserPydantic.model_validate(user_db)

这种方式既减少了冗余代码，又避免了手动映射可能出现的类型错误，mypy也能正确识别整个转换过程的类型匹配。

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