Function execute_sql

bmtools/tools/database/data/script.py:11–27

(sql)

prepare the test samples
'''
def execute_sql(sql):
    conn = psycopg2.connect(database='tpch10x',
                            user='xxx',
                            password='xxx',
                            host='xxx',
                            port=xxx)

    cur = conn.cursor()
    cur.execute(sql)
    # res = cur.fetchall()[0][0][0]
    res = cur.fetchall()

    conn.commit()
    cur.close()
    conn.close()

    return len(res)

# Load the JSON file as a dictionary
data = {}