Merge pull request #14 from NJUST-FishTeam/dev_comzyh

add NoTestDataError

.gitignore

@@ -64,3 +64,6 @@ Core
 *.in
 *.out
 *.log
+testdata/
+tmp/
+core_log.txt

judgesite/task.py

@@ -12,6 +12,10 @@
 from models import save_result
 
 
+class NoTestDataException(Exception):
+    pass
+
+
 class JudgeTask(object):
 
     def __init__(self, message):
@@ -24,18 +28,28 @@ def __init__(self, message):
         self.time_limit = str(task["time_limit"])
         self.memory_limit = str(task["memory_limit"])
 
+        self.result = ""
+        self.run_time = 0
+        self.run_memory = 0
+        self.others = ""
+
     def go(self):
         self._clean_files()
 
-        self._prepare_temp_dir()
-
-        self._dump_code_to_file()
+        try:
+            self._prepare_temp_dir()
 
-        self._prepare_testdata_file()
+            self._dump_code_to_file()
 
-        self._run()
+            self._prepare_testdata_file()
+        except NoTestDataException, e:
+            self.result = 'NoTestDataError'
+        except Exception, e:
+            raise e
+        else:
+            self._run()
 
-        self._read_result()
+            self._read_result()
 
         self._save_result()
 
@@ -55,8 +69,12 @@ def _dump_code_to_file(self):
 
     def _prepare_testdata_file(self):
         logging.info("Prepare testdata")
-        input_file = os.path.join(conf.testdata_path, self.testdata_id, "in.in")
-        output_file = os.path.join(conf.testdata_path, self.testdata_id, "out.out")
+        input_file = os.path.join(
+            conf.testdata_path, self.testdata_id, "in.in")
+        output_file = os.path.join(
+            conf.testdata_path, self.testdata_id, "out.out")
+        if not os.path.exists(input_file) or not os.path.exists(output_file):
+            raise NoTestDataException
         shutil.copy(input_file, conf.tmp_path)
         shutil.copy(output_file, conf.tmp_path)