chaloBEST/smsBEST/mumbai/app.py

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):
    stops = []
    s = s.strip()
    areas = ChaloBest.areas(q=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'])
            msg.respond("%s: %s (%s) to %s (%s)" % (
                    ",".join(routes), origin_name, origin_area, dest_name, dest_area))
        elif msg.text.find(" to ") != -1:

            from_txt = msg.text.split("to")[0].strip()
            to_txt = msg.text.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)
            msg.respond("%s to %s: %s" % (from_matches['name'], to_matches['name'], routesFound,))
            return
            

        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)