from django.shortcuts import render
from django.contrib.auth.decorators import login_required
from django.forms import inlineformset_factory
from grades.forms import GradingSystemForm
from grades.models import GradingSystem, GradeRange

# Create your views here.
@login_required
def grading_systems(request):
    user = request.user
    default_grading_system = GradingSystem.objects.get(name='default')
    grading_systems = GradingSystem.objects.filter(creator=user).exclude(
            name='default')
    return render(request, 'grading_systems.html', {'default_grading_system':
                  default_grading_system, 'grading_systems': grading_systems})


@login_required
def add_grading_system(request, system_id=None):
    user = request.user
    grading_system = None
    if system_id is not None:
        grading_system = GradingSystem.objects.get(id=system_id)
    GradeRangeFormSet = inlineformset_factory(GradingSystem, GradeRange,
                                              fields='__all__', extra=0)
    grade_form = GradingSystemForm(instance=grading_system)

    if request.method == 'POST':
        formset = GradeRangeFormSet(request.POST, instance=grading_system)
        grade_form = GradingSystemForm(request.POST, instance=grading_system)
        if grade_form.is_valid():
            system = grade_form.save(commit=False)
            system.creator = user
            system.save()
            system_id = system.id
        if formset.is_valid():
            formset.save()
        if 'add' in request.POST:
            GradeRangeFormSet = inlineformset_factory(GradingSystem, GradeRange,
                                                      fields='__all__', extra=1)
    formset = GradeRangeFormSet(instance=grading_system)

    return render(request, 'add_grades.html', {'formset': formset,
                  'grade_form': grade_form, "system_id": system_id})