diff --git a/bot.json.example b/bot.json.example index 5ac30a5b..79b49c50 100644 --- a/bot.json.example +++ b/bot.json.example @@ -12,5 +12,7 @@ "twitter-access-secret" : "", "trakt-api-key" : "", "bitly-api-key" : "", - "soundcloud-api-key" : "" + "soundcloud-api-key" : "", + "tfl-app-id" : "", + "tfl-app-key" : "" } diff --git a/modules/tfl.py b/modules/tfl.py new file mode 100644 index 00000000..59c8bc71 --- /dev/null +++ b/modules/tfl.py @@ -0,0 +1,45 @@ +import datetime +import Utils + +URL_BUS = "https://api.tfl.gov.uk/StopPoint/%s/Arrivals" +URL_BUS_SEARCH = "https://api.tfl.gov.uk/StopPoint/Search/%s" + +class Module(object): + def __init__(self, bot): + self.bot = bot + bot.events.on("received").on("command").on("tflbus" + ).hook(self.bus, min_args=1) + + def bus(self, event): + app_id = self.bot.config["tfl-api-id"] + app_key = self.bot.config["tfl-api-key"] + stop_id = event["args_split"][0] + if stop_id.isdigit(): + bus_search = Utils.get_url(URL_BUS_SEARCH % stop_id, get_params={ + "app_id": app_id, "app_key": app_key}, json=True) + print(bus_search) + if bus_search["matches"]: + bus_stop = bus_search["matches"][0] + real_stop_id = bus_stop["id"] + bus_stop = Utils.get_url(URL_BUS % real_stop_id, get_params={ + "app_id": app_id, "app_key": app_key}, json=True) + busses = [] + for bus in bus_stop: + bus_number = bus["lineName"] + bus_due_iso8601 = bus["expectedArrival"] + bus_due = datetime.datetime.strptime(bus_due_iso8601, + "%Y-%m-%dT%H:%M:%SZ") + time_until = bus_due-datetime.datetime.utcnow() + time_until = int(time_until.total_seconds()/60) + busses.append([bus_number, time_until]) + if busses: + busses = sorted(busses, key=lambda b: b[-1]) + event["stdout"].write("%s: %s" % (stop_id, ", ".join( + ["%s (in %d minutes)" % (line, due + ) for line, due in busses]))) + else: + event["stderr"].write("%s: No busses due" % stop_id) + else: + event["stderr"].write("Bus ID '%s' unknown" % stop_id) + else: + event["stderr"].write("Please provide a numeric bus stop ID")