import pyhs2
import sys
default_encoding = 'utf-8'
if sys.getdefaultencoding() != default_encoding:
reload(sys)
sys.setdefaultencoding(default_encoding)
import pyhs2
import sys
default_encoding = 'utf-8'
if sys.getdefaultencoding() != default_encoding:
reload(sys)
sys.setdefaultencoding(default_encoding)
class HiveClient:
def __init__(self, db_host, user, password, database, port=10000, authMechanism="PLAIN"):
"""
create connection to hive server2
"""
self.conn = pyhs2.connect(host=db_host,
port=port,
authMechanism=authMechanism,
user=user,
password=password,
database=database,
)
def query(self, sql):
"""
query
"""
with self.conn.cursor() as cursor:
cursor.execute(sql)
return cursor.fetch()
def close(self):
"""
close connection
"""
self.conn.close()
def main():
"""
main process
"""
hive_client = HiveClient(db_host='*.*.*.*', user='hive', password='hive', database='default',port=10000 , authMechanism='PLAIN')
print 'test'
sql = 'select jgbm , jgmc from test limit 2'
result = hive_client.query(sql)
print (result)
hive_client.close()
if __name__ == '__main__':
main()
python 操作hive
最新推荐文章于 2024-09-03 13:54:52 发布
该博客介绍了如何使用Python的pyhs2库连接到Hive服务器并执行查询。通过创建`HiveClient`类,设置数据库主机、用户名、密码、端口和认证机制来建立连接。类中包含query方法执行SQL查询并获取结果,以及close方法关闭连接。示例展示了查询默认数据库'default'中的'test'表。
摘要由CSDN通过智能技术生成