Commit 918e14a1 authored by Staś Kycia's avatar Staś Kycia
Browse files

reports fix

parent 8f8e3a1b
Loading
Loading
Loading
Loading
+7 −1
Changes for stats/admin.py: 7 added lines, 1 removed line.
Original line number Diff line number Diff line
import matplotlib
from django.contrib import admin
from django.contrib import admin, messages
from django.contrib.auth.decorators import permission_required
from django.core.exceptions import PermissionDenied
from django.db.models import Q
from django.http import HttpResponseRedirect
from django.shortcuts import get_object_or_404, render
from django.urls import path, reverse
from django.utils.decorators import method_decorator
@@ -131,6 +132,11 @@ class ElectionReportAdmin(admin.ModelAdmin):
    def report_view(self, request, report_id):
        report = get_object_or_404(ElectionReport, pk=report_id)

        if report.total_votes_count == 0:
            report.delete()
            messages.warning(request, "Raport usunięty ze względnu na brak wyborców")
            return HttpResponseRedirect(reverse("admin:stats_electionstats_changelist"))

        results_labels = list(report.option_votes.values_list("name", flat=True))
        results_values = list(report.option_votes.values_list("votes", flat=True))