31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
from rapidsms.apps.base import AppBase
|
|
import re
|
|
import arrest
|
|
|
|
MAX_MSG_LEN = 160
|
|
DIGIT = re.compile(r"\d{1,3}")
|
|
chalobest = arrest.Client("http://chalobest.in/1.0")
|
|
|
|
class App(AppBase):
|
|
def handle(self, msg):
|
|
if DIGIT.search(msg.text):
|
|
routes = chalobest.routes(q=msg.text.replace(" ", ""))
|
|
detail = chalobest.route[routes[0]]
|
|
stops = detail['stops']['features']
|
|
origin, destination = stops[0]['properties'], stops[-1]['properties']
|
|
msg.respond("%s: %s (%s) to %s (%s)" % (routes[0],
|
|
origin['display_name'], origin['area'],
|
|
destination['display_name'], destination['area']))
|
|
else:
|
|
stops = chalobest.stops(q=msg.text)['features']
|
|
response = ""
|
|
for feature in stops:
|
|
stop = feature['properties']
|
|
match = "%s (%s): %s" % (stop['official_name'], stop['area'], stop['routes'])
|
|
if len(response) + len(match) + 1 < MAX_MSG_LEN:
|
|
if response: response += ";"
|
|
response += match
|
|
else:
|
|
break
|
|
msg.respond(response)
|