扩展原有库

当你觉得 zhixuewang 库的功能不能满足需求时，你可以轻松地扩展它。

推荐通过编写函数或类，并接收 StudentAccount 或 TeacherAccount 实例作为参数来实现扩展。这样可以直接复用已有的登录会话（Session）。

基本思路

所有的 API 请求都依赖于 requests.Session 对象，该对象在 Account 实例中维护。你可以通过访问 account.session (或 _session) 来发起自定义请求。

示例代码

假设你想添加一个获取某个特定接口数据的功能：

from zhixuewang import StudentAccount, TeacherAccount

def get_custom_data(account: StudentAccount):
    """
    一个扩展示例：获取自定义接口数据
    """
    # 接口地址
    url = "https://www.zhixue.com/some/custom/api"
    
    # 使用 account 内置的 session 发送请求
    # session 会自动携带必要的 Cookie 和 Header
    response = account.get_session().get(url, params={"key": "value"})
    
    if response.status_code == 200:
        return response.json()
    else:
        print(f"请求失败: {response.status_code}")
        return None

# 使用示例
if __name__ == "__main__":
    # 假设你已经登录
    account = login_playwright("username", "password").to_student()
    
    # 调用扩展函数
    data = get_custom_data(account)
    print(data)

编写扩展模块

你可以将常用的扩展功能封装成一个模块（例如 my_extension.py）：

# my_extension.py
from zhixuewang import StudentAccount

class MyAdvancedStudent(StudentAccount):
    def get_super_info(self):
        # 调用原生方法
        exams = self.get_exams()
        # 做一些额外的处理...
        return f"你有 {len(exams)} 场考试"

使用时：

from my_extension import MyAdvancedStudent

student = MyAdvancedStudent(login_playwright("username", "password").get_session())
print(student.get_super_info())

提示

尽量复用库中已有的 models (如 Exam, Subject 等) 来解析数据，保持代码风格一致。