bestSMS/best/smsmesn/handlers/fromto.py
2011-08-06 19:53:50 +05:30

108 lines
3.0 KiB
Python

#!/usr/bin/env python
# vim: ai ts=4 sts=4 et sw=4
from rapidsms.contrib.handlers.handlers.keyword import KeywordHandler
# from new_best.models import *
import datetime
try:
import json
except:
import simplejson as json
from settings import JSON_PATH
from os.path import join
def getStops(data, s):
stops = []
for d in data:
if d[1].lower() == s:
return [d]
else:
if d[0].lower().find(s) != -1:
stops.append(d)
return stops
def getStop(data, s):
for d in data:
if d[1] == s:
return d
return [""]
def getOptions(stops):
arr = []
for s in stops:
arr.append(s[1])
stopStr = ",".join(arr[0:4])
return '(' + stopStr + ')'
def getBuses(frm, to):
# return "hi"
from_code = frm[1].strip()
to_code = to[1].strip()
# url = "http://mesn.org/Bestbus/busesfound.html"
data = json.loads(open(join(JSON_PATH, "stopbus.json")).read().replace("\\", "/").replace(",,,", ",").replace(",,", ","))
to = getStop(data, to_code)
frm = getStop(data, from_code)
# return to[0]
stopListFrom = frm[-1]
stopListTo = to[-1]
fromArr = stopListFrom.split(",")
fromArr = [val.split("|")[0] for val in fromArr]
toArr = stopListTo.split(",")
toArr = [val.split("|")[0] for val in toArr]
intersectArr = [val for val in fromArr if val in toArr]
fromToStr = "from " + frm[1] + " to " + to[1] + ": "
# return ",".join(fromArr) + ",".join(toArr)
if (len(intersectArr) == 0):
return fromToStr + "no direct buses found."
# return "from %s to %s" % (",".join(fromArr), ",".join(toArr),)
else:
return fromToStr + ",".join(intersectArr)
class FromToHandler(KeywordHandler):
keyword = "from"
def help(self):
self.respond("Send from <place> to <place>")
def handle(self, text):
txt = text.strip()
from_to = txt.split("to")
frm = from_to[0].strip()
to = from_to[1].strip()
# print frm
# print to
data = json.loads(open(join(JSON_PATH, "stoplist.json")).read().replace("\\", "/"))
froms = getStops(data, frm)
tos = getStops(data, to)
if len(froms) == len(tos) == 1:
self.respond(getBuses(froms[0], tos[0]))
else:
if len(froms) == 0:
fromString = "From location not found."
elif len(froms) == 1:
fromString = "From %s" % (froms[0][1])
else:
fromString = "From %s" % (getOptions(froms))
if len(tos) == 0:
toString = "To location not found."
elif len(tos) == 1:
toString = "to %s" % (tos[0][1])
else:
toString = "to %s" % (getOptions(tos))
toString = "To location not found."
self.respond(fromString + " " + toString)
def time_str(no):
whole = int(no)
decimal = (no - whole) * 10
mins = int(((decimal + .0) / 10) * 60)
return str(whole) + ":" + str(mins)