presto connection with python

presto_conn_example.py

from pyhive import presto
from pandas import read_sql


def get_presto_conn():
    return presto.connect(host='YOUR_PRESTO_HOST', username='YOUR_PRESTO_USERNAME', port='YOUR_PRESTO_PORT')
    
 
def get_query_result(presto_conn):
    query = 'YOUR_QUERY_HERE'

    query_result = read_sql(query, presto_conn)    
    
    ## your dataframes manipulations here
    
    return query_result
    
    
def run():
    presto_conn = get_presto_conn()
    query_result = get_query_result(presto_conn)

    

if __name__ == '__main__':
    run()