dayReport.py

# encoding=utf-8
import argparse
import datetime
import json
import random
import pytz

from bs4 import BeautifulSoup
from login import login

# 加载全局配置
with open("./config.json", "r", encoding="utf-8") as f:
    configs = f.read()
configs = json.loads(configs)
isDebug = False


def load_params(ss, mode):
    '''合并填报参数'''
    json_form = get_report_data(ss)  # 获取昨日填报信息

    params = {
        "DZ_JSDTCJTW": 36.5,
        "DZ_DBRQ": "%Y-%m-%d",
        "CZRQ": "%Y-%m-%d %H:%M:%S",
        "CREATED_AT": "%Y-%m-%d %H:%M:%S",
        "NEED_CHECKIN_DATE": "%Y-%m-%d"
    }
    params["DZ_JSDTCJTW"] = 36 + random.randint(1, 10) / 10
    if mode != '':
        try:
            local = configs['dailyReport'][mode]
            params.update(local)
        except Exception:
            print("【加载本地配置失败，使用昨日信息进行填报】")
    else:
        print("【使用昨日信息进行填报】")

    # get time
    today = datetime.datetime.now()
    yesterday = today - datetime.timedelta(days=1)

    today_list = ['CZRQ', 'CREATED_AT', 'NEED_CHECKIN_DATE']
    yesterday_list = ['DZ_DBRQ']
    for key in params.keys():
        # 填充日期
        if key in yesterday_list:
            params[key] = yesterday.strftime(params[key])
        elif key in today_list:
            params[key] = today.strftime(params[key])
        json_form[key] = params[key]
    # print(params)
    return json_form


def doReport(session, mode=''):
    """
    session: 已登录的 requests.session 对象
    mode:  填报配置 home, school 等，(留空使用昨天的填报信息)，可自行修改 config.json 文件
    """
    # 进入填报页面（获取sessionid）
    session.get("http://ehall.seu.edu.cn/appShow?appId=5821102911870447")

    url = 'http://ehall.seu.edu.cn/qljfwapp2/sys/lwReportEpidemicSeu/mobile/dailyReport/T_REPORT_EPIDEMIC_CHECKIN_SAVE.do'

    json_form = load_params(session, mode)

    res = session.post(url, data=json_form)
    try:
        if json.loads(res.text)['datas']['T_REPORT_EPIDEMIC_CHECKIN_SAVE'] == 1:
            print("填报成功！")
        else:
            print("填报失败！")
    except Exception:
        soup = BeautifulSoup(res.text, "html.parser")
        tag = soup.select('.underscore.bh-mt-16')
        if len(tag) > 1:
            print(tag[0].text.replace('\n', ''))
        else:
            print(res.text)
        print("填报失败！")


# 获取昨日填报信息
def get_report_data(ss):
    # 进入填报页面（获取sessionid）
    ss.get("http://ehall.seu.edu.cn/appShow?appId=5821102911870447")
    latest_url = "http://ehall.seu.edu.cn/qljfwapp2/sys/lwReportEpidemicSeu/modules/dailyReport/getLatestDailyReportData.do"
    wid_url = "http://ehall.seu.edu.cn/qljfwapp2/sys/lwReportEpidemicSeu/mobile/dailyReport/getMyTodayReportWid.do"
    userinfo_url = "http://ehall.seu.edu.cn/qljfwapp2/sys/lwReportEpidemicSeu/api/base/getUserDetailDB.do"
    last_res = ss.get(latest_url)
    wid_res = ss.get(wid_url)
    userinfo_res = ss.post(userinfo_url)
    try:
        tempFormData = {}
        userInfo = json.loads(userinfo_res.text)['data']
        # 载入当天填报模板
        try:
            wid_data = json.loads(
                wid_res.text)['datas']['getMyTodayReportWid']['rows'][0]
            tempFormData.update(wid_data)
        except Exception:
            print('【getMyTodayReportWid FAILED】')
            # raise       # 经测试：wid为空无影响

        # 载入昨日填报信息
        try:
            last_report = json.loads(
                last_res.text)['datas']['getLatestDailyReportData']['rows'][0]
            tempFormData.update(last_report)
        except Exception:
            print('getLatestDailyReportData FAILED】')
            raise

        # 载入用户信息
        tempFormData['USER_ID'] = configs['user']['cardnum']
        # tempFormData['PHONE_NUMBER'] = userInfo['PHONE_NUMBER']
        tempFormData['IDCARD_NO'] = userInfo['IDENTITY_CREDENTIALS_NO']
        tempFormData['GENDER_CODE'] = userInfo['GENDER_CODE']

        # tempFormData['CLASS_CODE'] = userInfo['CLASS_CODE']
        # tempFormData['CLASS'] = userInfo['CLASS']
        tempFormData['RYSFLB'] = userInfo['RYSFLB']
        tempFormData['USER_NAME'] = userInfo['USER_NAME']
        tempFormData['DEPT_CODE'] = userInfo['DEPT_CODE']  # 学院编号
        tempFormData['DEPT_NAME'] = userInfo['DEPT_NAME']
        if isDebug:
            print(tempFormData)
    except Exception as e:
        print(e)
        print("【获取填报信息失败，请手动填报】")
        exit()
    return tempFormData


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Test for argparse')
    parser.add_argument(
        '--config', '-c', help='采用的配置名称 如 school, home', default='')
    parser.add_argument(
        '--force', '-f', help='15:00 后是否仍然填报', action='store_true')
    parser.add_argument(
        '--debug', '-d', help='显示调试信息', action='store_true')
    parser.add_argument(
        '--user', '-u', help='一卡通号', default='')
    parser.add_argument(
        '--password', '-p', help='密码', default='')

    args = parser.parse_args()
    isDebug = args.debug

    # 覆盖账号信息
    if args.user != '' and args.password != '':
        configs['user']['cardnum'] = args.user
        configs['user']['password'] = args.password

    today = datetime.datetime.now(pytz.timezone('Asia/Shanghai'))
    print(today)
    if today.hour >= 15:  # 超过填报时间
        if args.force:
            print("【超过填报时间，但继续填报】")
        else:
            print("【超过填报时间！放弃填报】")
            exit()
    ss = login(configs['user']['cardnum'], configs['user']['password'])
    if ss:
        doReport(ss, args.config)