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)