push

app.py

@@ -1,8 +1,10 @@
+# encoding: utf8
 import os
 import shutil
 
 from flask import Flask, request, json
 import judge
+
 app = Flask(__name__)
 
 
@@ -39,11 +41,11 @@ def judge():
 
 def save_test_cases(input_output_map, td_total):
     os.mkdir('testcases')
-    for i in range(1, td_total+1):
+    for i in range(1, td_total + 1):
         in_file = open('testcases/' + str(i) + '.in', 'w')
-        in_file.write(input_output_map.keys[i-1])
+        in_file.write(input_output_map.keys[i - 1])
         out_file = open('testcases/' + str(i) + '.out', 'w')
-        out_file.write(input_output_map.values[i-1])
+        out_file.write(input_output_map.values[i - 1])
 
 
 def save_code(language, code):

judge.py

@@ -35,7 +35,8 @@ def compile_java(src_path):
 def compile_python(src_path):
     if os.system('python3 -m py_compile %s' % src_path) != 0:
         return False
-    os.remove('py_compile')
+    if os.path.exists('py_compile'):
+        os.remove('py_compile')
     return True