mo
15 lines
559 B
Python
15 lines
559 B
Python
import sys, json, urllib.request
|
|
def q(sql):
|
|
req=urllib.request.Request("http://10.0.21.50:8089/v1/statement", data=sql.encode(), headers={"X-Trino-User":"admin"})
|
|
d=json.load(urllib.request.urlopen(req))
|
|
rows=d.get("data") or []; nu=d.get("nextUri"); err=None
|
|
while nu:
|
|
j=json.load(urllib.request.urlopen(nu)); rows+=j.get("data") or [];
|
|
if j.get("error"): err=j["error"]
|
|
nu=j.get("nextUri")
|
|
return rows, err
|
|
sql=sys.argv[1]
|
|
rows,err=q(sql)
|
|
if err: print("ERROR:", err.get("message"))
|
|
else: print("OK", rows[:5])
|