just

from django.shortcuts import render
from django.views import View
from django.http import HttpResponseRedirect
from django.urls import reverse
from .forms import ExamForm, ExamSubjectForm
from .models import Exam
from .models import Batch
from .models import Subject


class ExamCreateView(View):
    # exam_type_class = ExamTypeForm
    exam_form_class = ExamForm
    subject_form_class = ExamSubjectForm
    template_name = 'exam/new.html'

    def get(self, request):
        exam_form = self.exam_form_class()
        subject_form_class = self.subject_form_class()
        return render(request, self.template_name, {'exam_form': exam_form, 'subject_form': subject_form_class})

    def post(self, request):
        if request.method == 'POST':
            exam_form = self.exam_form_class(request.POST)
            if exam_form.is_valid():
                exam_form.save()
            return HttpResponseRedirect(reverse('exam_subjects'))
        # if 'save-exam' in request.POST.keys():
        #     ## TODO: DATA IS NOT SAVING
        #     # exam_form = self.exam_form_class(request.POST,prefix='exam')
        #     # subject_form_class = self.subject_form_class(request.POST, prefix='subject')
        #     # if exam_form.is_valid() and subject_form_class.is_valid():
        #     #     exam = exam_form.save(commit=False)
        #     #     exam.examsubject_set = subject_form_class.save()
        #     return HttpResponseRedirect(reverse('get_exam'))
        # elif 'save-and-add-another-exam' in request.POST.keys():
        #     print(request.POST.keys())
        #     ## TODO: DATA IS NOT SAVING
        #     # exam_form = self.exam_form_class(request.POST)
        #     # subject_form_class = self.subject_form_class(request.POST)
        #     # if exam_form.is_valid():
        #     #     exam_form.save()
        #     # if subject_form_class.is_valid():
        #     #     subject_form_class.save()
        #     return render(request, self.template_name,
        #                   {'exam_form': self.exam_form_class(), 'subject_form': self.subject_form_class()})


class ExamListView(View):
    template_name = 'exam/list.html'
    exams = Exam.objects.order_by('date')

    def get(self, request):
        return render(request, self.template_name, {'exams': self.exams})


class ExamSubjectView(View):
    def getSubjects(request, id):
        template_name = 'exam/subjects_on_exam.html'
        exam = Exam.objects.get(id=id)
        batch = Batch.objects.get(id=exam.batch.id)
        subjects = Subject.objects.filter(academic_class__id=batch.batch_academic_class.id)
        return render(request, template_name, {'subjects': subjects})