Airflow 2.7.1升级后SQLAlchemy连接Oracle报错及测试问题求助

一、修复Oracle连接报错（TypeError: 'extra' is an invalid keyword argument） #

问题原因 #

Airflow 2.7.x版本中，OracleHook.get_uri()返回的连接字符串会包含额外的extra参数，而SQLAlchemy的Oracle驱动无法识别该参数，导致创建连接时抛出错误。

解决方案 #

方案1：使用OracleHook内置的SQLAlchemy引擎创建方法（推荐）

直接调用OracleHook.get_sqlalchemy_engine()获取已正确配置的引擎，避免手动处理URI的问题：

# 替换原有的conexao_orcl创建代码
oracle_hook = OracleHook(oracle_conn_id=ora_conn_id)
conexao_orcl = oracle_hook.get_sqlalchemy_engine()

方案2：手动清理URI中的extra参数

如果必须手动调用create_engine，可以解析并移除URI中的extra部分：

from urllib.parse import urlparse, urlunparse

oracle_hook = OracleHook(oracle_conn_id=ora_conn_id)
uri = oracle_hook.get_uri()
# 解析URI，移除query部分的extra参数
parsed_uri = urlparse(uri)
cleaned_uri = urlunparse(parsed_uri._replace(query=''))
conexao_orcl = create_engine(cleaned_uri)

二、启用Airflow连接测试功能 #

问题原因 #

Airflow默认禁用了连接测试功能，需要修改配置开启。

解决方案 #

方法1：修改airflow.cfg配置文件

找到Airflow配置文件（通常位于$AIRFLOW_HOME/airflow.cfg），在[core]部分添加或修改：

[core]
enable_connection_testing = True

修改完成后重启Airflow Webserver和Scheduler服务：

airflow webserver restart
airflow scheduler restart

方法2：通过环境变量临时开启

在启动Airflow服务前设置环境变量：

export AIRFLOW__CORE__ENABLE_CONNECTION_TESTING=True
# 重启服务
airflow webserver restart
airflow scheduler restart

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