import cgi import datetime import pytz import workers.conn import workers.val import workers.closest def application(env, re): if env["REQUEST_METHOD"] == "POST": args = cgi.parse_qs(env['wsgi.input'].readline().decode(), True) elif env["REQUEST_METHOD"] == "GET": args = cgi.parse_qs(env['QUERY_STRING'], True) else: re("405 Method Not Allowed", []) return conn = False val = False cfrm = None cto = None split = env["PATH_INFO"].split("/") while "" in split: split.remove("") if len(split) > 2: re("400 Bad Request", []) yield "

400 Bad Request

".encode() yield "Only one (validate) or two (conn) arguments may be passed as path." return if len(split) > 0: if len(split) == 1: val = True else: conn = True cto = split[1] cfrm = split[0] try: rtype = "conn" if conn else "val" if val else args["type"][0] except: re("400 Bad Request", []) yield "

400 Bad Request

".encode() yield "A request type must be provided.".encode() return json = "json" in args if rtype.lower() in ["conn", "connection"]: try: frm = cfrm or args["from"][0] to = cto or args["to"][0] if not frm or not to: raise ValueError() except: re("400 Bad Request", []) yield "

400 Bad Request

".encode() yield "\"from\" and \"to\" values are required for this type of request.".encode() return count = args["count"][0] if "count" in args and args["count"] else 1 date = args["date"][0] if "date" in args and args["date"] else datetime.datetime.strftime(datetime.datetime.now(pytz.timezone("Europe/Vienna")),"%d.%m.%Y") time = args["time"][0] if "time" in args and args["time"] else datetime.datetime.strftime(datetime.datetime.now(pytz.timezone("Europe/Vienna")),"%H:%M") mode = True if "mode" in args and args["mode"] and args["mode"][0].lower() == "arr" else False details = True if "details" in args else False try: count = int(count) if count < 0 or count > 10: raise ValueError() except: re("400 Bad Request", []) yield "

400 Bad Request

".encode() yield "The \"count\" value must be a value between 0 and 10." return try: outtime = datetime.datetime.strptime("%s %s" % (date, time), "%d.%m.%Y %H:%M") except: re("400 Bad Request", []) yield "

400 Bad Request

".encode() yield "The \"date\" value must be in DD.MM.YYYY format, the \"time\" value must be in HH:MM format." return try: output = workers.conn.worker(frm, to, count, outtime, mode, details, json) except Exception as e: re("500 Internal Server Error", []) yield "

500 Internal Server Error

".encode() if "debug" in args: yield str(e).encode() return re("200 OK", [("Content-Type", "application/json" if json else "text/xml")]) yield output.encode() return elif rtype.lower() in ["val", "validate"]: try: name = cfrm or args["name"] if not name: raise ValueError() except: re("400 Bad Request", []) yield "

400 Bad Request

".encode() yield "A \"name\" value is required for this type of request.".encode() return try: output = workers.val.worker(name, json) except Exception as e: re("500 Internal Server Error", []) yield "

500 Internal Server Error

".encode() if "debug" in args: yield str(e).encode() return re("200 OK", [("Content-Type", "application/json" if json else "text/xml")]) yield output.encode() return elif rtype.lower() in ["closest", "close", "near", "nearby"]: try: lat = float(args["lat"][0]) lon = float(args["lon"][0]) if (not lat and not lat == float(0)) or (not lon and not lon == float(0)): raise ValueError() except: re("400 Bad Request", []) yield "

400 Bad Request

".encode() yield "\"lat\" and \"lon\" values are required for this type of request.".encode() return distance = args["distance"][0] if "distance" in args and args["distance"] else 1000 try: distance = int(distance) if distance < 0 or distance > 10000: raise ValueError() except: re("400 Bad Request", []) yield "

400 Bad Request

" yield "\"distance\" must be a value between 0 and 10000." return try: output = workers.closest.worker(lat, lon, distance, json) except Exception as e: re("500 Internal Server Error", []) yield "

500 Internal Server Error

".encode() if "debug" in args: yield e.encode() return re("200 OK", [("Content-Type", "application/json" if json else "text/xml")]) yield output.encode() return else: re("400 Bad Request", []) yield "

400 Bad Request

".encode() yield "The request type you submitted is invalid.".encode() return