mirror of
https://gitlab.com/megazordpobeda/DataRush.git
synced 2026-05-23 07:27:10 +00:00
110 lines
3.4 KiB
Python
110 lines
3.4 KiB
Python
from django.contrib import admin
|
|
from django.db.models import Q, Sum
|
|
from django.shortcuts import get_object_or_404
|
|
from django.template.response import TemplateResponse
|
|
from django.urls import path, reverse
|
|
from django.utils.html import format_html
|
|
|
|
from apps.competition.models import Competition, State
|
|
from apps.task.admin import CompetitionTaskInline
|
|
from apps.task.models import CompetitionTask
|
|
from apps.user.models import User
|
|
|
|
|
|
@admin.register(Competition)
|
|
class CompetitionAdmin(admin.ModelAdmin):
|
|
list_display = (
|
|
"id",
|
|
"title",
|
|
"end_date",
|
|
"type",
|
|
"view_leaderboard",
|
|
)
|
|
search_fields = (
|
|
"title",
|
|
"description",
|
|
)
|
|
list_filter = (
|
|
"type",
|
|
"participation_type",
|
|
)
|
|
inlines = [CompetitionTaskInline]
|
|
|
|
def get_urls(self):
|
|
urls = super().get_urls()
|
|
custom_urls = [
|
|
path(
|
|
"leaderboard/",
|
|
self.admin_site.admin_view(self.leaderboard_view),
|
|
name="competition_leaderboard",
|
|
),
|
|
path(
|
|
"<uuid:competition_id>/leaderboard/",
|
|
self.admin_site.admin_view(self.competition_leaderboard_view),
|
|
name="competition_specific_leaderboard",
|
|
),
|
|
]
|
|
return custom_urls + urls
|
|
|
|
def view_leaderboard(self, obj):
|
|
url = reverse("admin:competition_specific_leaderboard", args=[obj.id])
|
|
return format_html('<a href="{}">перейти</a>', url)
|
|
|
|
view_leaderboard.short_description = "Лидерборд"
|
|
view_leaderboard.allow_tags = True
|
|
|
|
def competition_leaderboard_view(self, request, competition_id):
|
|
competition = get_object_or_404(Competition, id=competition_id)
|
|
|
|
competition_tasks = CompetitionTask.objects.filter(
|
|
competition=competition
|
|
)
|
|
|
|
leaderboard = (
|
|
User.objects.annotate(
|
|
total_score=Sum(
|
|
"competitiontasksubmission__earned_points",
|
|
filter=Q(
|
|
competitiontasksubmission__status="checked",
|
|
competitiontasksubmission__task__in=competition_tasks,
|
|
),
|
|
)
|
|
)
|
|
.exclude(total_score__isnull=True)
|
|
.order_by("-total_score")[:20]
|
|
)
|
|
|
|
context = dict(
|
|
self.admin_site.each_context(request),
|
|
title=f"Лидерборд для {competition.title}",
|
|
leaderboard=leaderboard,
|
|
competition=competition,
|
|
)
|
|
return TemplateResponse(
|
|
request, "admin/competition_leaderboard.html", context
|
|
)
|
|
|
|
def leaderboard_view(self, request):
|
|
leaderboard = (
|
|
User.objects.annotate(
|
|
total_score=Sum(
|
|
"competitiontasksubmission__earned_points",
|
|
filter=Q(competitiontasksubmission__status="checked"),
|
|
)
|
|
)
|
|
.exclude(total_score__isnull=True)
|
|
.order_by("-total_score")[:20]
|
|
)
|
|
|
|
context = dict(
|
|
self.admin_site.each_context(request),
|
|
title="Лидерборд соревнования",
|
|
leaderboard=leaderboard,
|
|
)
|
|
return TemplateResponse(
|
|
request, "admin/competition_leaderboard.html", context
|
|
)
|
|
|
|
|
|
admin.site.register(State)
|