#!/usr/bin/env python
# -*- coding: utf-8 -*-
import argparse
import json
import os

from worker import CanvasWorker, GiteaWorker, JOJWorker
from settings import *


def parse():
    parser = argparse.ArgumentParser(add_help=False)
    parser.add_argument('--help',
                        action='store_true',
                        help='show this help message and exit')
    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('-a', '--all', action='store_true', help='check all')
    parser.add_argument('-s',
                        '--score',
                        action='store_true',
                        help='generate score')
    parser.add_argument('-t', '--tidy', action='store_true', help='check tidy')
    parser.add_argument('-i',
                        '--indv',
                        action='store_true',
                        help='check indiviual submission')
    parser.add_argument('-g',
                        '--group',
                        action='store_true',
                        help='check group submission')
    parser.add_argument('-j',
                        '--joj',
                        action='store_true',
                        help='check joj score')
    parser.add_argument('-u',
                        '--upload',
                        action='store_true',
                        help='upload score to canvas')
    args = parser.parse_args()
    if args.help:
        parser.print_help()
        exit(0)
    if args.all:
        args.indv = True
        args.group = True
        args.joj = True
        args.tidy = True
        args.score = True
    return args


if __name__ == "__main__":
    hgroups = json.load(open("hgroups.json"))
    pwd = os.getcwd()
    args = parse()
    indvScores, groupScores, jojScores = None, None, None
    giteaWorker = GiteaWorker(args, hgroups,
                              [item[0] for item in JOJ_INFO["problemInfo"]])
    if args.indv:
        indvScores = giteaWorker.checkIndv()
    if args.group:
        groupScores = giteaWorker.checkGroup()
    if args.joj:
        jojWorker = JOJWorker(args, JOJ_COURSE_ID, SID, hgroups)
        jojScores = jojWorker.checkGroupJOJ(JOJ_INFO)
    if args.score:
        canvasWorker = CanvasWorker(args, RUBRIC, CANVAS_TOKEN, COURSE_ID,
                                    indvScores, groupScores, jojScores)
        canvasWorker.exportScores("scores.json")
        if args.upload:
            canvasWorker.grade2Canvas()
    if args.proj:
        giteaWorker.checkProj(args.proj, args.ms)