b@1551
|
1 import BaseHTTPServer
|
b@1551
|
2 from os import walk
|
b@1551
|
3 from os import path
|
b@1551
|
4 import urllib2
|
b@1551
|
5 import pickle
|
b@1551
|
6 import datetime
|
nickjillings@1542
|
7
|
b@1551
|
8 curSaveIndex = 0;
|
b@1551
|
9 curFileName = 'test-0.xml'
|
b@1551
|
10 while(path.isfile('saves/'+curFileName)):
|
b@1551
|
11 curSaveIndex += 1;
|
b@1551
|
12 curFileName = 'test-'+str(curSaveIndex)+'.xml'
|
nickjillings@1542
|
13
|
b@1551
|
14 print curFileName
|
nickjillings@1542
|
15
|
b@1551
|
16 def send404(s):
|
b@1551
|
17 s.send_response(404)
|
b@1551
|
18 s.send_header("Content-type", "text/html")
|
b@1551
|
19 s.end_headers()
|
b@1551
|
20
|
b@1551
|
21 def processFile(s):
|
b@1551
|
22 s.path = s.path[1:len(s.path)]
|
b@1551
|
23 st = s.path.rsplit(',')
|
b@1551
|
24 lenSt = len(st)
|
b@1551
|
25 fmt = st[lenSt-1].rsplit('.')
|
b@1551
|
26 size = path.getsize(urllib2.unquote(s.path))
|
b@1551
|
27 fileDump = open(urllib2.unquote(s.path))
|
b@1551
|
28 s.send_response(200)
|
b@1551
|
29
|
b@1551
|
30 if (fmt[1] == 'html'):
|
b@1551
|
31 s.send_header("Content-type", 'text/html')
|
b@1551
|
32 elif (fmt[1] == 'css'):
|
b@1551
|
33 s.send_header("Content-type", 'text/css')
|
b@1551
|
34 elif (fmt[1] == 'js'):
|
b@1551
|
35 s.send_header("Content-type", 'application/javascript')
|
b@1551
|
36 else:
|
b@1551
|
37 s.send_header("Content-type", 'application/octet-stream')
|
b@1551
|
38 s.send_header("Content-Length", size)
|
b@1551
|
39 s.end_headers()
|
b@1551
|
40 s.wfile.write(fileDump.read())
|
b@1551
|
41 fileDump.close()
|
b@1551
|
42
|
b@1551
|
43 def saveFile(self):
|
b@1551
|
44 global curFileName
|
b@1551
|
45 global curSaveIndex
|
b@1551
|
46 varLen = int(self.headers['Content-Length'])
|
b@1551
|
47 postVars = self.rfile.read(varLen)
|
b@1551
|
48 print curFileName
|
b@1551
|
49 file = open('saves/'+curFileName,'w')
|
b@1551
|
50 curSaveIndex += 1;
|
b@1551
|
51 curFileName = 'test-'+str(curSaveIndex)+'.xml'
|
b@1551
|
52 print curFileName
|
b@1551
|
53 file.write(postVars)
|
b@1551
|
54 file.close()
|
b@1551
|
55 self.send_response(200)
|
b@1551
|
56 self.send_header("Content-type", "text/xml")
|
b@1551
|
57 self.end_headers()
|
b@1551
|
58 self.wfile.write('<response><state>OK</state><file>saves/'+curFileName+'</file></response>')
|
nickjillings@1542
|
59
|
b@1551
|
60 class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):
|
b@1551
|
61 def do_HEAD(s):
|
b@1551
|
62 s.send_response(200)
|
b@1551
|
63 s.send_header("Content-type", "text/html")
|
b@1551
|
64 s.end_headers()
|
b@1551
|
65 def do_GET(request):
|
b@1551
|
66 if(request.client_address[0] == "127.0.0.1"):
|
b@1551
|
67 if (request.path == "/favicon.ico"):
|
b@1551
|
68 send404(request)
|
b@1551
|
69 else:
|
b@1551
|
70 if (request.path == '/'):
|
b@1551
|
71 request.path = '/index.html'
|
b@1551
|
72 processFile(request)
|
b@1551
|
73 else:
|
b@1551
|
74 send404(request)
|
b@1551
|
75 def do_POST(request):
|
b@1551
|
76 if(request.client_address[0] == "127.0.0.1"):
|
b@1551
|
77 if (request.path == "/save"):
|
b@1551
|
78 saveFile(request)
|
b@1551
|
79 else:
|
b@1551
|
80 send404(request)
|
b@1551
|
81
|
b@1551
|
82 def run(server_class=BaseHTTPServer.HTTPServer,
|
b@1551
|
83 handler_class=MyHandler):
|
b@1551
|
84 server_address = ('', 8000)
|
b@1551
|
85 httpd = server_class(server_address, handler_class)
|
b@1551
|
86 httpd.serve_forever()
|
b@1551
|
87
|
b@1551
|
88 run() |