Airflow执行命令报错：无法加载sqlalchemy.dialects:mysql插件

问题根源 #

报错sqlalchemy.exc.NoSuchModuleError: Can't load plugin: sqlalchemy.dialects:mysql，是因为Python环境缺少MySQL的SQLAlchemy适配驱动，导致SQLAlchemy无法解析MySQL连接协议。

解决步骤 #

方案1：安装mysqlclient（官方推荐） #

mysqlclient是MySQL官方维护的Python驱动，兼容性最优，执行以下命令：

# 先安装Ubuntu系统依赖
sudo apt-get install python3-dev default-libmysqlclient-dev build-essential
# 安装mysqlclient包
pip install mysqlclient

安装完成后，直接重新运行airflow db init或其他Airflow命令即可，无需修改现有配置。

方案2：使用pymysql驱动（备选） #

若安装mysqlclient遇到编译问题，可改用pymysql，需调整连接字符串：

1. 安装pymysql：

pip install pymysql

2. 修改airflow.cnf中的连接配置：
   • 将sql_alchemy_conn = mysql://airflow:password@localhost/airflowdb修改为：

     sql_alchemy_conn = mysql+pymysql://airflow:password@localhost/airflowdb
     

   • 将result_backend = db+mysql://airflow:password@localhost:3306/airflowdb修改为：

     result_backend = db+mysql+pymysql://airflow:password@localhost:3306/airflowdb
     

3. 保存配置后重新执行Airflow命令。

额外验证项 #

• 确认MySQL服务正常运行：sudo systemctl status mysql
• 确保airflow用户拥有airflowdb的全部权限，登录MySQL后执行：

GRANT ALL PRIVILEGES ON airflowdb.* TO 'airflow'@'localhost';
FLUSH PRIVILEGES;

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