pytyut.py

"""
@FILE_NAME : pytyut
-*- coding : utf-8 -*-
@Author : Zhaokugua
@Time : 2022/2/23 0:40
@Version V0.7 beta
"""
import requests
import re
import json


class Pytyut:
    default_headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) '
                      'Chrome/90.0.4430.212 Safari/537.36 Edg/90.0.818.66',
    }
    req_headers_add = {}  # 设置全局请求头（用于特殊节点的请求验证）
    node_link = None   # 设置默认节点
    login_pub_key = '''-----BEGIN PUBLIC KEY-----
    MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCoZG+2JfvUXe2P19IJfjH+iLmp
    VSBX7ErSKnN2rx40EekJ4HEmQpa+vZ76PkHa+5b8L5eTHmT4gFVSukaqwoDjVAVR
    TufRBzy0ghfFUMfOZ8WluH42luJlEtbv9/dMqixikUrd3H7llf79QIb3gRhIIZT8
    TcpN6LUbX8noVcBKuwIDAQAB
    -----END PUBLIC KEY-----
        '''

    def __init__(self, uid, pwd):
        """
        初始化用户信息
        :param uid: 学号
        :param pwd: 教务系统的密码
        """
        self.uid = uid
        self.__pwd = pwd
        self.session = None   # 初始化session

    def login(self, debug=False):
        """
        :param debug:是否打印调试信息
        登录教务系统
        :return: dict 成功info返回真实姓名 失败返回失败原因
        """
        if not self.node_link:
            print('未选择登录节点！') if debug else ''
            return None
        self.session = requests.Session()
        login_url = self.node_link + 'Login/CheckLogin'
        # 创建会话，获取ASP.NET_SessionId的Cookies
        self.default_headers.update(self.req_headers_add)
        self.session.get(self.node_link, headers=self.default_headers)
        login_data = {
            'username': self.__RSA_uid(self.uid),
            'password': self.__pwd,
            'code': '',
            'isautologin': 0,
        }
        headers_check_login = {
            'Accept': 'application / json, text / javascript, * / *; q = 0.01',
            'X-Requested-With': 'XMLHttpRequest',
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
                          'AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.101 Safari/537.36 Edg/91.0.864.48',
            'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
            'Origin': self.node_link,
            'Referer': self.node_link,
            'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6',
        }
        headers_check_login.update(self.req_headers_add)
        login_res = self.session.post(url=login_url, data=login_data, headers=headers_check_login)
        if '登录成功' in login_res.text:
            print(login_res.json()['message'][:-1], end='：') if debug else ''
            home_url = self.node_link + '/Home/Default'
            home_res = self.session.get(url=home_url, headers=self.default_headers).text
            html = home_res.replace(' ', '').replace('\n', '').replace('\t', '').replace('\r', '')
            name_pattern = '<small>Welcome,</small>([^*]*)</span><ic'
            self.real_name = re.search(name_pattern, html, ).group(1)
            print(self.real_name) if debug else ''
            return {'http_code': login_res.status_code, 'msg': '登录成功', 'info': self.real_name}
        else:
            print('登录失败：', end='') if debug else ''
            try:
                error_info = login_res.json()['message']
            except:
                error_info = '页面信息Json解码失败！源代码：\n' + login_res.text if login_res.text else '无源代码'
            print(error_info) if debug else ''
            return {'http_code': login_res.status_code, 'msg': '登录失败', 'info': error_info}

    @classmethod
    def auto_node_chose(cls, debug=False):
        """
        自动确认登录节点
        :param debug:是否打印调试信息
        :return:登录节点的链接，如：http://jxgl1.tyut.edu.cn/
        """
        test_url = 'http://jxgl1.tyut.edu.cn/'
        print("正在测试最快速的连接，请稍后...")if debug else ''
        try:
            print("节点1...", end='')if debug else ''
            req = requests.get(test_url, timeout=3, headers=cls.default_headers)
            print(req.elapsed.microseconds/1000 + req.elapsed.seconds*100, 'ms')if debug else ''
        except:
            print('超时')if debug else ''
            print("节点2...", end='')if debug else ''
            test_url = 'http://jxgl2.tyut.edu.cn/'
            try:
                req = requests.get(test_url, timeout=3, headers=cls.default_headers)
                print(req.elapsed.microseconds/1000 + req.elapsed.seconds*100, 'ms')if debug else ''
            except:
                print('超时')if debug else ''
                print("节点3...", end='')if debug else ''
                test_url = 'https://jxgl20201105.tyutmate.cn/'
                headers = {
                    'User-Agent': 'Mozilla/5.0 (Linux; Android 11; Redmi K30 Pro Build/RKQ1.200826.002; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/86.0.4240.99 XWEB/3171 MMWEBSDK/20211001 Mobile Safari/537.36 MMWEBID/6338 MicroMessenger/8.0.16.2040(0x2800105D) Process/toolsmp WeChat/arm64 Weixin NetType/WIFI Language/zh_CN ABI/arm64 miniProgram Edg/96.0.4664.110',
                    'Sec-Fetch-Site': 'same-site',
                    'Sec-Fetch-Mode': 'navigate',
                    'Sec-Fetch-User': '?1',
                    'Sec-Fetch-Dest': 'document',
                    'Referer': 'https://helper.tyutmate.cn/tyut/index.html?random=fuckyo',
                }

                try:
                    req = requests.get(test_url, timeout=10, headers=headers)
                    if req.status_code != 200:
                        print("服务器错误！请避免高峰期访问！", req.status_code)
                        return None
                    print(req.elapsed.microseconds / 1000 + req.elapsed.seconds * 1000, 'ms')if debug else ''
                    cls.req_headers_add = headers
                except:
                    print("所有节点均无响应，请检查网络！")
                    return None

        print("选择到节点：", test_url)if debug else ''
        return test_url

    @classmethod
    # RSA 公钥加密，用于登录教务系统处理用户名信息
    def __RSA_uid(cls, uid):
        import base64
        # 这里有可能出现导包导不进去的问题，把site-packages里面的crypto文件夹改为大写Crypto即可
        # 需要安装pycryptodome
        from Crypto.PublicKey import RSA
        from Crypto.Cipher import PKCS1_v1_5
        # 公钥在页面里面
        pub_key = cls.login_pub_key
        rsakey = RSA.importKey(pub_key)
        cipher = PKCS1_v1_5.new(rsakey)
        cipher_text = base64.b64encode(cipher.encrypt(uid.encode(encoding='utf-8')))
        value = cipher_text.decode('utf-8')
        return value

    def get_class_schedule(self):
        """
        获取自己的课表信息
        :return: 返回课表json信息
        """
        if not self.session:
            print('未登录')
            return None
        res = self.session.post(self.node_link + 'Tresources/A1Xskb/GetXsKb', headers=self.default_headers)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        return res.json()

    def get_class_scores(self):
        """
        获取课程成绩
        :return: 返回课程成绩json信息
        """
        if not self.session:
            print('未登录')
            return None
        req_data = {
            'order': 'zxjxjhh desc,kch',
        }
        req_url = self.node_link + 'Tschedule/C6Cjgl/GetKccjResult'
        res = self.session.post(req_url, headers=self.default_headers, data=req_data)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        # 正则匹配学年学期，按照学年学期分开每一个片段
        time_list = re.findall(r'\d{4}-\d{4}学年[\u4e00-\u9fa5]季', res.text)
        score_dict_list = []
        for i in range(len(time_list)):
            if i < len(time_list) - 1:
                html_part = res.text[res.text.find(time_list[i]): res.text.find(time_list[i + 1])]
            else:
                html_part = res.text[res.text.find(time_list[i]):]
            info_list = re.findall(r'tyle="vertical-align:middle; ">([^^]*?)</td>', html_part)
            for j in range(len(info_list) // 9):
                score_dict = {
                    'Xnxq': time_list[i],
                    'Kch': info_list[9 * j],
                    'Kxh': info_list[9 * j + 1],
                    'Kcm': info_list[9 * j + 2],
                    'Kcm_en': info_list[9 * j + 3],
                    'Xf': info_list[9 * j + 4],
                    'Kcsx': info_list[9 * j + 5],
                    'Kssj': info_list[9 * j + 6],
                    'Cj': info_list[9 * j + 7],
                    'Failed_reason': info_list[9 * j + 8],
                }
                score_dict_list.append(score_dict)
        return score_dict_list

    def get_test_info(self, xnxq, bydesk=False):
        """
        获取考试安排信息
        byDesk=True时，xnxq可以传任意参数，不受影响。
        :param bydesk: 使用教务系统主页的简化接口
        :param xnxq: 学年学期，如'2020-2021-1-1'表示2020-2021学年第一学期，同理'2020-2021-2-1'为第二学期
        :return:返回考试安排的json信息
        """
        if not self.session:
            print('未登录')
            return None
        if bydesk:
            req_url = self.node_link + 'Tschedule/C5KwBkks/GetKsxxByDesk'
            data = {
                'pagination[limit]': 15,
                'pagination[offset]': 1,
                'pagination[sort]': 'ksrq',
                'pagination[order]': 'asc',
                'pagination[conditionJson]': '{}',
            }
            res = self.session.post(req_url, headers=self.default_headers, data=data)
            if '出错' in res.text or '教学管理服务平台(S)' in res.text:
                print('登录失效！')
                return None
            html_text = json.loads(res.text)["rpath"]["m_StringValue"]
            html_text = html_text.replace('<font style="color: #ff0000">', '').replace('</font>', '')
            param = '''<tr><td height='20%' width='10%' style="vertical-align:middle; ">([^<]*?)</td><td height='20%' width='25%' style="vertical-align:middle; ">([^<]*?)</td><td height='20%' width='37%' style="vertical-align:middle; ">([^<]*?)</td> <td height='20%' width='30%' style="vertical-align:middle; ">([^<]*?)</td></tr>'''
            info_list = re.findall(param, html_text)
            return info_list

        req_url = self.node_link + 'Tschedule/C5KwBkks/GetKsxxByXhListPage'
        data = {
            'limit': 30,
            'offset': 0,
            'sort': 'ksrq',
            'order': 'desc',
            'conditionJson': '{"zxjxjhh":"' + xnxq + '"}'
        }
        res = self.session.post(req_url, headers=self.default_headers, data=data)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        return res.json()

    def get_major_class_tree(self, xnxq):
        """
        获取历届学院专业班级树的Json信息
        :param xnxq: 学年学期
        :return: list 返回历届学院专业班级树的json信息
        """
        if not self.session:
            print('未登录')
            return None
        data = {
            'zxjxjhh': xnxq,
        }
        req_url = self.node_link + 'Tschedule/Zhcx/GetNjxszyTreeByrwbjJson'
        res = self.session.post(req_url, data=data,headers=self.default_headers)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        return res.json()

    def get_class_schedule_by_bjh(self, xnxq, bjh):
        """
        获取历届学院专业班级树的Json信息
        :param bjh:班级号，专业班级简称
        :param xnxq: 学年学期
        :return: dict 返回历届学院专业班级树的json信息
        """
        if not self.session:
            print('未登录')
            return None
        class_data = {
            'zxjxjhh': xnxq,
            'bjh': bjh,
        }
        data = {
            'pagination[conditionJson]': str(class_data),
            'pagination[sort]': 'xsh,kch',
            'pagination[order]': 'asc',
        }
        req_url = self.node_link + 'Tschedule/Zhcx/GetSjjsSjddByBjh'
        res = self.session.post(req_url, data=data,headers=self.default_headers)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        return res.json()

    def get_my_info(self):
        """
        获取自己的Json信息
        :return: dict 返回自己的json信息，头像采用二进制字节串存储。
        """
        if not self.session:
            print('未登录')
            return None
        req_url = self.node_link + 'Home/StudentResult'
        res = self.session.get(req_url, headers=self.default_headers)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        avatar = self.session.get(self.node_link + 'Tresources/AXsgj/ZpResultXs', headers=self.default_headers).content
        param = '<div class="profile-info-name">([^^]*?)</div>[^^]*?<[^^]*?>(.*)</[^^]*?>'
        result = re.findall(param, res.text)
        # 列表生成法，可能会慢一些
        result_list = [(name.replace('：', '').replace('\n', '').replace(' ', '').replace('\r', ''), value) for name, value in result]
        result_dict = dict(result_list)
        # 字符串替换法，会导致日期和英文姓名中的空格消失
        # result_dict = eval(str(dict(result)).replace('：', '').replace(r'\n', '').replace(' ', '').replace(r'\r', ''))
        result_dict['avatar'] = avatar
        print('喵')
        return result_dict

    def get_now_xnxq(self):
        # 自动获取当前的学年学期
        # 采用选课的接口，可能对于其他页面不太准确，谨慎使用
        # 返回'2021-2022-2-1'之类的字符串
        html_txt = self.session.get(self.node_link + f'Tschedule/C4Xkgl/XkXsxkIndex', headers=self.default_headers).text
        result = re.findall('<option selected="selected" value="([^^]*?)">([^^]*?)</option>', html_txt)[0][0]
        return result

    def get_xq_page_list(self, xnxq='auto'):
        """
        获取该学期选课列表
        如果没有可以选的课total就是0
        :param xnxq: 学年学期
        :return:dict 返回可以进行选课的科目列表（不是详情）
        """
        if not self.session:
            print('未登录')
            return None
        req_url = self.node_link + 'Tschedule/C4Xkgl/GetXkPageListJson'
        xnxq = self.get_now_xnxq() if xnxq == 'auto' else xnxq
        conditionJson = {
            "zxjxjhh": xnxq,
        }
        data = {
            'limit': 30,
            'offset': 0,
            'sort': 'xh',
            'order': 'asc',
            'conditionJson': str(conditionJson).replace(r'\'', '').replace('+', '')
        }
        res = self.session.post(req_url, data=data, headers=self.default_headers)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        return res.json()

    def get_xk_kc_list(self, pid, xnxq='auto'):
        """
        获取选课课程列表
        :param xnxq: 学年学期，不传值会自动获取
        :param pid:  选课列表中的Id，比如0e902576-56cb-4e4f-a15a-3dce0b10a0b7（实际上是pid）
        :return: list 返回
        """
        if not self.session:
            print('未登录')
            return None
        xnxq = self.get_now_xnxq() if xnxq == 'auto' else xnxq
        conditionJson = {
            'zxjxjhh': xnxq,
            'kch': '',
            'pid': pid,
        }
        data = {
            'limit': 500,
            'offset': 0,
            'sort': 'kch,kxh',
            'order': 'asc',
            'conditionJson': str(conditionJson).replace('\\', '').replace('+', '')
        }
        # 这个接口是有教学任务的课（比如体育课）
        req_url = self.node_link + 'Tschedule/C4Xkgl/GetXkkcListByXh'
        res = self.session.post(req_url, data=data, headers=self.default_headers)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        json_info = res.json()
        # 判断有没有用错接口
        if json_info['total'] == 0:
            # 这个接口是没有教学任务的课（比如学习通上的选修课）
            url2 = self.node_link + 'Tschedule/C4Xkgl/GetXkkcListWithoutJxrwByXh'
            res2 = self.session.post(url=url2, data=data, headers=self.default_headers)
            return res2.json()
        return res.json()

    def get_chosen_course_list(self):
        """
        获取已选择的课程列表
        如果没有已经选的课total就是0
        :return:dict 返回已经选课的科目列表（不是详情）
        """
        if not self.session:
            print('未登录')
            return None
        req_url = self.node_link + 'Tschedule/C4Xkgl/GetYxkcListByXhAndZxjxjhh'
        conditionJson = '{"kch":"","nopid":"zk"}'
        data = {
            'limit': 30,
            'offset': 0,
            'sort': 'kch,kxh',
            'order': 'asc',
            'conditionJson': str(conditionJson).replace(r'\'', '').replace('+', '')
        }
        res = self.session.post(req_url, data=data, headers=self.default_headers)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        return res.json()

    def choose_course(self, json_info):
        """
        选课
        :param json_info: 课程的json信息，如下所示：（注意，里面的数据一定不能传错，否则可能能选上课但是选的课的数据不对）
        class_json_info = {
            'xsxkList[0][Kch]': 'A0000111',                                  # 课程号
            'xsxkList[0][Kcm]': '中国近代人物研究',                             # 课程名
            'xsxkList[0][Kxh]': '01',                                        # 课序号
            'xsxkList[0][Pid]': '0e902576-56cb-4e4f-a15a-3dce0b10a0b7',      # 课程的pid
            'xsxkList[0][Bkskrl]': 6666,                                     # 课程总共人数，必须转换为整数
            'xsxkList[0][Xf]': '1',                                          # 学分
        }
        :return:dict 返回操作成功的json数据
        """
        if not self.session:
            print('未登录')
            return None
        headers_adds = {
            'Accept': 'application / json, text / javascript, * / *; q = 0.01',
            'X-Requested-With': 'XMLHttpRequest',
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
                          'AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.101 Safari/537.36 Edg/91.0.864.48',
            'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
            'Origin': self.node_link,
            'Referer': self.node_link,
            'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6',
        }
        self.default_headers.update(headers_adds)
        self.default_headers.update(self.req_headers_add)
        req_url = self.node_link + 'Tschedule/C4Xkgl/XsxkSaveForm'
        data = json_info
        data['logModel[Rolename]'] = ''
        data['logModel[Menuname]'] = '学生选课'
        data['isjwc'] = 'false'
        data['encryptedUsername'] = self.__RSA_uid(self.uid)
        # 访问一个伪装URL获取Set-Cookie
        url_text = self.node_link + 'Tschedule/C4Xkgl/XkXsxkIndex?yhm=R3m18OaKO5IALl6PvzDPz6qFtj8pRuCcSydIgHmMN2Ios2c5yfv/d2Aup8pgksJXWQZE/L8wbkZ9vrOSTFdX8jr5Jz4ewiSmYICXNSWViR/vPCgx8bHgcYY+VR4JKT5siATlDPehxIhI25pvgb36g0rba4/3wQQ4nKYUYpVR0+4=&mm=c237053a74zyc431a8a9a6a46748'
        res_test = self.session.get(url=url_text, headers=self.default_headers).text
        text = res_test.replace(' ', '').replace('\n', '').replace('\t', '').replace('\r', '')
        name_pattern = '<inputname="__RequestVerificationToken"type="hidden"value="([^*]*)"/>'
        data['__RequestVerificationToken'] = re.search(name_pattern, text, ).group(1)
        res = self.session.post(req_url, data=data, headers=self.default_headers)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        return res.json()

    def remove_course(self, json_info, course_Id):
        """
        退课
        :param json_info: 课程的json信息，如下所示：（注意，里面的数据一定不能传错，否则可能能选上课但是选的课的数据不对）
        class_json_info = {
            'xsxkList[0][Kch]': 'A0000111',                                  # 课程号
            'xsxkList[0][Kcm]': '中国近代人物研究',                             # 课程名
            'xsxkList[0][Kxh]': '01',                                        # 课序号
            'xsxkList[0][Pid]': '0e902576-56cb-4e4f-a15a-3dce0b10a0b7',      # 课程的pid
            'xsxkList[0][Bkskrl]': 6666,                                     # 课程总共人数，必须转换为整数
            'xsxkList[0][Xf]': '1',                                          # 学分
        }
        :param course_Id: 在self.get_chosen_course_list()方法返回的数据里面获取到的Id，注意不是pid，要区分
        :return:dict 返回退选成功的json数据
        """
        if not self.session:
            print('未登录')
            return None
        headers_adds = {
            'Accept': 'application / json, text / javascript, * / *; q = 0.01',
            'X-Requested-With': 'XMLHttpRequest',
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
                          'AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.101 Safari/537.36 Edg/91.0.864.48',
            'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
            'Origin': self.node_link,
            'Referer': self.node_link,
            'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6',
        }
        self.default_headers.update(headers_adds)
        self.default_headers.update(self.req_headers_add)
        req_url = self.node_link + 'Tschedule/C4Xkgl/XsxkRemoveForm'
        data = json_info
        data['xsxkList[0][Id]'] = course_Id
        data['xsxkList[0][Xh]'] = self.uid
        data['xsxkList[0][Xm]'] = self.real_name
        data['isjwc'] = 'false'
        data['encryptedUsername'] = self.__RSA_uid(self.uid)
        # 访问一个伪装URL获取Set-Cookie
        url_text = self.node_link + 'Tschedule/C4Xkgl/XkXsxkIndex?yhm=R3m18OaKO5IALl6PvzDPz6qFtj8pRuCcSydIgHmMN2Ios2c5yfv/d2Aup8pgksJXWQZE/L8wbkZ9vrOSTFdX8jr5Jz4ewiSmYICXNSWViR/vPCgx8bHgcYY+VR4JKT5siATlDPehxIhI25pvgb36g0rba4/3wQQ4nKYUYpVR0+4=&mm=c237053a74zyc431a8a9a6a46748'
        res_test = self.session.get(url=url_text, headers=self.default_headers).text
        text = res_test.replace(' ', '').replace('\n', '').replace('\t', '').replace('\r', '')
        name_pattern = '<inputname="__RequestVerificationToken"type="hidden"value="([^*]*)"/>'
        data['__RequestVerificationToken'] = re.search(name_pattern, text, ).group(1)
        res = self.session.post(req_url, data=data, headers=self.default_headers)
        if '出错' in res.text or '教学管理服务平台(S)' in res.text:
            print('登录失效！')
            return None
        return res.json()