From 34523ab53f800a224c86de21c497f4d7663ea4b8 Mon Sep 17 00:00:00 2001 From: cuom1999 Date: Mon, 4 Apr 2022 17:43:56 -0500 Subject: [PATCH] Add median to problem admin --- judge/admin/problem.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/judge/admin/problem.py b/judge/admin/problem.py index cfa954c..cb5bec4 100644 --- a/judge/admin/problem.py +++ b/judge/admin/problem.py @@ -1,4 +1,5 @@ from operator import attrgetter +import statistics from django import forms from django.contrib import admin @@ -132,7 +133,7 @@ class ProblemAdmin(VersionAdmin): (_('Justice'), {'fields': ('banned_users',)}), (_('History'), {'fields': ('change_message',)}), ) - list_display = ['code', 'name', 'show_authors', 'points', 'vote_cnt', 'vote_mean', 'vote_std', 'is_public', 'show_public'] + list_display = ['code', 'name', 'show_authors', 'points', 'vote_cnt', 'vote_mean', 'vote_median', 'vote_std', 'is_public', 'show_public'] ordering = ['code'] search_fields = ('code', 'name', 'authors__user__username', 'curators__user__username') inlines = [LanguageLimitInline, ProblemClarificationInline, ProblemSolutionInline, ProblemTranslationInline] @@ -255,6 +256,10 @@ class ProblemAdmin(VersionAdmin): return obj._vote_cnt vote_cnt.admin_order_field = '_vote_cnt' + def vote_median(self, obj): + votes = obj.problem_points_votes.values_list('points', flat=True) + return statistics.median(votes) if votes else None + class ProblemPointsVoteAdmin(admin.ModelAdmin): list_display = ('vote_points', 'voter', 'voter_rating', 'voter_point', 'problem_name', 'problem_code', 'problem_points') @@ -266,6 +271,9 @@ class ProblemPointsVoteAdmin(admin.ModelAdmin): return request.user.has_perm('judge.edit_own_problem') return obj.problem.is_editable_by(request.user) + def lookup_allowed(self, key, value): + return True + def problem_code(self, obj): return obj.problem.code problem_code.short_description = _('Problem code')