Skip to content

Replace sasl with pyhive.get_installed_sasl for pure-sasl compatibility #55772

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
romsharon98 merged 4 commits into from
Sep 30, 2025
Merged

Replace sasl with pyhive.get_installed_sasl for pure-sasl compatibility #55772

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
1814038
Replace sasl with pyhive.get_installed_sasl for pure-sasl compatibility
kesem0811 Sep 17, 2025
7436c0a
fix ruff formating
kesem0811 Sep 25, 2025
938708c
cleanup: remove test DAG files to match UI version
kesem0811 Sep 25, 2025
5f1f682
Apply ruff formatting to hive.py
kesem0811 Sep 28, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 17 additions & 20 deletions providers/apache/hive/src/airflow/providers/apache/hive/hooks/hive.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
from __future__ import annotations

import contextlib
import csv
import os
import re
import socket
Expand All @@ -30,19 +31,21 @@
from deprecated import deprecated
from typing_extensions import overload

if TYPE_CHECKING:
import pandas as pd
import polars as pl

import csv

from airflow.configuration import conf
from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
from airflow.providers.apache.hive.version_compat import AIRFLOW_VAR_NAME_FORMAT_MAPPING, BaseHook
from airflow.providers.apache.hive.version_compat import (
AIRFLOW_VAR_NAME_FORMAT_MAPPING,
BaseHook,
)
from airflow.providers.common.sql.hooks.sql import DbApiHook
from airflow.security import utils
from airflow.utils.helpers import as_flattened_list

if TYPE_CHECKING:
import pandas as pd
import polars as pl


HIVE_QUEUE_PRIORITIES = ["VERY_HIGH", "HIGH", "NORMAL", "LOW", "VERY_LOW"]


Expand Down Expand Up @@ -573,21 +576,15 @@ def get_metastore_client(self) -> Any:
conn_socket = TSocket.TSocket(host, conn.port)

if conf.get("core", "security") == "kerberos" and auth_mechanism == "GSSAPI":
try:
import saslwrapper as sasl
except ImportError:
import sasl

def sasl_factory() -> sasl.Client:
sasl_client = sasl.Client()
sasl_client.setAttr("host", host)
sasl_client.setAttr("service", kerberos_service_name)
sasl_client.init()
return sasl_client

from pyhive.hive import get_installed_sasl
from thrift_sasl import TSaslClientTransport

transport = TSaslClientTransport(sasl_factory, "GSSAPI", conn_socket)
sasl_auth = "GSSAPI"
transport = TSaslClientTransport(
lambda: get_installed_sasl(host=host, sasl_auth=sasl_auth, service=kerberos_service_name),
sasl_auth,
conn_socket,
)
else:
transport = TTransport.TBufferedTransport(conn_socket)

Expand Down
Loading