From 2e49291d84dfc6f7e440e136ec910c03ded107a4 Mon Sep 17 00:00:00 2001
From: yuetsh <517252939@qq.com>
Date: Thu, 7 May 2026 07:34:23 -0600
Subject: [PATCH] add username for ai analysis

---
 ai/views/oj.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/ai/views/oj.py b/ai/views/oj.py
index 553eca7..02bc2e3 100644
--- a/ai/views/oj.py
+++ b/ai/views/oj.py
@@ -204,8 +204,14 @@ class AIDetailDataAPI(APIView):
     def get(self, request):
         start = request.GET.get("start")
         end = request.GET.get("end")
+        username = request.GET.get("username")
 
         user = request.user
+        if username and request.user.is_super_admin():
+            try:
+                user = User.objects.get(username=username)
+            except User.DoesNotExist:
+                return self.error("User not found")
 
         cache_key = get_cache_key(
             "ai_detail", user.id, user.class_name or "", start, end
@@ -385,8 +391,14 @@ class AIDurationDataAPI(APIView):
     def get(self, request):
         end_iso = request.GET.get("end")
         duration = request.GET.get("duration")
+        username = request.GET.get("username")
 
         user = request.user
+        if username and request.user.is_super_admin():
+            try:
+                user = User.objects.get(username=username)
+            except User.DoesNotExist:
+                return self.error("User not found")
 
         cache_key = get_cache_key(
             "ai_duration", user.id, user.class_name or "", end_iso, duration
@@ -668,7 +680,13 @@ class AIHintAPI(APIView):
 class AIHeatmapDataAPI(APIView):
     @login_required
     def get(self, request):
+        username = request.GET.get("username")
         user = request.user
+        if username and request.user.is_super_admin():
+            try:
+                user = User.objects.get(username=username)
+            except User.DoesNotExist:
+                return self.error("User not found")
         cache_key = get_cache_key("ai_heatmap", user.id, user.class_name or "")
         cached_result = cache.get(cache_key)
         if cached_result: