【Python】pyhive で Presto 接続

f:id:tm200:20210704143739p:plain


PyHiveでPresto接続方法のメモ。
認証方法はプロダクト毎に違うと思いますので、適宜変更。
この辺りを見れば実装方法が分かると思います。

github.com

インストール

$ pip3 install pyhive[presto]==0.6.4

実装

# -*- coding: utf-8 -*-
import base64
from pyhive import presto

def execute():
    cursor = presto.connect(
        '<ホスト名>',
        port=443,
        catalog='<カタログ名>',
        username='<ユーザー名>',
        protocol='https',
    ).cursor()

    # パラメータは辞書・リスト・tupleのいずれか
    cursor.execute('SELECT * FROM sample WHERE test_column = %s', parameters=('param_value',))
    print(cursor.fetchall())


if __name__ == '__main__':
    execute()

参考

逐次処理等がしたい場合はこちらを参考にすると良いかも。
機会があれば実装しようと思います。

github.com

www.python.org

オススメ書籍

色々と考えさせられる一冊です!
ページ数も少なく、時間がない人でも読める量だと思います(`・∀・´)