From 9a58766525f29845eaa930819e9c096b9991d6cf Mon Sep 17 00:00:00 2001
From: JJ <nicetry@noemail.com>
Date: Wed, 19 Mar 2025 22:22:10 +0000
Subject: first commit
---
src/tube_stat/cli.py | 77 ++++++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 77 insertions(+)
create mode 100644 src/tube_stat/cli.py
(limited to 'src/tube_stat/cli.py')
diff --git a/src/tube_stat/cli.py b/src/tube_stat/cli.py
new file mode 100644
index 0000000..c408804
--- /dev/null
+++ b/src/tube_stat/cli.py
@@ -0,0 +1,77 @@
+import click
+import json
+import requests
+
+with open("./config.json", "r") as e:
+ config = json.load(e)
+
+TUBE_LINES = (
+ "bakerloo", "victoria", "central", "circle", "district", "hammersmith-city",
+ "jubilee", "metropolitan", "northern", "piccadilly", "waterloo-city", "dlr",
+ "overground", "tram"
+)
+APP = config["APP"]
+KEY = config["KEY"]
+
+def get_line_color(line: str) -> tuple[int, int, int]:
+ match line:
+ case "bakerloo":
+ return (179, 99, 5)
+ case "victoria":
+ return (0, 152, 212)
+ case "central":
+ return (227, 32, 23)
+ case "circle":
+ return (255, 211, 0)
+ case "district":
+ return (0, 120, 42)
+ case "hammersmith-city":
+ return (243, 169, 187)
+ case "jubilee":
+ return (160, 165, 169)
+ case "metropolitan":
+ return (155, 0, 86)
+ case "northern":
+ return (0, 0, 0)
+ case "piccadilly":
+ return (0, 54, 136)
+ case "waterloo-city":
+ return (149, 205, 186)
+ case "dlr":
+ return (0, 164, 167)
+ case "overground":
+ return (238, 124, 14)
+ case "tram":
+ return (132, 184, 23)
+ case _:
+ return (128, 128, 128)
+
+@click.command()
+@click.option("--line", default=None, help="comma separated tube lines, e.g bakerloo,central")
+def cli(line):
+ if line:
+ requested_lines = line.split(",")
+ invalid_lines = [l for l in requested_lines if l not in TUBE_LINES]
+ if invalid_lines:
+ raise click.ClickException(f'Invalid tube lines: {', '.join(invalid_lines)}')
+ else:
+ requested_lines = ','.join(TUBE_LINES)
+ req_url = f'https://api.tfl.gov.uk/Line/{requested_lines}/Status?app_id={APP}&app_key={KEY}'
+ res = requests.get(req_url)
+ res_json = json.loads(res.text)
+ print_str = ""
+ for i, r in enumerate(res_json):
+ if len(res_json) - 1 != i:
+ line_break = "\n"
+ else:
+ line_break = ""
+ line_colour = get_line_color(r["id"])
+ print_str += f'{click.style(r["name"], bg=line_colour)}: {r["lineStatuses"][0]["statusSeverityDescription"]}{line_break}'
+ click.echo(print_str)
+
+if __name__ == "__main__":
+ try:
+ cli()
+ except click.ClickException as e:
+ click.echo(str(e), err=True)
+ raise click.exceptions.Exit(1)
--
cgit v1.2.3