Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

did some tweaks to tfl.py.

Browse source
This commit is contained in:
jesopo 2016-06-28 00:18:09 +01:00
parent f21618eeb1
commit 6da747981f
No known key found for this signature in database
GPG key ID: 0BBDEB2AEFCFFCB3
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
modules/tfl.py
View file

@ -17,7 +17,6 @@ class Module(object):
if stop_id.isdigit(): if stop_id.isdigit():
bus_search = Utils.get_url(URL_BUS_SEARCH % stop_id, get_params={ bus_search = Utils.get_url(URL_BUS_SEARCH % stop_id, get_params={
"app_id": app_id, "app_key": app_key}, json=True) "app_id": app_id, "app_key": app_key}, json=True)
print(bus_search)
if bus_search["matches"]: if bus_search["matches"]:
bus_stop = bus_search["matches"][0] bus_stop = bus_search["matches"][0]
real_stop_id = bus_stop["id"] real_stop_id = bus_stop["id"]
@ -34,9 +33,15 @@ class Module(object):
busses.append([bus_number, time_until]) busses.append([bus_number, time_until])
if busses: if busses:
busses = sorted(busses, key=lambda b: b[-1]) busses = sorted(busses, key=lambda b: b[-1])
for i, bus in enumerate(busses):
if bus[-1] == 0:
bus[-1] = "due"
elif bus[-1] == 1:
bus[-1] = "in 1 minute"
else:
bus[-1] = "in %d minutes" % bus[-1]
event["stdout"].write("%s: %s" % (stop_id, ", ".join( event["stdout"].write("%s: %s" % (stop_id, ", ".join(
["%s (in %d minutes)" % (line, due ["%s (%s)" % (number, due) for number, due in busses])))
) for line, due in busses])))
else: else:
event["stderr"].write("%s: No busses due" % stop_id) event["stderr"].write("%s: No busses due" % stop_id)
else: else: