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