145 lines
5.8 KiB
Python
145 lines
5.8 KiB
Python
from rapidsms.apps.base import AppBase
|
|
import re
|
|
import arrest
|
|
|
|
MAX_MSG_LEN = 160
|
|
DIGIT = re.compile(r"\d{1,3}")
|
|
PUNCT = re.compile(r"[^\w\s]")
|
|
"""
|
|
STYLE = {
|
|
"start": "-* ",
|
|
"repeat": " -*- ",
|
|
"end": " *-"
|
|
}
|
|
"""
|
|
|
|
STYLE = {"start": "", "repeat": "; ", "end": ""}
|
|
|
|
|
|
ChaloBest = arrest.Client("http://chalobest.in/1.0")
|
|
|
|
def get_routes_for_matches(stops):
|
|
# same_stops = []
|
|
# same_stops.append(stops[0])
|
|
# if len(stops) > 1:
|
|
# for s in stops[1:]:
|
|
# if s['properties']['official_name'] == stops[0]['properties']['official_name']:
|
|
# same_stops.append(s)
|
|
routes = []
|
|
for stop in stops:
|
|
routes.extend(stop['properties']['routes'].split(", "))
|
|
return routes
|
|
|
|
def get_stops_for_string(s):
|
|
'''
|
|
Returns a dict with name and stops to associate with given search string.
|
|
First looks at areas, if string matches an area, returns area as name and all stops within that area as stops.
|
|
If search term does not match an area, returns the first / best match of stop to string. If there are multiple stops with the same name, returns all of them in the list of stops.
|
|
'''
|
|
stops = []
|
|
s = s.strip()
|
|
# areas = ChaloBest.areas(q=s)
|
|
areas = [area for area in ChaloBest.areas(q=s) if area.lower().startswith(s)]
|
|
if len(areas) > 0:
|
|
for a in areas:
|
|
area = ChaloBest.area[a]
|
|
for stop in area['stops']['features']:
|
|
stops.append(stop)
|
|
return {
|
|
'name': ", ".join(areas),
|
|
'stops': stops
|
|
}
|
|
else:
|
|
stops_results = ChaloBest.stops(q=s)['features']
|
|
if len(stops_results) == 0:
|
|
return None
|
|
same_stops = []
|
|
same_stops.append(stops_results[0])
|
|
if len(stops_results) > 1:
|
|
for s in stops_results[1:]:
|
|
if s['properties']['official_name'] == same_stops[0]['properties']['official_name']:
|
|
same_stops.append(s)
|
|
return {
|
|
'name': same_stops[0]['properties']['display_name'],
|
|
'stops': same_stops
|
|
}
|
|
|
|
class App(AppBase):
|
|
def handle(self, msg):
|
|
if DIGIT.search(msg.text):
|
|
routes = ChaloBest.routes(q=msg.text.replace(" ", ""))
|
|
if not routes:
|
|
msg.respond("Sorry, we found no route marked '%(text)s'.", text=msg.text)
|
|
return
|
|
detail = None
|
|
for route in routes:
|
|
if route.replace(" ", "").upper() == msg.text.replace(" ", "").upper():
|
|
detail = ChaloBest.route[route]
|
|
if detail == None:
|
|
detail = ChaloBest.route[routes[0]]
|
|
detail = ChaloBest.route[routes[0]]
|
|
stops = detail['stops']['features']
|
|
origin, dest = stops[0]['properties'], stops[-1]['properties']
|
|
origin_name, dest_name = origin['display_name'], dest['display_name']
|
|
origin_area, dest_area = PUNCT.sub('', origin['area']), PUNCT.sub('', dest['area'])
|
|
url = "http://chalobest.in" + detail['route']['url']
|
|
response = "%s: %s (%s) to %s (%s) %s" % (
|
|
",".join(routes), origin_name, origin_area, dest_name, dest_area, url)
|
|
|
|
elif msg.text.find(" to ") != -1:
|
|
|
|
from_txt = msg.text.lower().split("to")[0].strip()
|
|
to_txt = msg.text.lower().split("to")[1].strip()
|
|
|
|
from_matches = get_stops_for_string(from_txt)
|
|
to_matches = get_stops_for_string(to_txt)
|
|
|
|
# stop1matches = ChaloBest.stops(q=stop1txt)['features']
|
|
if not from_matches:
|
|
msg.respond("Sorry, found no stop matching '%s'" % from_txt)
|
|
return
|
|
#best_match1 = stop1matches[0]
|
|
routes1 = set(get_routes_for_matches(from_matches['stops']))
|
|
#stop2matches = ChaloBest.stops(q=stop2txt)['features']
|
|
if not to_matches:
|
|
msg.respond("Sorry, found no stop matching '%s'" % to_txt)
|
|
#best_match2 = stop2matches[0]
|
|
routes2 = set(get_routes_for_matches(to_matches['stops']))
|
|
#routes1arr = set(routes1.split(", "))
|
|
#routes2arr = set(routes2.split(", "))
|
|
intersection = list(routes1.intersection(routes2))
|
|
if len(intersection) == 0:
|
|
msg.respond("Sorry, no direct buses found between %s and %s" % (from_matches['name'], to_matches['name'],))
|
|
return
|
|
routesFound = ", ".join(intersection)
|
|
response = "%s to %s: %s" % (from_matches['name'], to_matches['name'], routesFound,)
|
|
if len(response) > MAX_MSG_LEN:
|
|
response = response[0:MAX_MSG_LEN]
|
|
#msg.respond("%s to %s: %s" % (from_matches['name'], to_matches['name'], routesFound,))
|
|
|
|
|
|
|
|
else:
|
|
features = ChaloBest.stops(q=msg.text)['features']
|
|
if not features:
|
|
msg.respond("Sorry, we found no stops marked '%(text)s'.", text=msg.text)
|
|
return
|
|
stops = []
|
|
for feat in features:
|
|
stop = feat['properties']
|
|
if stops and stop["official_name"] == stops[-1]["official_name"]:
|
|
stops[-1]["routes"] += ", " + stop["routes"]
|
|
else:
|
|
stops.append(stop)
|
|
response = STYLE["start"]
|
|
for stop in stops:
|
|
match = stop["official_name"] + ": " + stop["routes"]
|
|
if len(response) > len(STYLE["repeat"]): response += STYLE["repeat"]
|
|
response += match
|
|
if len(response) > MAX_MSG_LEN: break
|
|
if len(response) > MAX_MSG_LEN:
|
|
response = response[:MAX_MSG_LEN-(len(STYLE["end"])+4)] + "..."
|
|
response += STYLE["end"]
|
|
|
|
msg.respond(response)
|