diff python/pythonServer.py @ 2934:260efd43fe52

#228
author Nicholas Jillings <nicholas.jillings@mail.bcu.ac.uk>
date Mon, 11 Sep 2017 17:41:40 +0100
parents 3bfc80d2ead0
children 8805556c0f42 03b0e5dc57e6
line wrap: on
line diff
--- a/python/pythonServer.py	Mon Sep 11 17:22:23 2017 +0100
+++ b/python/pythonServer.py	Mon Sep 11 17:41:40 2017 +0100
@@ -173,6 +173,35 @@
         self.wfile.write(bytes(reply, "utf-8"))
     curSaveIndex += 1
     curFileName = 'test-'+str(curSaveIndex)+'.xml'
+    
+def testSave(self):
+    self.send_response(200)
+    self.send_header("Content-type", "text/xml")
+    self.end_headers()
+    filename = "../saves/test-save.xml"
+    file = open(filename,'wb')
+    if sys.version_info[0] == 2:
+        file.write("<xml></xml>")
+    elif sys.version_info[0] == 3:
+        file.write(bytes("<xml></xml>", "utf-8"))
+    file.close()
+    message = ""
+    try:
+        wbytes = os.path.getsize(filename)
+    except OSError:
+        message = '<response state="error"><message>Could not open file</message></response>';
+        if sys.version_info[0] == 2:
+            self.wfile.write(message)
+        elif sys.version_info[0] == 3:
+            self.wfile.write(bytes(message, "utf-8"))
+        return
+    os.remove(filename)
+    message = '<response state="OK"><message>OK</message></response>';
+    if sys.version_info[0] == 2:
+        self.wfile.write(message)
+    elif sys.version_info[0] == 3:
+        self.wfile.write(bytes(message, "utf-8"))
+    
 
 def poolXML(s):
     pool = ET.parse('../tests/pool.xml')
@@ -243,6 +272,8 @@
             requestKey(request);
         elif (request.path.split('?',1)[0] == "/php/pool.php"):
             poolXML(request);
+        elif (request.path.split('?',1)[0] == "/php/test_write.php"):
+            testSave(request);
         else:
             request.path = request.path.split('?',1)[0]
             if (request.path == '/'):