From 9460238917ea6466dfec6883d9cba9d5fd5d80eb Mon Sep 17 00:00:00 2001 From: jesopo Date: Fri, 8 Nov 2019 17:58:04 +0000 Subject: [PATCH] move tfl.py to bitbot-modules --- modules/tfl.py | 105 ------------------------------------------------- 1 file changed, 105 deletions(-) delete mode 100644 modules/tfl.py diff --git a/modules/tfl.py b/modules/tfl.py deleted file mode 100644 index 62fc57df..00000000 --- a/modules/tfl.py +++ /dev/null @@ -1,105 +0,0 @@ -#--depends-on commands - -import collections, 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", "") - 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)