import pyhs2
import time
import csv

csv_out = open('data.csv', 'wb')
mywriter = csv.writer(csv_out)

def generic_user():
    gu = 'xxxxxx'
    return gu

def password():
    pw = 'xxxxx'
    return pw

with pyhs2.connect(host='xxxxxxxx',
               port=10000,
               authMechanism='PLAIN',
               user=generic_user(),
               password=password(),
               database='xxxxxxx') as conn:

    with conn.cursor() as cur:

q = raw_input('Enter query: ').replace('csc', 'CSC')
    print q

    #timer start
    start_time = time.time()

    #Execute query
    cur.execute(q)

col = []

for key in cur.getSchema():
        col.append(key['columnName'])

header = []
    header.append(col)

for rows in zip(header):
        mywriter.writerows(rows)

records = cur.fetch()
    for rows in zip(records):
        mywriter.writerows(rows)