import access as a
import pprint

pp = pprint.PrettyPrinter(indent=4)

athena = a.Athena("hay", "s3://haystac-pmo-athena/")
query  = athena.query('select * from "trajectories"."kitware" limit 10')
data   = query.results()
print(data.head(10))

# Everything *needs* to have a column in order for parquet to work, so scalar
# values have to be assigned something, so here we use `as count` to create
# a temporary column called `count`
print(athena.query("select count(*) as count from trajectories.kitware").results().head(1))