dailyFudan.py

from os import path as os_path
import os
os.environ['OPENSSL_CONF'] = os_path.join(os.getcwd(), 'openssl.cnf')
import time
from json import loads as json_loads
from json import dumps as json_dumps
from sys import exit as sys_exit
from sys import argv as sys_argv
import traceback

from py_sha2 import sha256

from lxml import etree
from requests import session
import logging
logging.basicConfig(level=logging.INFO,
                    format='%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s')

# WeChat notice
#get token via http://iyuu.cn/
import requests
from captcha_break import DailyFDCaptcha
from geo_disturbance import geoDisturbance
from requests.packages.urllib3.exceptions import InsecureRequestWarning
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
def iyuu(IYUU_TOKEN):
    url = f"https://iyuu.cn/{IYUU_TOKEN}.send"
    headers = {'Content-type': 'application/x-www-form-urlencoded'}
    def send(text, desp=""):
        desp = f'该脚本打卡会强制修改为在校状态，非在校同学切勿使用！！！\n{desp}'
        Form = {'text': text, 'desp': desp}
        return requests.post(url, data=Form, headers=headers, verify=False)
    return send

from ServerChan import ftqq
import UIS_Captcha

# fix random area bug
def set_q(iterO):
    res = list()
    for item in iterO:
        if item not in res:
            res.append(item)
    return res



class Fudan:
    """
    建立与复旦服务器的会话，执行登录/登出操作
    """
    UA = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36 Edg/107.0.1418.62"

    # 初始化会话
    def __init__(self,
                 uid, psw,
                 url_login='https://uis.fudan.edu.cn/authserver/login'):
        """
        初始化一个session，及登录信息
        :param uid: 学号
        :param psw: 密码
        :param url_login: 登录页，默认服务为空
        """
        self.session = session()
        self.session.headers['User-Agent'] = self.UA
        self.session.verify = False
        self.url_login = url_login

        self.uid = uid
        self.psw = psw

        self.headers = {
            "Accept-Encoding": "gzip, deflate, br",
            "Accept-Language": "zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6",
            "Cache-Control": "no-cache",
            "Connection": "keep-alive",
            "DNT": "1",
            "Host": "uis.fudan.edu.cn",
            "Origin": "https://uis.fudan.edu.cn",
            "Pragma": "no-cache",
            "sec-ch-ua": '"Microsoft Edge";v="107", "Chromium";v="107", "Not=A?Brand";v="24"',
            "sec-ch-ua-mobile": "?0",
            "sec-ch-ua-platform": '"Windows"',
            "Sec-Fetch-Dest": "empty",
            "Sec-Fetch-Mode": "cors",
            "Sec-Fetch-Site": "same-origin",
            "X-Requested-With": "XMLHttpRequest"
        }
        self.headers.update({
            "Referer"   : self.url_login,
            "User-Agent": self.UA
            
        })

    def _page_init(self):
        """
        检查是否能打开登录页面
        :return: 登录页page source
        """
        logging.debug("Initiating——")
        page_login = self.session.get(self.url_login)

        logging.debug("return status code " + str(page_login.status_code))

        if page_login.status_code == 200:
            logging.debug("Initiated——")
            return page_login.text
        else:
            logging.debug("Fail to open Login Page, Check your Internet connection\n")
            self.close()

    def login(self):
        """
        执行登录
        """
        page_login = self._page_init()

        logging.debug("parsing Login page——")
        html = etree.HTML(page_login, etree.HTMLParser())

        logging.debug("getting tokens")
        data = {
            "username": self.uid,
            "password": self.psw,
            "service" : "https://zlapp.fudan.edu.cn/site/ncov/fudanDaily"
        }

        # 获取登录页上的令牌
        data.update(
                zip(
                        html.xpath("/html/body/form/input/@name"),
                        html.xpath("/html/body/form/input/@value")
                )
        )
        if UIS_Captcha.needCaptcha(self.session, self.headers, self.uid):
            from captcha_break_dddd import dddd_3
            for i in range(6):
                tmp_res = UIS_Captcha.captcha(self.session, self.headers)
                tmp_res = dddd_3(tmp_res)
                data.update({
                    'captchaResponse': tmp_res
                    })
                logging.debug("Login ing—— captcha")
                post = self.session.post(
                self.url_login,
                data=data,
                headers=self.headers,
                allow_redirects=False)
                logging.debug("return status code %d" % post.status_code)
                if post.status_code == 302:
                    logging.debug("登录成功")
                    return True
                elif post.status_code == 200:
                    logging.debug("验证码错误")
                else:
                    logging.debug("登录失败，请检查账号信息")
                    self.close()
                    return False
            logging.debug("验证码错误次数过多，请更换识别模块")
            self.close()
            return False
                
        logging.debug("Login ing——")
        post = self.session.post(
                self.url_login,
                data=data,
                headers=self.headers,
                allow_redirects=False)

        logging.debug("return status code %d" % post.status_code)

        if post.status_code == 302:
            logging.debug("登录成功")
            return True
        else:
            logging.debug("登录失败，请检查账号信息")
            self.close()
            return False

    def logout(self):
        """
        执行登出
        """
        exit_url = 'https://uis.fudan.edu.cn/authserver/logout?service=/authserver/login'
        expire = self.session.get(exit_url).headers.get('Set-Cookie')

        if '01-Jan-1970' in expire:
            logging.debug("登出完毕")
        else:
            logging.debug("登出异常")

    def close(self):
        """
        执行登出并关闭会话
        """
        self.logout()
        self.session.close()
        logging.debug("关闭会话")
        # sys_exit() 通知完成再退出 注释掉

class Zlapp(Fudan):
    last_info = ''

    def check(self):
        """
        检查
        """
        logging.debug("检测是否已提交")
        get_info = self.session.get(
                'https://zlapp.fudan.edu.cn/ncov/wap/fudan/get-info')
        last_info = get_info.json()

        logging.info("上一次提交日期为: %s " % last_info["d"]["info"]["date"])

        position = last_info["d"]["info"]['geo_api_info']
        position = json_loads(position)

        if s_sfzx.__name__ == '<lambda>':
            logging.info("上一次提交地址为: %s" % position['formattedAddress'])
        else:
            logging.info("上一次提交地址为: ***" )
        # logging.debug("上一次提交GPS为", position["position"])

        today = time.strftime("%Y%m%d", time.localtime())

        if last_info["d"]["info"]["date"] == today:
            logging.info("今日已提交")
            self.close()
            global gl_info
            gl_info = last_info
            geo_api_info = json_loads(last_info["d"]["info"]["geo_api_info"])
            province = geo_api_info["addressComponent"].get("province", "")
            city = geo_api_info["addressComponent"].get("city", "") or province
            district = geo_api_info["addressComponent"].get("district", "")
            gl_info['dailyFudan'] = " ".join(set_q((province, city, district)))
            gl_info['geoDisturbance'] = geoDisturbance(last_info["d"]["info"]["geo_api_info"])
            gl_info = json_dumps(gl_info, indent=4, ensure_ascii=False)
            return True
        else:
            logging.info("未提交")
            self.last_info = last_info["d"]["info"]
            self.old_info = last_info["d"]["oldInfo"]
            return False

    def checkin(self, captcha):
        """
        提交
        """
        headers = {
            "Host"      : "zlapp.fudan.edu.cn",
            "Referer"   : "https://zlapp.fudan.edu.cn/site/ncov/fudanDaily?from=history",
            "DNT"       : "1",
            "TE"        : "Trailers",
            "User-Agent": self.UA
        }

        logging.debug("提交中")

        geo_api_info = json_loads(self.last_info["geo_api_info"])
        province = geo_api_info["addressComponent"].get("province", "")
        city = geo_api_info["addressComponent"].get("city", "") or province
        district = geo_api_info["addressComponent"].get("district", "")
        self.last_info.update(
                {
                    "tw"      : "13",
                    "province": province,
                    "city"    : city,
                    "area"    : " ".join(set_q((province, city, district))),
                    "ismoved" : 0,
                    "geo_api_info" : geoDisturbance(self.last_info["geo_api_info"])
                }
        )
        # logging.debug(self.last_info)
        for i in range(3):
            captcha_text = captcha()
            #captcha_text = 'abcd'
            self.last_info.update({
                'sfzx': s_sfzx(self.old_info),
                'code': captcha_text
            })
            save = self.session.post(
                    'https://zlapp.fudan.edu.cn/ncov/wap/fudan/save',
                    data=self.last_info,
                    headers=headers,
                    allow_redirects=False)
            logging.info(save.text)
            save_msg = json_loads(save.text)["m"]
            if save_msg != '验证码错误':
                break
            else:
                captcha.reportError()
                print('captcha.reportError')

def f_decode(lc_psw):
    if lc_psw.startswith('|||base64|||'):
        from base64 import b64decode
        lc_psw = lc_psw[12:]
        lc_psw = b64decode(lc_psw)
        lc_psw = lc_psw.decode(encoding = 'utf-8')
    return lc_psw


def get_account():
    """
    获取账号信息
    """
    uid, psw, *IYUU_TOKEN = sys_argv[1].strip().split(' ')
    global Check_value
    Check_value = False
    if len(IYUU_TOKEN) == 4:
        # https://tool.oschina.net/encrypt?type=2
        Check_value = IYUU_TOKEN.pop()
        statement = f'{uid}认同平安复旦对抗疫的重要意义，将自觉遵守防疫政策；{uid}仅在长期停留原处时使用本代码以减少不必要的劳动；{uid}如有出行，将立即手动更新自己的位置信息；如出现任何违反防疫政策的行为，{uid}同意自己承担全部责任。'
        Check_value = (sha256(statement, True) == Check_value)
    return uid, psw, IYUU_TOKEN

gl_info = "快去手动填写！"
if __name__ == '__main__':
    if True:
        print("平安复旦已结束")
        sys_exit(0)
    uid, psw, IYUU_TOKE = get_account()
    # print(f'Check_value: {Check_value}')
    if Check_value:
        def s_sfzx(last_info):
            print(f"last_info sfzx {last_info.get('sfzx')}")
            return last_info.get('sfzx', "1")
    else:
        s_sfzx = lambda x: "1"
    if IYUU_TOKE: #有token则通知，无token不通知
        if len(IYUU_TOKE) != 3:
            logging.error("请正确配置微信通知功能和验证码打码功能～\n")
            sys_exit(1)
        uname = IYUU_TOKE[1]
        pwd = IYUU_TOKE[2]
        IYUU_TOKE = IYUU_TOKE[0]
        if IYUU_TOKE.startswith('IYUU'):
            iy_info = iyuu(IYUU_TOKE)
        elif IYUU_TOKE.startswith('SCT'):
            iy_info = ftqq(IYUU_TOKE)
        else:
            def iy_info(text, desp=""):
                pass
    else:
        def iy_info(text, desp=""):
            pass
        logging.error("请按readme操作，以正确完成配置～\n")
        sys_exit(1)
    psw = f_decode(psw)
    pwd = f_decode(pwd)

    if (True):
        try:
            from FDU_daily_fudan import dailyFudan
            suc = dailyFudan(uid, psw, uname, pwd, iy_info, s_sfzx)
        except:
            suc = False
            print(traceback.format_exc())

        if suc:
            sys_exit()

    # logging.debug("ACCOUNT：" + uid + psw)
    zlapp_login = 'https://uis.fudan.edu.cn/authserver/login?' \
                  'service=https://zlapp.fudan.edu.cn/site/ncov/fudanDaily'
    daily_fudan = Zlapp(uid, psw, url_login=zlapp_login)
    if not daily_fudan.login():
        iy_info("平安复旦：登陆失败", gl_info)
        sys_exit()

    if daily_fudan.check():
        iy_info("平安复旦：今日已提交", gl_info)
        sys_exit()

    def captcha_info(message):
        iy_info(message, gl_info)
    captcha = DailyFDCaptcha(uname,pwd,daily_fudan,captcha_info)
    daily_fudan.checkin(captcha)

    # 再检查一遍
    if daily_fudan.check():
        iy_info("平安复旦：今日已提交", gl_info)
    else:
        iy_info("平安复旦：本次提交失败", gl_info)

    daily_fudan.close()
    sys_exit()