aoxiangRobot

#!/usr/bin/env python3
# -*- coding: utf-8 -*-


import argh
from argh import arg


@arg('-t', '--term', help='terms, for example: 19', nargs='+')
def grade(**kwargs):
    """
    获取考试成绩
    """
    from src.grade import Grade

    username = kwargs['username']
    password = open(kwargs['password']).read().strip()

    g = Grade(username=username, password=password)

    if kwargs['term']:
        g.output(*kwargs['term'])
    else:
        g.output()


@arg('-t', '--term', help='terms, for example: 19', nargs='+', required=True)
def exam(**kwargs):
    """
    获取考试安排
    """
    from src.exam import Exam

    username = kwargs['username']
    password = open(kwargs['password']).read().strip()

    e = Exam(username=username, password=password)
    e.output(*kwargs['term'])


@arg('-t', '--term', help='terms, for example: 19', required=True)
def courses(**kwargs):
    """
    获取我的课程信息
    """
    from src.classTable import ClassTable

    username = kwargs['username']
    password = open(kwargs['password']).read().strip()

    e = ClassTable(username=username, password=password)
    e.output(kwargs['term'])


def me(**kwargs):
    """
    获取当前账号信息
    """
    from rich.console import Console
    from rich.table import Table
    from src.netreq.aoxiang import Aoxiang

    username = kwargs['username']
    password = open(kwargs['password']).read().strip()

    console = Console()

    res = Aoxiang(username=username, password=password).userInfo

    for i in res:
        table = Table(title=i, show_header=False, width=120)

        info = list(res[i].items())

        if len(info) % 2:
            info.append(('', ''))

        table.add_column('', width=20)
        table.add_column('', width=40)
        table.add_column('', width=20)
        table.add_column('', width=40)

        for x in range(0, len(info), 2):
            table.add_row(*(info[x] + info[x + 1]))
        table.row_styles = ['none', 'dim']

        console.print(table)


@arg('-s', '--start', help='time start, for example: 2020-9-1, today by default')
@arg('-e', '--end', help='time end, for example: 2020-12-31, today + 180 days by default')
@arg('-o', '--output', help='output classtable to the file in `ics` format', required=True)
@arg('-a', '--alarm', help='alarm before class, default: 20, set 0 to disable alarm', default=20)
def classTable(**kwargs):
    """
    获取课表
    """
    import os
    from src import log
    from src.classTable import ClassTable

    if os.path.exists(kwargs['output']):
        if not log.yes_or_no(f'{kwargs["output"]} exists, overwrite it?'):
            log.e('Abort.')

    username = kwargs['username']
    password = open(kwargs['password']).read().strip()

    ct = ClassTable(username=username, password=password)
    ct.export(kwargs['start'], kwargs['end'], **kwargs)


@arg('-p', '--province', default=None, help='对应疫情填报中的省，在学校/在西安可不填')
@arg('-c', '--city', default=None, help='对应疫情填报中的市，直辖市填市辖区或县，在学校/在西安可不填')
@arg('-l', '--location', default='在学校', help='对应疫情填报中的当前位置（在学校/在西安）或区/县字段，查看所有允许的位置，看：`http://yqtb.nwpu.edu.cn/wx/js/eams.area.data.js`')
def yqtb(**kwargs):
    """
    疫情填报
    """
    from rich.console import Console
    from rich.table import Table

    from src.netreq.aoxiang import Aoxiang

    username = kwargs['username']
    password = open(kwargs['password']).read().strip()

    res = Aoxiang(username, password).yqtb(province=kwargs['province'], city=kwargs['city'], location=kwargs['location'])

    table = Table(title='Result')

    table.add_column('A')
    table.add_column('B')

    for i in res:
        table.add_row(i, res[i])
    table.row_styles = ['dim', 'none']

    Console().print(table)


if __name__ == '__main__':
    parser = argh.ArghParser()
    parser.add_argument('-u', '--username', required=True, help='username')
    parser.add_argument('-p', '--password', required=True, help='password FILE path')

    parser.add_commands([grade, exam, courses, classTable, me, yqtb])

    parser.dispatch()