diff --git a/ai/urls/oj.py b/ai/urls/oj.py index e59bffb..8a3434e 100644 --- a/ai/urls/oj.py +++ b/ai/urls/oj.py @@ -3,13 +3,13 @@ from django.urls import path from ..views.oj import ( AIAnalysisAPI, AIDetailDataAPI, - AIWeeklyDataAPI, + AIDurationDataAPI, AIHeatmapDataAPI, ) urlpatterns = [ path("ai/detail", AIDetailDataAPI.as_view()), - path("ai/weekly", AIWeeklyDataAPI.as_view()), + path("ai/duration", AIDurationDataAPI.as_view()), path("ai/analysis", AIAnalysisAPI.as_view()), path("ai/heatmap", AIHeatmapDataAPI.as_view()), ] diff --git a/ai/views/oj.py b/ai/views/oj.py index e916c72..11f987f 100644 --- a/ai/views/oj.py +++ b/ai/views/oj.py @@ -24,7 +24,20 @@ from ai.models import AIAnalysis CACHE_TIMEOUT = 300 DIFFICULTY_MAP = {"Low": "简单", "Mid": "中等", "High": "困难"} DEFAULT_CLASS_SIZE = 45 -GRADE_THRESHOLDS = [(20, "S"), (50, "A"), (85, "B")] + +# 评级阈值配置:(百分位上限, 评级) +GRADE_THRESHOLDS = [ + (10, "S"), # 前10%: S级 - 卓越 + (35, "A"), # 前35%: A级 - 优秀 + (75, "B"), # 前75%: B级 - 良好 + (100, "C"), # 其余: C级 - 及格 +] + +# 小规模参与惩罚配置:(最小人数, 等级降级映射) +SMALL_SCALE_PENALTY = { + "threshold": 10, + "downgrade": {"S": "A", "A": "B"}, +} def get_cache_key(prefix, *args): @@ -36,16 +49,44 @@ def get_difficulty(difficulty): def get_grade(rank, submission_count): + """ + 计算题目完成评级 + + 评级标准: + - S级:前10%,卓越水平(10%的人) + - A级:前35%,优秀水平(25%的人) + - B级:前75%,良好水平(40%的人) + - C级:75%之后,及格水平(25%的人) + + 特殊规则: + - 参与人数少于10人时,S级降为A级,A级降为B级(避免因人少而评级虚高) + + Args: + rank: 用户排名(1表示第一名) + submission_count: 总AC人数 + + Returns: + 评级字符串 ("S", "A", "B", "C") + """ + # 边界检查 if not rank or rank <= 0 or submission_count <= 0: return "C" - if submission_count < DEFAULT_CLASS_SIZE // 3: - return "S" - - top_percent = round(rank / submission_count * 100) + + # 计算百分位(0-100),使用 (rank-1) 使第一名的百分位为0 + percentile = (rank - 1) / submission_count * 100 + + # 根据百分位确定基础评级 + base_grade = "C" for threshold, grade in GRADE_THRESHOLDS: - if top_percent < threshold: - return grade - return "C" + if percentile < threshold: + base_grade = grade + break + + # 小规模参与惩罚:人数太少时降低评级 + if submission_count < SMALL_SCALE_PENALTY["threshold"]: + base_grade = SMALL_SCALE_PENALTY["downgrade"].get(base_grade, base_grade) + + return base_grade def get_class_user_ids(user): @@ -219,7 +260,7 @@ class AIDetailDataAPI(APIView): } -class AIWeeklyDataAPI(APIView): +class AIDurationDataAPI(APIView): @login_required def get(self, request): end_iso = request.GET.get("end") @@ -228,7 +269,7 @@ class AIWeeklyDataAPI(APIView): user = request.user cache_key = get_cache_key( - "ai_weekly", user.id, user.class_name or "", end_iso, duration + "ai_duration", user.id, user.class_name or "", end_iso, duration ) cached_result = cache.get(cache_key) if cached_result: @@ -239,7 +280,7 @@ class AIWeeklyDataAPI(APIView): time_config = self._parse_duration(duration) start = datetime.fromisoformat(end_iso) - time_config["total_delta"] - weekly_data = [] + duration_data = [] for i in range(time_config["show_count"]): start = start + time_config["delta"] period_end = start + time_config["delta"] @@ -272,10 +313,10 @@ class AIWeeklyDataAPI(APIView): user_first_ac, by_problem, user.id ) - weekly_data.append(period_data) + duration_data.append(period_data) - cache.set(cache_key, weekly_data, CACHE_TIMEOUT) - return self.success(weekly_data) + cache.set(cache_key, duration_data, CACHE_TIMEOUT) + return self.success(duration_data) def _parse_duration(self, duration): unit, count = duration.split(":") @@ -332,7 +373,7 @@ class AIAnalysisAPI(APIView): @login_required def post(self, request): details = request.data.get("details") - weekly = request.data.get("weekly") + duration = request.data.get("duration") api_key = get_env("AI_KEY") @@ -342,7 +383,7 @@ class AIAnalysisAPI(APIView): client = OpenAI(api_key=api_key, base_url="https://api.deepseek.com") system_prompt = "你是一个风趣的编程老师,学生使用判题狗平台进行编程练习。请根据学生提供的详细数据和每周数据,给出用户的学习建议,最后写一句鼓励学生的话。请使用 markdown 格式输出,不要在代码块中输出。" - user_prompt = f"这段时间内的详细数据: {details}\n每周或每月的数据: {weekly}" + user_prompt = f"这段时间内的详细数据: {details}\n每周或每月的数据: {duration}" analysis_chunks = [] saved_instance = None @@ -355,7 +396,7 @@ class AIAnalysisAPI(APIView): user=request.user, provider="deepseek", model="deepseek-chat", - data={"details": details, "weekly": weekly}, + data={"details": details, "duration": duration}, system_prompt=system_prompt, user_prompt="这段时间内的详细数据,每周或每月的数据。", analysis="".join(analysis_chunks).strip(),