Python连接Hive相关配置_MY测试之道的博客-CSDN博客

"""
hive数据库相关操作
"""

from pyhive import hive

class HiveUtils():

    def __init__(self, host, port=10000, username=None, password=None, database=None, auth=None):
        try:
            self.conn = hive.Connection(host=host,port=port, username=username, password=password, database=database, auth=auth)
            self.cur = self.conn.cursor()
            print("初始化连接hive数据库成功！")
        except Exception as e:
            raise f"初始化连接hive失败，失败原因: {e}"

    def execute_sql(self, sql: str):
        """执行SQL语句"""
        if not isinstance(sql, str):
            raise ValueError("参数：sql 数据类型错误！应是str类型")
        try:
            results = None
            self.cur.execute(sql)
            if "show" in sql or "select" in sql:
                results = self.cur.fetchall()
            self.conn.commit()
            print("SQL语句执行成功！")
            return results
        except Exception as e:
            raise f"SQL语句执行失败！，失败原因: {e}"

    def close(self):
        """关闭游标和hive连接"""
        self.cur.close()
        self.conn.close()