PLD_Builder/request_handler_server.py

   1 #!/usr/bin/python
   2
   3 import string
   4 import cgi
   5 import time
   6 import StringIO
   7 from config import config, init_conf
   8
   9 from os import curdir, sep
  10 from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
  11
  12 import request_handler
  13
  14 class MyHandler(BaseHTTPRequestHandler):
  15
  16         def do_GET(self):
  17                 self.send_error(401);
  18
  19         def do_POST(self):
  20                 global rootnode
  21                 try:
  22                         length = int(self.headers.getheader('content-length'))
  23                         ctype, pdict = cgi.parse_header(self.headers.getheader('content-type'))
  24                         if ctype != 'application/x-www-form-urlencoded':
  25                                 self.send_error(401)
  26                                 self.end_headers()
  27                                 return
  28
  29                         query = self.rfile.read(length)
  30                         sio = StringIO.StringIO()
  31                         sio.write(query)
  32                         sio.seek(0)
  33
  34                         filename = self.headers.getheader('x-filename')
  35
  36                         if not request_handler.handle_request_main(sio, filename = filename):
  37                                 self.send_error(500)
  38                                 self.end_headers()
  39                                 return
  40
  41                         self.send_response(200)
  42                         self.end_headers()
  43
  44                 except Exception, e:
  45                         self.send_error(500)
  46                         self.end_headers()
  47                         raise
  48                         pass
  49
  50 def main():
  51         try:
  52                 init_conf()
  53                 host = ""
  54                 port = config.request_handler_server_port
  55
  56                 server = HTTPServer((host, port), MyHandler)
  57                 print 'started httpserver on :%d...' % port
  58                 server.serve_forever()
  59         except KeyboardInterrupt:
  60                 print '^C received, shutting down server'
  61                 server.socket.close()
  62
  63 if __name__ == '__main__':
  64         main()
  65