get_sport_measure_score.py

# Copyright (C) 2022 by the XiDian Open Source Community.
#
# This file is part of xidian-scripts.
#
# xidian-scripts is free software: you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by the
# Free Software Foundation, either version 3 of the License, or (at your
# option) any later version.
#
# xidian-scripts is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
# for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with xidian-scripts.  If not, see <http://www.gnu.org/licenses/>.
#

import pkg_resources
import subprocess
import sys
import os
try:
    pkg_resources.require(('libxduauth'))
except (pkg_resources.DistributionNotFound, pkg_resources.VersionConflict):
    subprocess.check_call([
        sys.executable, '-m', 'pip', 'install', 'libxduauth'
    ])

try:
    import credentials
    USERNAME = credentials.SPORTS_USERNAME
    PASSWORD = credentials.SPORTS_PASSWORD
except ImportError:
    USERNAME, PASSWORD = [os.getenv(i) for i in ('SPORTS_USER', 'SPORTS_PASS')]
if not USERNAME or not PASSWORD:
    print('请设置环境变量 SPORTS_USER 和 SPORTS_PASS')
    exit(1)

from libxduauth import SportsSession
# For a nice output.
from prettytable import PrettyTable


def get_sport_measure_detail(session, measure_score_id):
    form = PrettyTable()
    form.field_names = ["项目", "结果", "单位", "分数"]
    response = session.post(session.BASE_URL + 'measure/getStuScoreDetail',
                            data={
                                "meaScoreId": measure_score_id
                            }).json()
    for each in response['data']:
        if 'actualScore' in each:
            form.add_row([each["examName"], each["actualScore"],
                         each["examunit"], each["score"]])
        else:
            form.add_row([each["examName"], "未录入", each["examunit"], "未录入"])
    form.align = "l"
    print(form)


def get_sport_measure_general(session):
    response = session.post(session.BASE_URL + 'measure/getStuTotalScore',
                            data={
                                "userId": session.user_id
                            }).json()
    for i in response['data']:
        if 'meaScoreId' in i:
            print("{} 年度的成绩是 {} ，等级是{}。详情如下：".format(
                i['year'], i['totalScore'], i['rank']))
            get_sport_measure_detail(session, i["meaScoreId"])
        else:
            print("目前四次体测成绩总分是 {}".format(i['totalScore']))


if __name__ == '__main__':
    ses = SportsSession(USERNAME, PASSWORD)
    get_sport_measure_general(ses)