xuyue/OnlineJudge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add username for ai analysis

Browse Source
This commit is contained in:
yuetsh
2026-05-07 07:34:23 -06:00
parent df097dd610
commit 2e49291d84
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
ai/views/oj.py
View File

@@ -204,8 +204,14 @@ class AIDetailDataAPI(APIView):
def get(self, request): def get(self, request):
start = request.GET.get("start") start = request.GET.get("start")
end = request.GET.get("end") end = request.GET.get("end")
username = request.GET.get("username")
user = request.user 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( cache_key = get_cache_key(
"ai_detail", user.id, user.class_name or "", start, end "ai_detail", user.id, user.class_name or "", start, end
@@ -385,8 +391,14 @@ class AIDurationDataAPI(APIView):
def get(self, request): def get(self, request):
end_iso = request.GET.get("end") end_iso = request.GET.get("end")
duration = request.GET.get("duration") duration = request.GET.get("duration")
username = request.GET.get("username")
user = request.user 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( cache_key = get_cache_key(
"ai_duration", user.id, user.class_name or "", end_iso, duration "ai_duration", user.id, user.class_name or "", end_iso, duration
@@ -668,7 +680,13 @@ class AIHintAPI(APIView):
class AIHeatmapDataAPI(APIView): class AIHeatmapDataAPI(APIView):
@login_required @login_required
def get(self, request): def get(self, request):
username = request.GET.get("username")
user = request.user 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 "") cache_key = get_cache_key("ai_heatmap", user.id, user.class_name or "")
cached_result = cache.get(cache_key) cached_result = cache.get(cache_key)
if cached_result: if cached_result: