17 1 10 49

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):
    def examCreate(request):
        # exam_type_class = ExamTypeForm
        exam_form_class = ExamForm
        subject_form_class = ExamSubjectForm
        template_name = 'exam/new.html'

        if request.method == 'POST':
            exam_form = exam_form_class(request.POST)
            if exam_form.is_valid():
                exam_form.save()
                return HttpResponseRedirect(reverse('exam_subjects', args=(9,)))
            else:
                return render(request, 'exam/error.html',
                              {'error': exam_form.errors})
        return render(request, template_name, {'exam_form': exam_form_class(), 'subject_form': 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})