54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
import time
|
|
|
|
import requests
|
|
|
|
from qm_try_analysis.logging import info
|
|
|
|
TELEMETRY_BASE_URL = "https://sql.telemetry.mozilla.org/api/"
|
|
|
|
|
|
def query(key, query, p_params):
|
|
headers = {"Authorization": f"Key {key}"}
|
|
start_url = TELEMETRY_BASE_URL + f"queries/{query}/refresh?{p_params}"
|
|
info(f"Starting job using url {start_url}")
|
|
resp = requests.post(url=start_url, headers=headers)
|
|
job = resp.json()["job"]
|
|
job_id = job["id"]
|
|
info(f"Started job {job_id}")
|
|
|
|
poll_url = TELEMETRY_BASE_URL + f"jobs/{job_id}"
|
|
info(f"Polling query status from {poll_url}")
|
|
poll = True
|
|
status = 0
|
|
qresultid = 0
|
|
while poll:
|
|
print(".", end="", flush=True)
|
|
resp = requests.get(url=poll_url, headers=headers)
|
|
status = resp.json()["job"]["status"]
|
|
if status > 2:
|
|
# print(resp.json())
|
|
poll = False
|
|
qresultid = resp.json()["job"]["query_result_id"]
|
|
else:
|
|
time.sleep(0.2)
|
|
print(".")
|
|
info(f"Finished with status {status}")
|
|
|
|
if status == 3:
|
|
results_url = TELEMETRY_BASE_URL + f"queries/78691/results/{qresultid}.json"
|
|
|
|
info(f"Querying result from {results_url}")
|
|
resp = requests.get(url=results_url, headers=headers)
|
|
return resp.json()
|
|
|
|
return {"query_result": {"data": {"rows": {}}}}
|
|
|
|
|
|
def getLastEventTimeAbs(rows):
|
|
if len(rows) == 0:
|
|
return 0
|
|
return rows[len(rows) - 1]["submit_timeabs"]
|