iSmartAuto/automaton/spider/spider.py

from hashlib import md5
from random import random

import httpx
from loguru import logger

from .captcha import recognize
from ..utils import Tree


class Spider(httpx.AsyncClient):
    def __init__(self):
        super().__init__()

    async def login(self, username, password):
        self.cookies.clear()  # 重置 cookies
        logger.info('正在获取验证码...')
        result = await self.get(f'http://sso.ismartlearning.cn/captcha.html?{random()}')
        code = recognize(result.content)
        token = md5(password.encode()).hexdigest()
        info = (await self.post(
            'http://sso.ismartlearning.cn/v2/tickets-v2',
            data={
                'username': username,
                'password': md5(token.encode() + b'fa&s*l%$k!fq$k!ld@fjlk').hexdigest(),  # 啥时候炸了就写成动态获取的
                'captcha': code
            },
            headers={
                'X-Requested-With': 'XMLHttpRequest',
                'Origin': 'http://me.ismartlearning.cn',
                'Referer': 'http://me.ismartlearning.cn/'
            }
        )).json()
        logger.debug(info['result'])

        assert info['result']['code'] == -26  # 断言登录结果
        return info['result']

    async def get_courses(self):  # 获取课程列表
        logger.info('正在获取课程列表...')
        courses = (await self.post(
            'https://school.ismartlearning.cn/client/course/list-of-student?status=1',
            data={
                'pager.currentPage': 1,
                'pager.pageSize': 32767
            }
        )).json()['data']
        return courses['list']

    async def get_books(self, course_id):  # 获取某课程的书籍列表
        logger.info('正在获取书籍列表...')
        await self.get_courses()  # 必须有这个请求，否则后面会报错
        books = (await self.post(
            'http://school.ismartlearning.cn/client/course/textbook/list-of-student',
            data={
                'courseId': course_id
            }
        )).json()['data']
        return books

    async def get_tasks(self, book_id, book_type, course_id):  # 获取某书籍的任务树
        logger.info('正在获取任务列表...')
        tasks = (await self.post(
            'http://school.ismartlearning.cn/client/course/textbook/chapters',
            data={
                'bookId': book_id,
                'bookType': book_type,
                'courseId': course_id
            }
        )).json()['data']
        id_record = {task['id']: Tree(task) for task in tasks}
        root = Tree({
            'book_id': tasks[0]['book_id'],
            'unitStudyPercent': 0
        })
        for task_id in id_record:
            node = id_record[task_id]
            node_name = (f'{node.task["name"]} ' if 'name' in node.task else '') + f'[id:{node.task["id"]}]'
            if 'parent_id' in node.task:
                if (parent_id := node.task['parent_id']) in id_record:
                    id_record[parent_id].child.append(node)
                else:
                    logger.warning(f'任务已忽略（父节点不存在）：{node_name}')
            else:
                root.child.append(node)
        return root

    async def get_paper(self, paper_id):  # 获取任务点信息（包括题目和答案）
        ticket = (await self.post(
            'http://sso.ismartlearning.cn/v1/serviceTicket',
            data={
                'service': 'http://xot-api.ismartlearning.cn/client/textbook/paperinfo'
            }
        )).json()['data']['serverTicket']
        paper_info = (await self.post(
            'http://xot-api.ismartlearning.cn/client/textbook/paperinfo',
            data={
                'paperId': paper_id
            },
            headers={
                'Origin': 'http://me.ismartlearning.cn',
                'Referer': 'http://me.ismartlearning.cn/',
                'X-Requested-With': 'XMLHttpRequest',
                'Accept-Encoding': 'gzip, deflate'
            },
            params={
                'ticket': ticket
            }
        )).json()['data']
        return paper_info

    async def user_info(self):
        logger.info('正在获取用户信息...')
        return (await self.post(
            'https://school.ismartlearning.cn/client/user/student-info')
        ).json()