import pyhs2
import getpass
import sys
import csv

if __name__ == "__main__":
	query = sys.argv[1]		#query without a semicolon at the end
	pwd   = getpass.getpass()
	data = []
	header = []
	with pyhs2.connect(host='hostname', port=10000, authMechanism="PLAIN", user='username', password=pwd) as conn:
		with conn.cursor() as cur:
			#Execute query
			cur.execute(str(query))

			columnInfo = cur.getSchema()
			for i in columnInfo:
				header.append(str(i.get("columnName").split(".")[1]))
			data.append(header)

			#Fetch table results
			for row in cur.fetch():
				data.append(row)

			with open('output.csv', 'wb') as f:
				wtr = csv.writer(f, delimiter=',')
				wtr.writerows(data)