from locustdb import Client
import numpy as np
import argparse
parser = argparse.ArgumentParser(description='Query locustDB')
parser.add_argument('column_name', type=str, help='column name to query')
args = parser.parse_args()
colname = args.column_name
client = Client("http://localhost:8080")
results = client.query(f'SELECT "{colname}" FROM "avid-wildflower-3446"')
scolname = colname.replace('[', '_').replace(']', '_')
with open(scolname + ".txt", "w") as f:
np.savetxt(f, [(r if r is not None else np.nan) for r in results[colname]])