From 94be33c7a2c97883ab2a05c4d892e11b11745f07 Mon Sep 17 00:00:00 2001 From: yuetsh <517252939@qq.com> Date: Sun, 18 Jan 2026 20:10:46 +0800 Subject: [PATCH] add login summary --- account/views/oj.py | 8 +++ ai/urls/oj.py | 2 + ai/views/oj.py | 124 ++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 134 insertions(+) diff --git a/account/views/oj.py b/account/views/oj.py index 6907a73..1b7909e 100644 --- a/account/views/oj.py +++ b/account/views/oj.py @@ -209,7 +209,11 @@ class UserLoginAPI(APIView): if user.is_disabled: return self.error("Your account has been disabled") if not user.two_factor_auth: + prev_login = user.last_login auth.login(request, user) + request.session["prev_login"] = ( + datetime2str(prev_login) if prev_login else "" + ) return self.success("Succeeded") # `tfa_code` not in post data @@ -217,7 +221,11 @@ class UserLoginAPI(APIView): return self.error("tfa_required") if OtpAuth(user.tfa_token).valid_totp(data["tfa_code"]): + prev_login = user.last_login auth.login(request, user) + request.session["prev_login"] = ( + datetime2str(prev_login) if prev_login else "" + ) return self.success("Succeeded") else: return self.error("Invalid two factor verification code") diff --git a/ai/urls/oj.py b/ai/urls/oj.py index 8a3434e..26cd13b 100644 --- a/ai/urls/oj.py +++ b/ai/urls/oj.py @@ -5,6 +5,7 @@ from ..views.oj import ( AIDetailDataAPI, AIDurationDataAPI, AIHeatmapDataAPI, + AILoginSummaryAPI, ) urlpatterns = [ @@ -12,4 +13,5 @@ urlpatterns = [ path("ai/duration", AIDurationDataAPI.as_view()), path("ai/analysis", AIAnalysisAPI.as_view()), path("ai/heatmap", AIHeatmapDataAPI.as_view()), + path("ai/login_summary", AILoginSummaryAPI.as_view()), ] diff --git a/ai/views/oj.py b/ai/views/oj.py index c824116..737ca5a 100644 --- a/ai/views/oj.py +++ b/ai/views/oj.py @@ -9,9 +9,11 @@ from django.db.models import Min, Count from django.db.models.functions import TruncDate from django.http import StreamingHttpResponse from django.utils import timezone +from django.utils.dateparse import parse_datetime from utils.api import APIView from utils.openai import get_ai_client +from utils.shortcuts import datetime2str from account.models import User from problem.models import Problem @@ -521,6 +523,128 @@ class AIDurationDataAPI(APIView): return "C" + +class AILoginSummaryAPI(APIView): + @login_required + def get(self, request): + user = request.user + end_time = timezone.now() + start_time = self._resolve_start_time(request, user, end_time) + + if end_time - start_time < timedelta(days=1): + summary = { + "start": datetime2str(start_time), + "end": datetime2str(end_time), + "new_problem_count": 0, + "submission_count": 0, + "accepted_count": 0, + "solved_count": 0, + "flowchart_submission_count": 0, + } + return self.success({"summary": summary, "analysis": ""}) + + problems_qs = Problem.objects.filter( + create_time__gte=start_time, + create_time__lte=end_time, + contest_id__isnull=True, + visible=True, + ) + new_problem_count = problems_qs.count() + + submissions_qs = Submission.objects.filter( + user_id=user.id, create_time__gte=start_time, create_time__lte=end_time + ) + submission_count = submissions_qs.count() + accepted_count = submissions_qs.filter(result=JudgeStatus.ACCEPTED).count() + solved_count = ( + submissions_qs.filter(result=JudgeStatus.ACCEPTED) + .values("problem_id") + .distinct() + .count() + ) + flowchart_submission_count = FlowchartSubmission.objects.filter( + user_id=user.id, create_time__gte=start_time, create_time__lte=end_time + ).count() + + summary = { + "start": datetime2str(start_time), + "end": datetime2str(end_time), + "new_problem_count": new_problem_count, + "submission_count": submission_count, + "accepted_count": accepted_count, + "solved_count": solved_count, + "flowchart_submission_count": flowchart_submission_count, + } + + analysis = "" + analysis_error = "" + if submission_count >= 3: + analysis, analysis_error = self._get_ai_analysis(summary) + + data = {"summary": summary, "analysis": analysis} + if analysis_error: + data["analysis_error"] = analysis_error + return self.success(data) + + def _resolve_start_time(self, request, user, end_time): + start_raw = request.session.get("prev_login") or request.GET.get("start") + start_time = parse_datetime(start_raw) if start_raw else None + + if start_time and timezone.is_naive(start_time): + start_time = timezone.make_aware( + start_time, timezone.get_current_timezone() + ) + + if not start_time: + if user.last_login and user.last_login < end_time: + start_time = user.last_login + elif user.create_time: + start_time = user.create_time + else: + start_time = end_time - timedelta(days=7) + + if start_time >= end_time: + start_time = end_time - timedelta(days=1) + + return start_time + + def _get_ai_analysis(self, summary): + try: + client = get_ai_client() + except Exception as exc: + return "", str(exc) + + system_prompt = ( + "你是 OnlineJudge 的学习助教。" + "请根据统计数据给出简短分析(1-2句)，再给出一行结论，" + "结论用“结论：”开头。" + ) + user_prompt = ( + f"时间范围：{summary['start']} 到 {summary['end']}\n" + f"新题目数：{summary['new_problem_count']}\n" + f"提交次数：{summary['submission_count']}\n" + f"AC 次数：{summary['accepted_count']}\n" + f"AC 题目数：{summary['solved_count']}\n" + f"流程图提交数：{summary['flowchart_submission_count']}\n" + ) + + try: + completion = client.chat.completions.create( + model="deepseek-chat", + messages=[ + {"role": "system", "content": system_prompt}, + {"role": "user", "content": user_prompt}, + ], + ) + except Exception as exc: + return "", str(exc) + + if not completion.choices: + return "", "" + + content = completion.choices[0].message.content or "" + return content.strip(), "" + class AIAnalysisAPI(APIView): @login_required def post(self, request):