Apache Airflow调用InfluxDB遇ModuleNotFoundError错误求助

以下是针对该问题的排查和解决步骤：

1. 确认版本兼容性 #

Python 3.11需要搭配Airflow 2.6.0及以上版本，同时InfluxDB Provider版本需与Airflow主版本适配。执行以下命令安装匹配的依赖：

# 确保Airflow版本符合Python3.11要求
pip install "apache-airflow>=2.6.0" --constraint "https://raw.githubusercontent.com/apache/airflow/constraints-2.6.3/constraints-3.11.txt"
# 安装适配的InfluxDB Provider
pip install apache-airflow-providers-influxdb==3.2.0

注：constraints文件需对应你的Airflow主版本，避免依赖冲突。

2. 验证安装环境一致性 #

• 执行which python（WSL）或where python（Windows），确认当前使用的Python解释器路径。
• 执行pip list | grep apache-airflow-providers-influxdb，检查该包是否出现在输出列表中。
• 若使用虚拟环境，确保Airflow的webserver、scheduler服务是在该虚拟环境中启动的。

3. 重启Airflow服务 #

安装Provider后必须重启Airflow服务，确保新的Provider被加载：

# WSL环境下的重启命令（根据实际启动方式调整）
airflow webserver -D
airflow scheduler -D

4. 排查特殊场景问题 #

• 若使用Docker部署Airflow，需在镜像中添加Provider安装步骤，或基于官方镜像自定义构建。
• 检查是否存在多个Python环境，导致安装的包不在Airflow实际使用的环境中（比如系统Python与虚拟环境Python混淆）。

额外代码优化提示 #

你的代码还有几处需要完善：

• urls、token、organization变量未赋值，需补充实际的InfluxDB连接信息。
• Flux查询中的|> filter(fn: (r) => r["osmajor"] == )缺少具体值，需补充为类似"18"这样的条件。
• InfluxDBHook初始化时需指定连接ID（比如InfluxDBHook(influxdb_conn_id="your_influx_conn")），无参初始化无法正确获取连接。

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