Airflow：如何将PyMySQL与MySQL Hook一起使用？

Question

我试图在Apache Airflow中使用MySQL。我已经使用PymySQL来查询MySQL数据库，并且没有安装MySQLDb。因此我得到了错误：

 from airflow.hooks.mysql_hook import MySqlHook
  File "/anaconda3/anaconda/lib/python3.6/site-packages/airflow/hooks/mysql_hook.py", line 15, in <module>
    import MySQLdb
ModuleNotFoundError: No module named 'MySQLdb'

如果我进入mysql_hook.py课，我已经看到他们正在使用MySQLDB

import MySQLdb
import MySQLdb.cursors

除了手动导入pymysql而不是钩子，我有什么替代方案？

我使用的是Python 3.6

谢谢

Answer 1

为了使MySQL钩子能够在Airflow中工作，您需要安装MySQL额外的包。如果您已经使用pip安装了Airflow，那么以下命令将执行：

pip install apache-airflow[mysql]

这将安装mysqlclient包，这是此处所需的包。