如何使用mypy对alembic进行正确的类型检查

1. 安装Alembic的类型补充包 #

直接安装官方维护的类型包就能解决大部分类型提示问题，执行以下命令：

pip install types-alembic

这个包提供了Alembic核心模块（包括op、context）的完整类型注释，Mypy就能正确识别这些属性了。

2. 检查Mypy配置（必要时调整） #

如果安装类型包后仍有报错，检查你的Mypy配置文件（mypy.ini、pyproject.toml或setup.cfg）：

• 确保配置中没有禁用类型包的识别，比如不要针对alembic模块设置ignore_missing_imports = true
• 如果你同时使用SQLAlchemy，建议启用Mypy的SQLAlchemy插件，在配置中添加：

  [mypy]
  plugins = mypy.plugins.sqlalchemy
  

• 可以单独为alembic模块添加配置段，确保类型检查正常：

  [mypy-alembic.*]
  strict = false
  

3. 备选：调整导入方式（不推荐） #

如果暂时不想安装类型包，可以在迁移脚本中显式导入对应模块替代直接引用alembic.op或alembic.context：

from alembic.operations import Operations as op
from alembic.runtime.environment import EnvironmentContext as context

不过这种方式会改变原有代码结构，不如安装类型包来得方便。

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