python 操作hive

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()