Fix JOJWorker: judge specified hgroup only

1 files changed, 4 insertions, 3 deletions

diff --git a/worker/JOJWorker.py b/worker/JOJWorker.py
index fd7566d..df761f8 100644
--- a/worker/JOJWorker.py
+++ b/worker/JOJWorker.py
@@ -5,7 +5,7 @@ import requests
 import zipfile
 import time
 import os
-
+from config import HGROUP_NO
 
 class JOJWorker():
     def __init__(self, args, courseID, sid, hgroups, logger=Logger()):
@@ -81,6 +81,7 @@ class JOJWorker():
         return res
 
     def checkGroupJOJProcess(self, groupNum, hwNum, jojInfo, fns, problemID):
+        self.logger.debug(f"{groupNum} h{hwNum}: {repr(jojInfo)} \n{fns} \n{problemID}")
         groupName = f"hgroup-{groupNum:02}"
         hwDir = os.path.join('hwrepos', groupName, f"h{hwNum}")
         if not os.path.exists(hwDir): return 0
@@ -104,11 +105,11 @@ class JOJWorker():
     def checkGroupJOJ(self, jojInfo):
         res = {}
         hwNum = self.args.hw
-        for i, (key, value) in enumerate(self.hgroups.items()):
+        for key, value in self.hgroups.items():
             with multiprocessing.Pool(len(jojInfo["problemInfo"])) as p:
                 scores = p.starmap(
                     self.checkGroupJOJProcess,
-                    [[i, hwNum, jojInfo, fns, problemID]
+                    [[HGROUP_NO, hwNum, jojInfo, fns, problemID]
                      for fns, problemID, _ in jojInfo["problemInfo"]])
             scores = [(scores[i], jojInfo["problemInfo"][i][2])
                       for i in range(len(scores))]