1 #-*- coding:utf-8 -*- 2 #author: lichmama 3 #email: nextgodhand@163.com 4 #filename: httpd.py 5 import io 6 import os 7 import sys 8 import urllib 9 from BaseHTTPServer import HTTPServer 10 from SimpleHTTPServer import SimpleHTTPRequestHandler 11 12 class MyRequestHandler(SimpleHTTPRequestHandler): 13 protocol_version = "HTTP/1.1" 14 server_version = "PSHS/0.1" 15 sys_version = "Python/2.7.x" 16 target = "D:/web" 17 18 def do_GET(self): 19 if self.path == "/" or self.path == "/index": 20 content = open("signin.html", "rb").read() 21 self.send_head(content) 22 else: 23 path = self.translate_path(self.path) 24 if os.path.exists(path): 25 extn = os.path.splitext(path)[1].lower() 26 content = open(path, "rb").read() 27 self.send_head(content, type=self.extensions_map[extn]) 28 else: 29 content = open("404.html", "rb").read() 30 self.send_head(content, code=404) 31 self.send_content(content) 32 33 def do_POST(self): 34 if self.path == "/signin": 35 data = self.rfile.read(int(self.headers["content-length"])) 36 data = urllib.unquote(data) 37 data = self.parse_data(data) 38 try: 39 uid = data["uid"] 40 if uid != "": 41 content = open("success.html", "rb").read() 42 content = content.replace("$uid", uid) 43 self.send_head(content) 44 #do-something-in-backend 45 if not os.path.exists(self.target + "/" + uid): 46 os.mkdir(self.target + "/" + uid) 47 else: 48 content = "400, bad request." 49 self.send_head(content, code=400) 50 except KeyError: 51 content = "400, bad request." 52 self.send_head(content, code=400) 53 else: 54 content="403, forbiden." 55 self.send_head(content, code=403) 56 self.send_content(content) 57 58 def parse_data(self, data): 59 ranges = {} 60 for item in data.split("&"): 61 k, v = item.split("=") 62 ranges[k] = v 63 return ranges 64 65 def send_head(self, content, code=200, type="text/html"): 66 self.send_response(code) 67 self.send_header("Content-Type", type) 68 self.send_header("Content-Length", str(len(content))) 69 self.end_headers() 70 71 def send_content(self, content): 72 f = io.BytesIO() 73 f.write(content) 74 f.seek(0) 75 self.copyfile(f, self.wfile) 76 f.close() 77 78 if __name__ == "__main__": 79 if len(sys.argv) == 2: 80 #set the target where to mkdir, and default "D:/web" 81 MyRequestHandler.target = sys.argv[1] 82 try: 83 server = HTTPServer(("", 8080), MyRequestHandler) 84 print "pythonic-simple-http-server started, serving at http://localhost:8080" 85 server.serve_forever() 86 except KeyboardInterrupt: 87 server.socket.close()