From 11657edd4519e0432aa75324fdb0f9e9640b6494 Mon Sep 17 00:00:00 2001
From: BoYanZh <32470225+BoYanZh@users.noreply.github.com>
Date: Mon, 5 Oct 2020 02:48:10 +0800
Subject: update: gitea worker and minor fix

---
 VG101GradeHelper.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'VG101GradeHelper.py')

diff --git a/VG101GradeHelper.py b/VG101GradeHelper.py
index 677a25e..507296b 100644
--- a/VG101GradeHelper.py
+++ b/VG101GradeHelper.py
@@ -4,7 +4,7 @@ import argparse
 import json
 import os
 
-from worker import CanvasWorker, GitWorker, JOJWorker
+from worker import CanvasWorker, GitWorker, JOJWorker, GiteaWorker
 from settings import *
 
 
@@ -16,6 +16,11 @@ def parse():
     parser.add_argument('-h', '--hw', type=int, help='# homework')
     parser.add_argument('-p', '--proj', type=int, help='# project')
     parser.add_argument('-m', '--ms', type=int, help='# milestone')
+    parser.add_argument('-r',
+                        '--rejudge',
+                        type=int,
+                        default=-1,
+                        help='rejudge group num or stu ID')
     parser.add_argument('-a', '--all', action='store_true', help='check all')
     parser.add_argument('-s',
                         '--score',
@@ -63,7 +68,7 @@ if __name__ == "__main__":
     args = parse()
     indvScores, groupScores, jojScores = {}, {}, {}
     gitWorker = GitWorker(args, hgroups,
-                              [item[0] for item in JOJ_INFO["problemInfo"]])
+                          [item[0] for item in JOJ_INFO["problemInfo"]])
     if args.indv:
         indvScores = gitWorker.checkIndv()
     if args.group:
@@ -78,4 +83,7 @@ if __name__ == "__main__":
         if args.upload:
             canvasWorker.grade2Canvas()
     if args.proj:
-        gitWorker.checkProj(args.proj, args.ms)
+        projScores = gitWorker.checkProj(args.proj, args.ms)
+        giteaWorker = GiteaWorker(args, GITEA_BASE_URL, ORG_NAME, GITEA_TOKEN,
+                                  hgroups)
+        giteaWorker.raiseIssues(projScores)
-- 
cgit v1.2.3