import minerva
import pprint

pp = pprint.PrettyPrinter(indent=4)

# Create the Minerva object which gives you access to the account under the
# profile `hay`
m = minerva.Minerva("hay")

# Get the Athena object
athena = m.athena("s3://haystac-pmo-athena/")

# Everything *needs* to have a column in order for unloading to parquet to work,
# so scalar values have to be assigned something, so here we use `as count` to
# create a temporary column called `count`
#query  = athena.query(
#    """
#    select round(longitude, 3) as lon, count(*) as count
#    from trajectories.basline
#    where agent = 4
#    group by round(longitude, 3)
#    order by count(*) desc
#    """
#)

query = athena.query("select * from trajectories.basline where agent < 100 limit 100")
data   = query.results()

pp.pprint(data.head(10))

# We also get important statistics
print(query.runtime)
print(query.cost)