From f901e1e49e13faf35db375bbc95f1d94cea464dc Mon Sep 17 00:00:00 2001 From: Frederick Yin Date: Sun, 10 Oct 2021 18:17:38 +0800 Subject: Fix JOJWorker: judge specified hgroup only --- worker/JOJWorker.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'worker') 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))] -- cgit v1.2.3