Initial switch label scripts 🚂

1 files changed, 66 insertions, 0 deletions

diff --git a/gondul.py b/gondul.py
new file mode 100644
index 0000000..65c18fa
--- /dev/null
+++ b/gondul.py
@@ -0,0 +1,66 @@
+import base64
+import json
+import os
+import re
+import urllib.parse
+import urllib.request
+
+GONDUL_USERNAME = os.getenv("GONDUL_USERNAME", "")
+GONDUL_PASSWORD = os.getenv("GONDUL_PASSWORD", "")
+GONDUL_API = os.getenv("GONDUL_API", "https://tg18.gondul.gathering.org/api")
+GONDUL_SWITCHES_ENDPOINT = os.getenv(
+    "GONDUL_SWITCHES_ENDPOINT", "/public/switches")
+
+
+def _generate_credentials(username, password):
+    return base64.standard_b64encode(
+        (username + ":" + password)
+        .encode("utf-8")).decode("utf-8")
+
+
+def _do_switches_request(
+        api=GONDUL_API,
+        endpoint=GONDUL_SWITCHES_ENDPOINT,
+        credentials=_generate_credentials(GONDUL_USERNAME, GONDUL_PASSWORD)):
+    switches_url = api + endpoint
+
+    # Build request
+    request = urllib.request.Request(switches_url)
+    request.add_header("Authorization", "Basic " + credentials)
+    resp = urllib.request.urlopen(request, timeout=5)
+    assert resp.status == 200, "HTTP return was not 200 OK"
+
+    # Read response
+    body = resp.read().decode("utf-8")
+    data = json.loads(body)
+    assert "switches" in data, "Missing switches object from HTTP response"
+
+    switches = data.get("switches")
+    print("Found {} switches in Gondul".format(len(switches)))
+    return switches
+
+
+def _match_switches(switches, match="^e(.*)"):
+    pattern = re.compile(match)
+
+    included_switches = []
+    for switch in switches:
+        include = re.search(pattern, switch)
+        if include:
+            included_switches.append(switch)
+
+    print("'{}' matches {} switches.".format(match, len(included_switches)))
+    return included_switches
+
+
+def _sort_switches(switches):
+    # The lambda returns two values to compare on;
+    # * The switch number (e77-4) - picks out the number 77
+    # * The number of the switch in relation to other switches on the same row
+    # E.g. "e77-4" will return 4
+    return sorted(switches, key=lambda x: (int(x[1:].split("-")[0]), x.split("-")[1]))
+
+
+def fetch_gondul_switches(match="^e(.*)"):
+    # credentials = _generate_credentials()
+    return _sort_switches(_match_switches(_do_switches_request()))