diff options
Diffstat (limited to 'worker')
| -rw-r--r-- | worker/JOJWorker.py | 7 |
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))] |