From 4565bb4dd8b821bb1c9d7fbd1797f233375dda03 Mon Sep 17 00:00:00 2001 From: makefu Date: Tue, 19 Sep 2017 15:25:27 +0200 Subject: fix europastats --- europastats/attractions.py | 21 +++++++++++++++------ europastats/weather.py | 15 +++++++++++++-- 2 files changed, 28 insertions(+), 8 deletions(-) (limited to 'europastats') diff --git a/europastats/attractions.py b/europastats/attractions.py index 9a6c087..53bbfa4 100644 --- a/europastats/attractions.py +++ b/europastats/attractions.py @@ -1,7 +1,15 @@ #!/usr/bin/env python """ - takes a list of openweathermap ids and a config dict with - "openweathermap_apikey" + tag fields: + status (open closed) + type + name + metrics: + statuscode (0-6) + waitingtime + unused: + _ts (timestamp) + """ from datetime import datetime as dt import sys @@ -24,7 +32,7 @@ status_codes = [ api_url = "https://api.europapark.de/api-5.4/" def get_db(id): - with open(abspath(join(dirname(__name__),"europastats/data",id+".json")),'rb') as f: + with open(abspath(join(dirname(__file__),"data",id+".json")),'rb') as f: return json.load(f) name_mapping = { a["code"]: a for a in get_db("attractions") } @@ -59,12 +67,13 @@ def get_data(mock=False): resp = get_live("waitingtimes") for v in resp: data = {} + data["status"] = status_codes[v["status"]] - data["_name"] = "Attraktion {}".format( name_mapping[v["code"]]["nameGerman"]) - data["_id"] = "europapark-{}-{}".format(name,v["code"]) + data["statuscode"] = v["status"] + data["name"] = "Attraktion {}".format( name_mapping[v["code"]]["nameGerman"]) + data["type"] = name data["waitingtime"] = v["time"] data["_ts"] = now.isoformat(timespec="seconds") + "Z" - data["_db"] = name yield data def main(): diff --git a/europastats/weather.py b/europastats/weather.py index db0879c..3395e4e 100644 --- a/europastats/weather.py +++ b/europastats/weather.py @@ -1,5 +1,16 @@ #!/usr/bin/env python """ + tags fields: + name + type + offset + metrics: + perspiration + maxTemp + ... + unused: + _ts + """ from datetime import datetime as dt from europastats.attractions import get_db, get_live @@ -23,10 +34,10 @@ def get_data(mock=False): offset = 0 for v in resp: data = { - "_id": "europapark-weather-{}".format(offset), "_name": "Europapark Wetter {}".format(off[offset]), + "offset": offset, "_ts": now.isoformat(timespec="seconds") + "Z", - "_db": "weather" + "type": "weather" } data.update(v) yield data -- cgit v1.2.3