diff --git a/airflow/providers/apache/hdfs/hooks/webhdfs.py b/airflow/providers/apache/hdfs/hooks/webhdfs.py index e99e20127e7ca..fdfdfc0c5a49e 100644 --- a/airflow/providers/apache/hdfs/hooks/webhdfs.py +++ b/airflow/providers/apache/hdfs/hooks/webhdfs.py @@ -52,7 +52,12 @@ class WebHDFSHook(BaseHook): :param proxy_user: The user used to authenticate. """ - def __init__(self, webhdfs_conn_id: str = "webhdfs_default", proxy_user: str | None = None): + conn_type = "webhdfs" + conn_name_attr = "webhdfs_conn_id" + default_conn_name = "webhdfs_default" + hook_name = "Apache WebHDFS" + + def __init__(self, webhdfs_conn_id: str = default_conn_name, proxy_user: str | None = None): super().__init__() self.webhdfs_conn_id = webhdfs_conn_id self.proxy_user = proxy_user diff --git a/airflow/providers/apache/hdfs/provider.yaml b/airflow/providers/apache/hdfs/provider.yaml index 8984b1652b35c..f796b23873d61 100644 --- a/airflow/providers/apache/hdfs/provider.yaml +++ b/airflow/providers/apache/hdfs/provider.yaml @@ -70,3 +70,7 @@ hooks: - integration-name: WebHDFS python-modules: - airflow.providers.apache.hdfs.hooks.webhdfs + +connection-types: + - hook-class-name: airflow.providers.apache.hdfs.hooks.webhdfs.WebHDFSHook + connection-type: webhdfs