from provider.aufe.jwc import JWCClient from provider.aufe.jwc.model import Course, EvaluationRequestParam import asyncio import random from datetime import datetime from typing import Dict, List, Optional, Callable from dataclasses import dataclass, field from enum import Enum from loguru import logger class TaskStatus(Enum): """任务状态枚举""" IDLE = "idle" # 空闲 INITIALIZING = "initializing" # 初始化中 RUNNING = "running" # 运行中 PAUSED = "paused" # 暂停 COMPLETED = "completed" # 完成 FAILED = "failed" # 失败 TERMINATED = "terminated" # 已终止 @dataclass class EvaluationStats: """评价统计信息""" total_courses: int = 0 pending_courses: int = 0 success_count: int = 0 fail_count: int = 0 current_index: int = 0 status: TaskStatus = TaskStatus.IDLE message: str = "" course_list: List[Course] = field(default_factory=list) current_countdown: int = 0 current_course: Optional[Course] = None start_time: Optional[datetime] = None end_time: Optional[datetime] = None error_message: str = "" @dataclass class CurrentCourseInfo: """当前评价课程信息""" is_evaluating: bool = False course_name: str = "" teacher_name: str = "" progress_text: str = "" countdown_seconds: int = 0 current_index: int = -1 total_pending: int = 0 class Constants: """常量定义""" # 等待评价的冷却时间（秒） COUNTDOWN_SECONDS = 140 # 2分20秒 # 随机评价文案 - 总体评价文案 ZGPGS = [ "老师授课生动形象,课堂氛围活跃。", "教学方法新颖,能够激发学习兴趣。", "讲解耐心细致,知识点清晰易懂。", "对待学生公平公正,很有亲和力。", "课堂管理有序,效率高。", "能理论联系实际,深入浅出。", "作业布置合理,有助于巩固知识。", "教学经验丰富,讲解深入浅出。", "关注学生反馈,及时调整教学。", "教学资源丰富,便于学习。", "课堂互动性强,能充分调动积极性。", "教学重点突出,难点突破到位。", "性格开朗,课堂充满活力。", "批改作业认真,评语有指导性。", "教学目标明确,条理清晰。", ] # 额外描述性文案 NICE_0000000200 = [ "常把晦涩理论生活化,知识瞬间亲近起来。", "总用类比解难点,复杂概念秒懂。", "引入行业前沿案例,打开视野新窗口。", "设问巧妙引深思,激发自主探寻答案。", "常分享学科冷知识,拓宽知识边界。", "用跨学科视角解题,思维更灵动。", "鼓励尝试多元解法,创新思维被激活。", "常分享科研趣事,点燃学术热情。", "用思维导图梳理知识,结构一目了然。", "常把学习方法倾囊相授,效率直线提升。", "用历史事件类比,知识记忆更深刻。", "常鼓励跨学科学习,综合素养渐涨。", "分享行业大咖故事,奋斗动力满满。", "总能挖掘知识背后的趣味,学习味十足。", "常组织知识竞赛,学习热情被点燃。", ] # 建议文案 NICE_0000000201 = [ "无", "没有", "没有什么建议,老师很好", "继续保持这么好的教学风格", "希望老师继续分享更多精彩案例", "感谢老师的悉心指导", ] class EvaluationTaskManager: """评价任务管理器 - 基于学号管理""" def __init__(self, jwc_client: JWCClient, user_id: str): """ 初始化评价任务管理器 Args: jwc_client: JWC客户端实例 user_id: 用户学号 """ self.jwc_client = jwc_client self.user_id = user_id self.stats = EvaluationStats() self._task: Optional[asyncio.Task] = None self._stop_event = asyncio.Event() self._progress_callbacks: List[Callable[[EvaluationStats], None]] = [] logger.info(f"初始化评价任务管理器，用户ID: {user_id}") def add_progress_callback(self, callback: Callable[[EvaluationStats], None]): """添加进度回调函数""" self._progress_callbacks.append(callback) def _notify_progress(self): """通知所有进度回调""" for callback in self._progress_callbacks: try: callback(self.stats) except Exception as e: logger.error(f"进度回调执行失败: {str(e)}") async def initialize(self) -> bool: """ 初始化评价环境 Returns: bool: 初始化是否成功 """ try: self.stats.status = TaskStatus.INITIALIZING self.stats.message = "正在检查网络..." self._notify_progress() # 检查网络连接 if not await self.jwc_client.check_network_connection(): self.stats.status = TaskStatus.FAILED self.stats.message = "网络连接失败，请确保连接到校园网或VPN" self.stats.error_message = "网络连接失败" self._notify_progress() return False # 验证环境和Cookie self.stats.message = "正在验证登录状态..." self._notify_progress() if not await self.jwc_client.validate_environment_and_cookie(): self.stats.status = TaskStatus.FAILED self.stats.message = "登录状态失效，请重新登录" self.stats.error_message = "Cookie验证失败" self._notify_progress() return False # 获取Token self.stats.message = "正在获取Token..." self._notify_progress() token = await self.jwc_client.get_token() if not token: self.stats.status = TaskStatus.FAILED self.stats.message = "获取Token失败，可能是评教系统未开放" self.stats.error_message = "Token获取失败" self._notify_progress() return False # 获取课程列表 self.stats.message = "正在获取课程列表..." self._notify_progress() courses = await self.jwc_client.fetch_evaluation_course_list() if not courses: self.stats.status = TaskStatus.FAILED self.stats.message = "未获取到课程列表，请稍后再试" self.stats.error_message = "课程列表获取失败" self._notify_progress() return False # 更新统计信息 pending_courses = [ course for course in courses if getattr(course, "is_evaluated", "否") != "是" ] self.stats.course_list = courses self.stats.total_courses = len(courses) self.stats.pending_courses = len(pending_courses) self.stats.status = TaskStatus.IDLE self.stats.message = ( f"初始化完成，找到 {self.stats.pending_courses} 门待评价课程" ) self.stats.current_course = None logger.info( f"用户 {self.user_id} 初始化完成，待评价课程: {self.stats.pending_courses}" ) self._notify_progress() return True except Exception as e: self.stats.status = TaskStatus.FAILED self.stats.message = f"初始化异常: {str(e)}" self.stats.error_message = str(e) logger.error(f"用户 {self.user_id} 初始化失败: {str(e)}") self._notify_progress() return False async def evaluate_course(self, course: Course, token: str) -> bool: """ 评价单门课程 Args: course: 课程信息 token: CSRF Token Returns: bool: 评价是否成功 """ try: # 设置当前课程 self.stats.current_course = course # 如果课程已评价，则跳过 if getattr(course, "is_evaluated", "否") == "是": logger.info(f"课程已评价，跳过: {course.evaluation_content}") return True # 第一步：访问评价页面 if not await self.jwc_client.access_evaluation_page(token, course): return False course_name = course.evaluation_content logger.info(f"正在准备评价: {course_name}") self.stats.message = "已访问评价页面，等待服务器倒计时完成后提交评价..." self._notify_progress() # 等待服务器倒计时 server_wait_time = Constants.COUNTDOWN_SECONDS # 显示倒计时 for second in range(server_wait_time, 0, -1): # 检查是否被终止 if self._stop_event.is_set(): self.stats.status = TaskStatus.TERMINATED self.stats.message = "任务已被终止" self._notify_progress() return False self.stats.current_countdown = second self.stats.message = f"服务器倒计时: {second} 秒，然后提交评价..." self._notify_progress() await asyncio.sleep(1) self.stats.current_countdown = 0 self.stats.message = "倒计时结束，正在提交评价..." self._notify_progress() # 生成评价数据 evaluation_ratings = {} for i in range(180, 202): key = f"0000000{i}" if i == 200: evaluation_ratings[key] = random.choice(Constants.NICE_0000000200) elif i == 201: evaluation_ratings[key] = random.choice(Constants.NICE_0000000201) else: evaluation_ratings[key] = f"5_{random.choice(['0.8', '1'])}" # 创建评价请求参数 evaluation_param = EvaluationRequestParam( token_value=token, questionnaire_code=( course.questionnaire.questionnaire_number if course.questionnaire else "" ), evaluation_content=( course.id.evaluation_content_number if course.id else "" ), evaluated_people_number=course.id.evaluated_people if course.id else "", zgpj=random.choice(Constants.ZGPGS), rating_items=evaluation_ratings, ) # 提交评价 response = await self.jwc_client.submit_evaluation(evaluation_param) success = response.result == "success" if success: logger.info(f"课程评价成功: {course_name}") else: logger.error(f"课程评价失败: {course_name}, 错误: {response.msg}") # 清除当前课程信息 self.stats.current_course = None self.stats.current_countdown = 0 return success except Exception as e: logger.error(f"评价课程异常: {str(e)}") return False async def start_evaluation_task(self) -> bool: """ 开始评价任务 确保一个用户只能有一个运行中的任务 Returns: bool: 任务是否成功启动 """ # 检查当前状态 if self.stats.status == TaskStatus.RUNNING: logger.warning(f"用户 {self.user_id} 的评价任务已在运行中") return False if self.stats.status == TaskStatus.INITIALIZING: logger.warning(f"用户 {self.user_id} 的评价任务正在初始化中") return False # 检查是否有未完成的异步任务 if self._task and not self._task.done(): logger.warning(f"用户 {self.user_id} 已有任务在执行") return False # 确保任务已经初始化 if self.stats.status == TaskStatus.IDLE and len(self.stats.course_list) == 0: logger.warning(f"用户 {self.user_id} 任务未初始化，请先调用initialize") return False # 重置停止事件 self._stop_event.clear() # 创建新任务 self._task = asyncio.create_task(self._evaluate_all_courses()) logger.info(f"用户 {self.user_id} 开始评价任务") return True async def _evaluate_all_courses(self): """批量评价所有课程（内部方法）""" try: # 获取Token token = await self.jwc_client.get_token() if not token: self.stats.status = TaskStatus.FAILED self.stats.message = "获取Token失败" self._notify_progress() return # 获取待评价课程 pending_courses = [ course for course in self.stats.course_list if getattr(course, "is_evaluated", "否") != "是" ] if not pending_courses: self.stats.status = TaskStatus.COMPLETED self.stats.message = "所有课程已评价完成！" self._notify_progress() return # 开始评价流程 self.stats.status = TaskStatus.RUNNING self.stats.success_count = 0 self.stats.fail_count = 0 self.stats.current_course = None self.stats.start_time = datetime.now() index = 0 while index < len(pending_courses): # 检查是否被终止 if self._stop_event.is_set(): self.stats.status = TaskStatus.TERMINATED self.stats.message = "任务已被终止" self.stats.end_time = datetime.now() self._notify_progress() return course = pending_courses[index] self.stats.current_index = index self.stats.current_course = course course_name = getattr( course.questionnaire, "questionnaire_name", course.evaluation_content, ) self.stats.message = f"正在处理第 {index + 1}/{len(pending_courses)} 门课程: {course_name}" self._notify_progress() # 评价当前课程 success = await self.evaluate_course(course, token) if success: self.stats.success_count += 1 self.stats.message = f"课程评价成功: {course_name}" else: self.stats.fail_count += 1 self.stats.message = f"课程评价失败: {course_name}" self._notify_progress() # 评价完一门课程后，重新获取课程列表 self.stats.message = "正在更新课程列表..." self._notify_progress() # 重新获取课程列表 updated_courses = await self.jwc_client.fetch_evaluation_course_list() if updated_courses: self.stats.course_list = updated_courses pending_courses = [ course for course in updated_courses if getattr(course, "is_evaluated", "否") != "是" ] self.stats.total_courses = len(updated_courses) self.stats.pending_courses = len(pending_courses) self.stats.message = ( f"课程列表已更新，剩余待评价课程: {self.stats.pending_courses}" ) self._notify_progress() # 给服务器一些处理时间 if pending_courses and index < len(pending_courses) - 1: self.stats.message = "准备处理下一门课程..." self._notify_progress() await asyncio.sleep(3) index += 1 # 评价完成 self.stats.status = TaskStatus.COMPLETED self.stats.current_course = None self.stats.end_time = datetime.now() self.stats.message = f"评价完成！成功: {self.stats.success_count}，失败: {self.stats.fail_count}" logger.info( f"用户 {self.user_id} 评价任务完成，成功: {self.stats.success_count}，失败: {self.stats.fail_count}" ) self._notify_progress() except Exception as e: self.stats.status = TaskStatus.FAILED self.stats.error_message = str(e) self.stats.message = f"评价任务异常: {str(e)}" self.stats.end_time = datetime.now() logger.error(f"用户 {self.user_id} 评价任务异常: {str(e)}") self._notify_progress() async def pause_task(self) -> bool: """ 暂停任务 Returns: bool: 是否成功暂停 """ if self.stats.status != TaskStatus.RUNNING: return False self.stats.status = TaskStatus.PAUSED self.stats.message = "任务已暂停" logger.info(f"用户 {self.user_id} 任务已暂停") self._notify_progress() return True async def resume_task(self) -> bool: """ 恢复任务 Returns: bool: 是否成功恢复 """ if self.stats.status != TaskStatus.PAUSED: return False self.stats.status = TaskStatus.RUNNING self.stats.message = "任务已恢复" logger.info(f"用户 {self.user_id} 任务已恢复") self._notify_progress() return True async def terminate_task(self) -> bool: """ 终止任务 Returns: bool: 是否成功终止 """ if self.stats.status not in [TaskStatus.RUNNING, TaskStatus.PAUSED]: return False # 设置停止事件 self._stop_event.set() # 如果有运行中的任务，等待其完成 if self._task and not self._task.done(): try: await asyncio.wait_for(self._task, timeout=5.0) except asyncio.TimeoutError: self._task.cancel() try: await self._task except asyncio.CancelledError: pass self.stats.status = TaskStatus.TERMINATED self.stats.message = "任务已终止" self.stats.end_time = datetime.now() logger.info(f"用户 {self.user_id} 任务已终止") self._notify_progress() return True def get_current_course_info(self) -> CurrentCourseInfo: """ 获取当前评价课程信息 Returns: CurrentCourseInfo: 当前课程信息 """ # 如果没有运行评价任务 if self.stats.status != TaskStatus.RUNNING: return CurrentCourseInfo( is_evaluating=False, progress_text="当前无评价任务" ) # 正在评价但还没有确定是哪门课程 if ( self.stats.current_index < 0 or self.stats.current_index >= len(self.stats.course_list) or self.stats.current_course is None ): return CurrentCourseInfo( is_evaluating=True, progress_text="准备中...", total_pending=self.stats.pending_courses, ) # 正在评价特定课程 course = self.stats.current_course pending_courses = [ c for c in self.stats.course_list if getattr(c, "is_evaluated", "否") != "是" ] index = self.stats.current_index + 1 total = len(pending_courses) countdown_text = ( f" (倒计时: {self.stats.current_countdown}秒)" if self.stats.current_countdown > 0 else "" ) course_name = course.evaluation_content[:20] if len(course.evaluation_content) > 20: course_name += "..." return CurrentCourseInfo( is_evaluating=True, course_name=course_name, teacher_name=course.evaluated_people, progress_text=f"正在评价({index}/{total}): {course_name} - {course.evaluated_people}{countdown_text}", countdown_seconds=self.stats.current_countdown, current_index=self.stats.current_index, total_pending=total, ) def get_task_status(self) -> EvaluationStats: """ 获取任务状态 Returns: EvaluationStats: 任务统计信息 """ return self.stats def get_user_id(self) -> str: """获取用户ID""" return self.user_id # 全局任务管理器字典，以学号为键 _task_managers: Dict[str, EvaluationTaskManager] = {} def get_task_manager( user_id: str, jwc_client: Optional[JWCClient] = None ) -> Optional[EvaluationTaskManager]: """ 获取或创建任务管理器 一个用户只能有一个活跃的任务管理器 Args: user_id: 用户学号 jwc_client: JWC客户端（创建新管理器时需要） Returns: Optional[EvaluationTaskManager]: 任务管理器实例 """ if user_id in _task_managers: existing_manager = _task_managers[user_id] # 检查现有任务的状态 current_status = existing_manager.get_task_status().status # 如果任务已完成、失败或终止，自动清理 if current_status in [ TaskStatus.COMPLETED, TaskStatus.FAILED, TaskStatus.TERMINATED, ]: logger.info(f"自动清理用户 {user_id} 的已完成任务") del _task_managers[user_id] else: # 返回现有的管理器 return existing_manager # 创建新的管理器 if jwc_client is None: return None manager = EvaluationTaskManager(jwc_client, user_id) _task_managers[user_id] = manager logger.info(f"为用户 {user_id} 创建新的任务管理器") return manager def remove_task_manager(user_id: str) -> bool: """ 移除任务管理器 Args: user_id: 用户学号 Returns: bool: 是否成功移除 """ if user_id in _task_managers: del _task_managers[user_id] return True return False def get_all_task_managers() -> Dict[str, EvaluationTaskManager]: """获取所有任务管理器""" return _task_managers.copy()