move tfl.py to bitbot-modules

modules/tfl.py

@@ -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", "<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)