fix submission subsitude codes

Author: zdszero

judge/testing.py

@@ -147,14 +147,18 @@ def __init__(self, submission: Submission):
         self.sub_dirpath = PathManager.submission_dir(str(submission.user_id), submission.id)
         if not exists(self.sub_dirpath):
             os.makedirs(self.sub_dirpath, exist_ok=True)
-        prob_dir = PathManager.problem_dir(submission.problem._id) # use display id
-        print('problem id:', submission.problem._id)
-        print('problem dir:', prob_dir)
+        problem: Problem = submission.problem
+        prob_dir = PathManager.problem_dir(problem._id) # use display id
         if not exists(prob_dir):
             raise Exception("problem dir {} not exists".format(prob_dir))
         for filename in os.listdir(prob_dir):
-            print(f'copy {join(prob_dir, filename)} to {self.sub_dirpath}')
             shutil.copy2(join(prob_dir, filename), self.sub_dirpath)
+        for index, codename in enumerate(problem.code_names):
+            filecontent = submission.code_list[index]
+            codepath = join(self.sub_dirpath, codename)
+            with open(codepath, "w") as wfp:
+                print(f"substitude {codepath} with user implemented")
+                wfp.write(filecontent)
 
     def judge(self):
         tester_path = join(self.sub_dirpath, TESTER_NAME)
@@ -174,25 +178,27 @@ def judge(self):
         failed_info = []
         with open(join(log_path, "results.json")) as fp:
             res = json.load(fp)
+            grade = res['grade']
             self.sub.grade = res['grade']
             for idx in res['failed']:
                 log_fp = open(join(log_path, f"testcase{idx}.log"), "r")
                 testcase_fp = open(join(self.sub_dirpath, TESTCASE_NAME), "r")
                 testcase_json_data = json.load(testcase_fp)
+                # notice the index here, logical index starts from 1
                 if "config" in testcase_json_data:
                     # global config
                     config = testcase_json_data["config"]
                 else:
-                    cur_data = testcase_json_data["testcases"][idx] 
+                    cur_data = testcase_json_data["testcases"][idx-1] 
                     if "config" in cur_data:
                         # specific config for testcase
                         config = ["config"]
                     else:
                         # no config
                         config = None
                 displayed_test = {
-                    "input": testcase_json_data["testcases"][idx]["input"],
-                    "expected_output":testcase_json_data["testcases"][idx]["output"],
+                    "input": testcase_json_data["testcases"][idx-1]["input"],
+                    "expected_output":testcase_json_data["testcases"][idx-1]["output"],
                 }
                 # config, "testcase"
                 failed_info.append({
@@ -204,5 +210,10 @@ def judge(self):
                 log_fp.close()
                 testcase_fp.close()
         self.sub.failed_info = failed_info
-        self.sub.result = JudgeStatus.FINISHED
+        if grade == 0:
+            self.sub.result = JudgeStatus.ALL_FAILED
+        elif grade == 100:
+            self.sub.result = JudgeStatus.ALL_PASSED
+        else:
+            self.sub.result = JudgeStatus.SOME_PASSED
         self.sub.save()

submission/models.py

@@ -11,8 +11,10 @@
 class JudgeStatus:
     PENDING      = 0
     JUDGING      = 1
-    FINISHED     = 2
-    SYSTEM_ERROR = 3
+    SYSTEM_ERROR = 2
+    ALL_PASSED   = 3
+    SOME_PASSED  = 4
+    ALL_FAILED   = 5
 
 
 class Submission(models.Model):

submission/serializers.py

@@ -28,7 +28,7 @@ class SubmissionSafeModelSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = Submission
-        exclude = ("info", "contest", "ip")
+        exclude = ("contest", "ip")
 
 
 class SubmissionListSerializer(serializers.ModelSerializer):

submission/views/user.py

@@ -70,7 +70,7 @@ def check_contest_permission(self, request):
     @validate_serializer(CreateSubmissionSerializer)
     @login_required
     def post(self, request):
-        print(request.data)
+        # print(request.data)
         data = request.data
         hide_id = False