summaryrefslogtreecommitdiff
path: root/contrib/i3bar-snippet.py
diff options
context:
space:
mode:
authorclonejo <clonejo@shakik.de>2021-08-17 15:39:24 +0200
committerDaniel Friesel <derf@chaosdorf.de>2021-12-01 18:29:36 +0100
commit3829b26fea9947f38cd739d3edce79f193a17464 (patch)
treeca3f0ca0a4c2794ba856db733b03eb1ce61820a4 /contrib/i3bar-snippet.py
parent0117a042664c37f8c4d0cdf79dfc36b7b77132c7 (diff)
contrib: JSON snippet generator for i3bar
Diffstat (limited to 'contrib/i3bar-snippet.py')
-rwxr-xr-xcontrib/i3bar-snippet.py110
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) + ",")