server.py

#!/usr/bin/python
from http.server import BaseHTTPRequestHandler, HTTPServer
import simplejson
import numpy as np
from processData import genCatStr
from fitData import fitData

stateL = ["NULL",
"Alabama",
"Alaska",
"Arizona",
"Arkansas",
"California",
"Colorado",
"Connecticut",
"District of Columbia",
"Delaware",
"Florida",
"Georgia",
"Hawaii",
"Idaho",
"Illinois",
"Indiana",
"Iowa",
"Kansas",
"Kentucky",
"Louisiana",
"Maine",
"Maryland",
"Massachusetts",
"Michigan",
"Minnesota",
"Mississippi",
"Missouri",
"Montana",
"Nebraska",
"Nevada",
"New Hampshire",
"New Jersey",
"New Mexico",
"New York",
"North Carolina",
"North Dakota",
"Ohio",
"Oklahoma",
"Oregon",
"Pennsylvania",
"Rhode Island",
"South Carolina",
"South Dakota",
"Tennessee",
"Texas",
"Utah",
"Vermont",
"Virginia",
"Washington",
"West Virginia",
"Wisconsin",
"Wyoming"]

stateDict = { stateL[i]: i for i in range(0,52) }

PORT_NUMBER = 8080


# This class will handles any incoming request from
# the browser
class myHandler(BaseHTTPRequestHandler):
    # Handler for the GET requests
    def do_GET(self):
        self.send_response(200)
        print("Get request: %s" % self.path)
        # Send the html message
        if self.path == "/" or self.path == "index.html":
            self.send_header('Content-type', 'text/html')
            self.end_headers()
            f = open("index.html","r")
            fstr = bytes(f.read(),"utf-8")
            self.wfile.write(fstr)
            f.close()
        #elif self.path[0:3] == "/js":
        #    self.send_header('Content-type', 'text/javascript')
        #    self.end_headers()
        #    f = open("js"+self.path[3:],"r")
        #    fstr = bytes(f.read(),"utf-8")
        #    self.wfile.write(fstr)
        #    f.close()
        #elif self.path[0:4] == "/css":
        #    self.send_header('Content-type', 'text/javascript')
        #    self.end_headers()
        #    f = open("css"+self.path[4:],"r")
        #    fstr = bytes(f.read(),"utf-8")
        #    self.wfile.write(fstr)
        #    f.close()
        else:
            if self.path[-3:] == "css":
                self.send_header('Content-type', 'text/css')
                self.end_headers()
            elif self.path[-2:] == "js":
                self.send_header('Content-type', 'text/javascript')
                self.end_headers()
            try:
                f = open(self.path[1:],"r")
                fstr = bytes(f.read(), "utf-8")
                self.wfile.write(fstr)
                f.close()
            except:
                print("Unable to serve file %s. File not found." % self.path)
        return

    def do_POST(self):
        if self.path == "/quote":
            length = int(self.headers["Content-Length"])
            request = str(self.rfile.read(length), "utf-8")
            print("Request: %s" % request)
            reqdict = simplejson.loads(request)
            print(reqdict)
            age = int(reqdict["age"])
            sex = 0 if reqdict["sex"] == "M" else 1
            height = int(reqdict["height"])
            weight = int(reqdict["weight"])
            try:
                state = stateDict[reqdict["state"]]
            except:
                state = stateDict["NULL"]
            long = reqdict["longitude"]
            lat = reqdict["latitude"]
            marst = reqdict["maritalstatus"]
            tobac = reqdict["tobacco"]
            peepcv = int(reqdict["dependents"])
            optins = reqdict["total"]
            anninc = int(reqdict["annualincome"])
            preconds = reqdict["medical"]

            linestr = "%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d," % (age, sex, height, weight, state, long, lat, marst, tobac, optins, anninc, peepcv)
            linestr += genCatStr(preconds)
            print(len(linestr))

            bronze, silver, gold, platinum, purch = fitData(linestr)

            returndict={"bronze": 0
                       ,"silver": 0
                       ,"gold": 0
                       ,"platinum": 0
                       ,"purchase": 0}
            returndict["bronze"] = round(bronze.item())
            returndict["silver"] = round(silver.item())
            returndict["gold"]   = round(gold.item())
            returndict["platinum"] = round(platinum.item())
            returndict["purchase"] = round(purch)

            returnjson = simplejson.dumps(returndict)

            response = bytes(returnjson, "utf-8")  # create response

            self.send_response(200)  # create header
            self.send_header('Access-Control-Allow-Origin', '*')
            self.send_header("Content-Length", str(len(response)))
            self.end_headers()

            self.wfile.write(response)  # send response
            return
        else:
            self.send_response(404)
            return

    def do_OPTIONS(self):
        self.send_response(200, "ok")
        self.send_header('Access-Control-Allow-Origin', '*')
        self.send_header('Access-Control-Allow-Credentials', 'true')
        self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
        self.send_header("Access-Control-Allow-Headers", "X-Requested-With, Content-type")
        self.send_header("Access-Control-Max-Age", "36000")
        self.end_headers()
        return

try:
    # Create a web server and define the handler to manage the
    # incoming request
    server = HTTPServer(('', PORT_NUMBER), myHandler)
    print('Started httpserver on port', PORT_NUMBER)

    # Wait forever for incoming htto requests
    server.serve_forever()

except KeyboardInterrupt:
    server.socket.close()
    print('^C received, shutting down the web server')