diff options
| -rwxr-xr-x | contrib/i3bar-snippet.py | 110 |
1 files changed, 110 insertions, 0 deletions
diff --git a/contrib/i3bar-snippet.py b/contrib/i3bar-snippet.py new file mode 100755 index 0000000..5d7b9e7 --- /dev/null +++ b/contrib/i3bar-snippet.py @@ -0,0 +1,110 @@ +#!/usr/bin/python3 + +# This script queries the Travelynx API if you are checked into a train. If +# yes, marudor.de is additionally queried for the next stop, and a JSON object +# like this is written to stdout: +# {"full_text": "RE26824, next: D\u00fcren at <span fgcolor=\"#ff0000\">15:38+5</span>, dest: Aachen Hbf at <span fgcolor=\"#ff0000\">16:07+5</span>", "markup": "pango"}, +# The script then exits. +# +# Configuration: +# - Place your API key from https://travelynx.de/account at +# ~/.config/travelynx.conf . +# - Then integrate into whatever generates your i3bar input. +# - Make sure you use i3bar with a pango font, so that the color tags are +# picked up. + + +from datetime import datetime +import dateutil +import dateutil.parser +import json +from pathlib import Path +import requests +import sys +import xdg # not pyxdg! + + +def format_stop(stop_name, scheduled_arrival_timestamp, delay): + color = "#ffffff" + if delay > 0: + if delay <= 2: + color = "#ffff00" + else: + color = "#ff0000" + delayStr = "{:+.0f}".format(delay) + else: + delayStr = "" + if isinstance(scheduled_arrival_timestamp, int): + scheduled_arrival_time = datetime.fromtimestamp(scheduled_arrival_timestamp) + else: + # We assume it's datetime already. + scheduled_arrival_time = scheduled_arrival_timestamp + return f'{stop_name} at <span fgcolor="{color}">{scheduled_arrival_time:%H:%M}{delayStr}</span>' + + +api_key_path = Path(xdg.xdg_config_home(), "travelynx.conf") +if api_key_path.exists(): + with api_key_path.open("r") as f: + api_key = f.read().strip() +else: + print( + f"Could not find Travelyxn API key at {api_key_path}.", + file=sys.stderr, + ) + sys.exit(1) + + +api_base = f"https://travelynx.de/api/v1/status/{api_key}" +try: + res = requests.get(api_base) +except requests.exceptions.ConnectionError: + print( + json.dumps({"full_text": "Could not connect to travelynx", "color": "#ff0000"}) + + "," + ) + sys.exit() + +j = res.json() +# print(json.dumps(j, sort_keys=True, indent=4), file=sys.stderr) + +if not j["checkedIn"]: + sys.exit() + +out_fields = [] + +train = "{}{}".format(j["train"]["type"], j["train"]["no"]) +out_fields.append(train) +destination_name = j["toStation"]["name"] +scheduled_arrival_timestamp = j["toStation"]["scheduledTime"] +delay = (j["toStation"]["realTime"] - j["toStation"]["scheduledTime"]) / 60 + +try: + details_res = requests.get(f"https://marudor.de/api/hafas/v2/details/{train}") + details = details_res.json() + # print(json.dumps(details, sort_keys=True, indent=4), file=sys.stderr) + next_stop_name = details["currentStop"]["station"]["title"] + if next_stop_name == destination_name: + out_fields.append("next") + else: + next_scheduled_arrival_time = dateutil.parser.isoparse( + details["currentStop"]["arrival"]["scheduledTime"] + ) + next_scheduled_arrival_time = next_scheduled_arrival_time.astimezone( + dateutil.tz.tzlocal() + ) + next_delay = details["currentStop"]["arrival"]["delay"] + out_fields.append( + "next: " + + format_stop(next_stop_name, next_scheduled_arrival_time, next_delay) + ) +except requests.exceptions.ConnectionError: + pass + +out_fields.append( + "dest: " + format_stop(destination_name, scheduled_arrival_timestamp, delay) +) + +s = ", ".join(out_fields) + +out_obj = {"full_text": s, "markup": "pango"} +print(json.dumps(out_obj) + ",") |