妈妈再也不用担心我忘记填报疫情——巧用云函数完成自动化数据填报

class NWPU_Yqtb_Site(object):
    def __init__(self):
        # 在 `NWPU_Yqtb_Site` 实例中，首先建立一个 session 并对登录页发起 get：
        self.session = requests.Session()
        # 从 CAS 登录页作为登录过程起点
        self.session.get(url_cas_login)
        # 供填报时使用的数据
        self.name = ""
        self.xymc = ""
        self.xssjhm = ""
        self.szcsbm = ""
        self.szcsmc = ""
        self.hsjc = ""
        self.sign = ""
        self.timeStamp = ""
        self.data_for_submit = None

header = {
    'referer': url_cas_login,
    'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_16_0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/85.0.4183.26 Safari/537.36',
    'content-Type': 'application/x-www-form-urlencoded',
    'origin': 'https://uis.nwpu.edu.cn',
    'cookie': 'SESSION=' + str((session.cookies.values()[0])),
    # [0] 是生成的 `SESSION`
    'authority': 'uis.nwpu.edu.cn',
    'path': '/cas/login',
    'scheme': 'https',
    'accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9',
    'upgrade-insecure-requests': '1',
    'sec-fetch-dest': 'document',
    'sec-fetch-mode': 'navigate',
    'sec-fetch-site': 'same-origin',
    'sec-fetch-user': '?1',
    'cache-control': 'no-cache'
}
data = {
    'username': username,
    'password': password,
    '_eventId': 'submit',
    'currentMenu': '1',
    'execution': 'e2s1',
    'submit': 'One moment please...',
    'geolocation': '',
}

res_login = self.session.post(url_cas_login, data=data_for_login, headers=header_for_login)
if (res_login.text.find('欢迎使用')) != -1:
    print('登陆成功！')
else:
    print('登录失败！请检查「登录信息」一栏用户名及密码是否正确')
    if user_config.SC_switcher == 1:
        Pusher.sc_push_when_login_failed(self)
    exit()

header_for_submit = {
    "Host": "yqtb.nwpu.edu.cn",
    "Origin": "http://yqtb.nwpu.edu.cn",
    "Referer": "http://yqtb.nwpu.edu.cn/wx/ry/jrsb.jsp",
    "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8",
    "Cookie": "JSESSIONID=" + str((self.session.cookies.values()[2])),
    "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_16_0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/85.0.4183.26 Safari/537.36",
}
self.data_for_submit = {
    'actionType': 'addRbxx',
    'userLoginId': self.username,
    'sfjt': '0',
    'sfjcry': '0',
    'sfjcqz': '0',
    'sfjkqk': '0',
    'sfyzz': '0',
    'sfqz': '0',
    'glqk': '0',
    'tbly': 'sso',
    'userType': '2',
    'userName': self.name,
    'bdzt': '1',
    'xymc': self.xymc,
    'xssjhm': self.xssjhm,
    'szcsmc': self.szcsmc,
    'szcsbm': self.szcsbm,
    'hsjc': self.hsjc,
}

HeadersForm = {...}
tbDataForm = {...}

# `self.judge_last_report_is_today(
#       self.get_last_report_time(data_for_init, header_for_init))` 的意义：
# 判断系统当前日期是否与最后一次填报的日期一致。
# 如条件为 `False`，则不一致，认为今日未填报，执行 POST；
# 否则跳过 POST 步骤，不再重复填报。
if not (self.judge_last_report_is_today(self.get_last_report_time(data_for_init, header_for_init))):
    url_ry_util_with_token = url_ry_util + '?sign=' + self.sign + '&timeStamp=' + self.timeStamp
    self.session.post(url=url_ry_util_with_token,
                        data=self.data_for_submit,
                        headers=header_for_submit)
else:
    print('今日已填报，无需重复填报！')
    return

# 判断填报成功
if (self.judge_last_report_is_today(self.get_last_report_time(data_for_init, header_for_init))):
    print('申报成功！')
    if user_config.SC_switcher == 1:
        Pusher.scPush(self)
else:
    print('申报失败，请重试！')
    if user_config.SC_switcher == 1:
        Pusher.sc_push_when_wrong_info(self)


def judge_last_report_is_today(self, report_time):
    report_time = self.string_toDatetime(report_time)
    if report_time.date() == datetime.today().date():
        return True
    else:
        return False

# 获取最近一次填报的时间，判断是否为今天
def get_last_report_time(self, data, header):
    last_report_time = \
        etree.HTML(self.get_last_report(data, header)).xpath(
            '/html/body/div[1]/div[2]/div/div[2]/div[1]/div[2]/text()')[0]
    return last_report_time