server.py

import time
import socket
import json
import processor
import sys
import urlparse
import os

try:
    import BaseHTTPServer
except:
    os.system("pip install BaseHTTPServer")
    time.sleep(1)
    import BaseHTTPServer
'''import admin
if not admin.isUserAdmin():
    admin.runAsAdmin()'''

HOST_NAME = ""
PORT_NUMBER = 8080
SOURCE = ""
START = os.getcwd()
for arg in sys.argv:
    try:
        PORT_NUMBER = int(arg)
        break
    except:
        PORT_NUMBER = 8080

class handler(BaseHTTPServer.BaseHTTPRequestHandler):
    '''def do_HEAD(self, s):
        s.send_response(200)
        s.send_header("Content-type", "text/html")
        s.send_header("Access-Control-Allow-Origin", "*")
        s.end_headers()'''
    def do_GET(self):
        #This is the general handler for GET requests
        self.handleF()	
        pass
    def do_POST(self):
        #This is the general handler for POST requests
        self.handleF()
        pass

    def get_parameters(self, path):
        parts = path.split("/", 1)
        l = len(parts)
        if(l <= 1):
            return None, None
        last = parts[-1]
        arguments = last.split("?", 1)
        last = arguments[0]
        if(len(arguments) > 1):
            parameters = arguments[1]
            parameters = parameters.split("&")
            dict = {}
            for item in parameters:
                parts = item.split("=")
                dict[parts[0]] = parts[1]
            return last, dict
        else:
            return last, None

    
    def handleF(self):
        #function need to be changed for 404 to be returned properly
        os.chdir(START)
        path, params = self.get_parameters(self.path)
        if(not ".ico" in path):
            path = SOURCE + path
        
        if((path == "" or "." not in path[-4:-1]) and os.path.isfile(path + "index.psp")):
            path += "index.psp"
            
        if(not os.path.isfile(path)):
            with open(os.getcwd()+"//404.html") as file:
                print("404 not found!")
                self.send_response(200)
                self.wfile.write(file.read())
            self.send_header("Content-type","text/html")
            return
            
        req = request("GET", self)
        self.send_response(200)
        self.send_header("Access-Control-Allow-Origin", "*")
        
        if("psp" in path.split(".")[1]):
            st = processor.process(path, params, req)
            self.send_header("Content-type", "text/html")
        elif("jpg" in path.split(".")[1].lower()):
            with open(path, "rb") as file:
                st = file.read()
            self.send_header("Content-type", "image/jpeg")
        elif("png" in path.split(".")[1].lower()):
            with open(path, "rb") as file:
                st = file.read()
            self.send_header("Content-type", "image/png")
        elif(path == "close" or path == "exit" or path == "stop"):
            raise KeyboardInterrupt()
        else:
            with open(path, "r") as file:
                st = file.read()
            self.send_header("Content-type", "text/html")
        
        
        self.end_headers()
        self.wfile.write(st)
        #self.wfile.write("You enetered path: " + path +" and parameters: " + json.dumps(params))

def run_server():
    server_class = BaseHTTPServer.HTTPServer
    httpd = server_class(('', PORT_NUMBER), handler)
    print("PSAP Running on:" + str(PORT_NUMBER))
    httpd.serve_forever()
    httpd.server_close()
    

class request:
    def __init__(self, method, req):
        if(method == "POST"):
            self.method = "POST"
            parsed = urlparse.urlparse(req.path)
            self.post = urlparse.parse_qs(parsed.query)
            print(self.post)
            self.get = None
            k = self.post.keys()
            for key in k:
                self.post[key] = self.post[key][0]
        else:
            self.method = "GET"
            parsed = urlparse.urlparse(req.path)
            self.get = urlparse.parse_qs(parsed.query)
            self.post = None
            k = self.get.keys()
            for key in k:
                self.get[key] = self.get[key][0]