nickjillings@1453: import BaseHTTPServer nickjillings@1453: from os import walk nickjillings@1453: from os import path nickjillings@1453: from os import listdir nickjillings@1453: import inspect nickjillings@1453: import os nickjillings@1453: import urllib2 nickjillings@1453: import pickle nickjillings@1453: import datetime nickjillings@1453: nickjillings@1453: # Go to right folder. nickjillings@1453: scriptdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))) # script directory nickjillings@1453: os.chdir(scriptdir) # does this work? nickjillings@1453: nickjillings@1453: PSEUDO_PATH = 'example_eval/' nickjillings@1453: pseudo_files = [] nickjillings@1453: for filename in listdir(PSEUDO_PATH): nickjillings@1453: if filename.endswith('.xml'): nickjillings@1453: pseudo_files.append(filename) nickjillings@1453: nickjillings@1453: curSaveIndex = 0; nickjillings@1453: curFileName = 'test-0.xml' nickjillings@1453: while(path.isfile('saves/'+curFileName)): nickjillings@1453: curSaveIndex += 1; nickjillings@1453: curFileName = 'test-'+str(curSaveIndex)+'.xml' nickjillings@1453: nickjillings@1453: print "Next save - " + curFileName nickjillings@1453: pseudo_index = curSaveIndex % len(pseudo_files) nickjillings@1453: print "Next test in pseudo-random queue - " + pseudo_files[pseudo_index] nickjillings@1453: nickjillings@1453: def send404(s): nickjillings@1453: s.send_response(404) nickjillings@1453: s.send_header("Content-type", "text/html") nickjillings@1453: s.end_headers() nickjillings@1453: nickjillings@1453: def processFile(s): nickjillings@1453: s.path = s.path[1:len(s.path)] nickjillings@1453: st = s.path.rsplit(',') nickjillings@1453: lenSt = len(st) nickjillings@1453: fmt = st[lenSt-1].rsplit('.') nickjillings@1453: size = path.getsize(urllib2.unquote(s.path)) nickjillings@1453: fileDump = open(urllib2.unquote(s.path)) nickjillings@1453: s.send_response(200) nickjillings@1453: nickjillings@1453: if (fmt[1] == 'html'): nickjillings@1453: s.send_header("Content-type", 'text/html') nickjillings@1453: elif (fmt[1] == 'css'): nickjillings@1453: s.send_header("Content-type", 'text/css') nickjillings@1453: elif (fmt[1] == 'js'): nickjillings@1453: s.send_header("Content-type", 'application/javascript') nickjillings@1453: else: nickjillings@1453: s.send_header("Content-type", 'application/octet-stream') nickjillings@1453: s.send_header("Content-Length", size) nickjillings@1453: s.end_headers() nickjillings@1453: s.wfile.write(fileDump.read()) nickjillings@1453: fileDump.close() nickjillings@1453: nickjillings@1453: def saveFile(self): nickjillings@1453: global curFileName nickjillings@1453: global curSaveIndex nickjillings@1453: varLen = int(self.headers['Content-Length']) nickjillings@1453: postVars = self.rfile.read(varLen) nickjillings@1453: print curFileName nickjillings@1453: file = open('saves/'+curFileName,'w') nickjillings@1453: curSaveIndex += 1; nickjillings@1453: curFileName = 'test-'+str(curSaveIndex)+'.xml' nickjillings@1453: print curFileName nickjillings@1453: file.write(postVars) nickjillings@1453: file.close() nickjillings@1453: self.send_response(200) nickjillings@1453: self.send_header("Content-type", "text/xml") nickjillings@1453: self.end_headers() nickjillings@1453: self.wfile.write('OKsaves/'+curFileName+'') nickjillings@1453: nickjillings@1453: class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler): nickjillings@1453: def do_HEAD(s): nickjillings@1453: s.send_response(200) nickjillings@1453: s.send_header("Content-type", "text/html") nickjillings@1453: s.end_headers() nickjillings@1453: def do_GET(request): nickjillings@1453: global pseudo_index nickjillings@1453: global pseudo_files nickjillings@1453: global PSEUDO_PATH nickjillings@1453: if(request.client_address[0] == "127.0.0.1"): nickjillings@1453: if (request.path == "/favicon.ico"): nickjillings@1453: send404(request) nickjillings@1453: else: nickjillings@1453: if (request.path == '/'): nickjillings@1453: request.path = '/index.html' nickjillings@1453: elif (request.path == '/pseudo.xml'): nickjillings@1453: request.path = '/'+PSEUDO_PATH + pseudo_files[pseudo_index] nickjillings@1453: print request.path nickjillings@1453: pseudo_index += 1 nickjillings@1453: pseudo_index %= len(pseudo_files) nickjillings@1453: processFile(request) nickjillings@1453: else: nickjillings@1453: send404(request) nickjillings@1453: nickjillings@1453: def do_POST(request): nickjillings@1453: if(request.client_address[0] == "127.0.0.1"): nickjillings@1453: if (request.path == "/save"): nickjillings@1453: saveFile(request) nickjillings@1453: else: nickjillings@1453: send404(request) nickjillings@1453: nickjillings@1453: def run(server_class=BaseHTTPServer.HTTPServer, nickjillings@1453: handler_class=MyHandler): nickjillings@1453: server_address = ('', 8000) nickjillings@1453: httpd = server_class(server_address, handler_class) nickjillings@1453: httpd.serve_forever() nickjillings@1453: nickjillings@1453: run()