105 lines
3.9 KiB
Python
105 lines
3.9 KiB
Python
#--depends-on commands
|
|
|
|
import collections, datetime, re, typing, urllib.parse
|
|
from src import ModuleManager, utils
|
|
|
|
URL_LINE = "https://api.tfl.gov.uk/Line/Mode/tube/Status"
|
|
|
|
URL_STOP_SEARCH = "https://api.tfl.gov.uk/StopPoint/Search/%s"
|
|
URL_STOP_ARRIVALS = "https://api.tfl.gov.uk/StopPoint/%s/Arrivals"
|
|
|
|
LINES = {
|
|
"waterloo and city": "waterloo-city",
|
|
"waterloo & city": "waterloo-city",
|
|
"hammersmith and city": "hammersmith-city",
|
|
"hammersmith & city": "hammersmith-city"
|
|
}
|
|
|
|
GOOD_COLOR = utils.irc.color("Good service", utils.consts.GREEN)
|
|
BAD_COLOR = utils.irc.color("Issues", utils.consts.ORANGE)
|
|
|
|
class Module(ModuleManager.BaseModule):
|
|
_name = "TFL"
|
|
|
|
|
|
@utils.hook("received.command.tubeline")
|
|
@utils.kwarg("help", "Show status of Tube lines")
|
|
@utils.kwarg("usage", "[line]")
|
|
def line(self, event):
|
|
lines = utils.http.request(URL_LINE, json=True)
|
|
|
|
if event["args_split"]:
|
|
line_query = event["args"].strip().lower()
|
|
line_query = LINES.get(line_query, line_query)
|
|
|
|
found = None
|
|
for line in lines.data:
|
|
if line["id"] == line_query:
|
|
found = line
|
|
break
|
|
if found:
|
|
status = found["lineStatuses"][0]
|
|
reason = None
|
|
if "reason" in status:
|
|
reason = " (%s)" % status["reason"].strip()
|
|
|
|
event["stdout"].write("%s status: %s%s" % (
|
|
found["name"], status["statusSeverityDescription"], reason))
|
|
else:
|
|
event["stderr"].write("Unknown line '%s'" % line_query)
|
|
else:
|
|
good = []
|
|
bad = []
|
|
for line in lines.data:
|
|
status = line["lineStatuses"][0]
|
|
if status["statusSeverity"] == 10:
|
|
good.append(line["name"])
|
|
else:
|
|
bad.append(line["name"])
|
|
|
|
good_str = ", ".join(good)
|
|
bad_str = ", ".join(bad)
|
|
if good and bad:
|
|
event["stdout"].write("%s: %s | %s: %s" %
|
|
(GOOD_COLOR, good_str, BAD_COLOR, bad_str))
|
|
elif good:
|
|
event["stdout"].write("%s on all lines" % GOOD_COLOR)
|
|
else:
|
|
event["stdout"].write("%s reported on all lines" % BAD_COLOR)
|
|
|
|
@utils.hook("received.command.tubestop")
|
|
@utils.kwarg("help", "Show arrivals for a given Tube station")
|
|
@utils.kwarg("usage", "<station>")
|
|
def stop(self, event):
|
|
query = event["args"].strip()
|
|
station = utils.http.request(
|
|
URL_STOP_SEARCH % urllib.parse.quote(query), json=True)
|
|
|
|
if station.data["matches"]:
|
|
station = station.data["matches"][0]
|
|
arrivals = utils.http.request(URL_STOP_ARRIVALS % station["id"],
|
|
json=True)
|
|
destinations = collections.OrderedDict()
|
|
now = utils.datetime_utcnow().replace(second=0, microsecond=0)
|
|
print(now.isoformat())
|
|
|
|
arrivals = sorted(arrivals.data, key=lambda a: a["expectedArrival"])
|
|
for train in arrivals:
|
|
destination = train["destinationNaptanId"]
|
|
if not destination in destinations:
|
|
arrival = utils.iso8601_parse(train["expectedArrival"])
|
|
if now >= arrival:
|
|
arrival = "Due"
|
|
else:
|
|
arrival = "In %s" % utils.to_pretty_time(
|
|
(arrival-now).total_seconds(), max_units=1,
|
|
minimum_unit=utils.UNIT_MINUTE)
|
|
|
|
destinations[destination] = "%s (%s)" % (
|
|
train["towards"], arrival)
|
|
|
|
if destinations:
|
|
event["stdout"].write("%s: %s" % (
|
|
station["name"], ", ".join(destinations.values())))
|
|
else:
|
|
event["stdout"].write("Unknown station '%s'" % query)
|