from django.shortcuts import get_object_or_404, render
from django.http import HttpResponseRedirect
from django.core.urlresolvers import reverse
from django.views import generic
from .models import Problem, Cislo, Reseni, VysledkyKCislu

# Create your views here.

class ProblemView(generic.DetailView):
    model = Problem
    template_name = 'seminar/problem.html'

class RadekVysledkovky(object):
    pass

class CisloView(generic.DetailView):
    model = Cislo
    template_name = 'seminar/cislo.html'
    def get_context_data(self, **kwargs):
        context = super(CisloView, self).get_context_data(**kwargs)

        vysledky = VysledkyKCislu.objects.filter(cislo = context['cislo']).select_related("resitel")
        reseni = Reseni.objects.filter(cislo_body = context['cislo']).select_related("resitel")

        problemy = sorted(list(set([r.problem for r in reseni])), key=lambda x:x.kod)
        problem_index = {}
        for i in range(len(problemy)):
            problem_index[problemy[i].id] = i

        vysledky_resitele = {}
        vysledkovka = []
        posledni_body = 100000
        for vi in range(len(vysledky)):
            v = vysledky[vi]
            tv = RadekVysledkovky()
            tv.resitel = v.resitel
            tv.vysledek = v
            tv.body = ['']*len(problemy)
            tv.poradi = ''
            if posledni_body > v.body_celkem:
                posledni_body = v.body_celkem
                tv.poradi = vi + 1
            vysledky_resitele[v.resitel.id] = tv
            vysledkovka.append(tv)

        for r in reseni:
            vysledky_resitele[r.resitel.id].body[problem_index[r.problem.id]] = r.body

        context['vysledkovka'] = vysledkovka
        context['problemy'] = problemy
        return context