ExternalVendorCode/sondehub-infra
1
0
Fork 0
mirror of https://github.com/projecthorus/sondehub-infra.git synced 2025-01-30 15:43:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add launch sites function

Browse Source
This commit is contained in:
Michaela 2021-10-09 11:24:55 +11:00
parent 05d96cfb48
commit 1e39bb0b1e
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
predict_updater/lambda_function.py
View File

@ -340,6 +340,17 @@ def get_reverse_predictions():
logging.debug("Finished ES Request") logging.debug("Finished ES Request")
return { x['_source']['serial'] : x['_source'] for x in results['hits']['hits']} return { x['_source']['serial'] : x['_source'] for x in results['hits']['hits']}
def get_launch_sites():
path = "sites/_search"
payload = {
"size": 10000
}
logging.debug("Start ES Request")
results = es_request(json.dumps(payload), path, "POST")
logging.debug("Finished ES Request")
return {x['_source']['station']: x['_source'] for x in results['hits']['hits']}
def bulk_upload_es(index_prefix,payloads): def bulk_upload_es(index_prefix,payloads):
body="" body=""
for payload in payloads: for payload in payloads:
@ -641,10 +652,10 @@ if __name__ == "__main__":
# _descent = get_standard_prediction(conn, _now, -34.0, 138.0, 24000.0, burst_altitude=24000.5) # _descent = get_standard_prediction(conn, _now, -34.0, 138.0, 24000.0, burst_altitude=24000.5)
# print(f"Got {len(_descent)} data points for descent prediction.") # print(f"Got {len(_descent)} data points for descent prediction.")
test = predict( # test = predict(
{},{} # {},{}
) # )
print(get_launch_sites())
# for _serial in test: # for _serial in test:
# print(f"{_serial['serial']}: {len(_serial['data'])}") # print(f"{_serial['serial']}: {len(_serial['data'])}")