ari/minerva
1
0
Fork 0
forked from bellwether/minerva
Code Pull requests Activity

properly handles no results being returned

Browse source
This commit is contained in:
Ari Brown 2023-07-27 15:27:48 -04:00
parent 8c498ba5dd
commit ac657f671d
2 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
test.py
View file

@ -4,12 +4,19 @@ import pprint
pp = pprint.PrettyPrinter(indent=4)
athena = m.Athena("hay", "s3://haystac-pmo-athena/")
query = athena.query('select * from "trajectories"."kitware" limit 10')
query = athena.query(
"""SELECT *
FROM trajectories.kitware
WHERE ST_Disjoint(
ST_GeometryFromText('POLYGON((103.6 1.2151693, 103.6 1.5151693, 104.14797 1.5151693, 104.14797 1.2151693, 103.6 1.2151693))'),
ST_Point(longitude, latitude)
)
""")
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))
#print(athena.query("select count(*) as count from trajectories.kitware").results().head(1))