from django.shortcuts import render, redirect
from django.contrib.auth.models import User
from django.contrib.auth import authenticate, login
from .forms import UserSignUpForm, UserLoginForm
from .models import Teacher, Student

def index(request):
    return render(request, 'school/index.html')


def sign_up(request):
    context ={}
    who ={"teacher": Teacher, "student": Student}
    form = UserSignUpForm(request.POST or None)

    if request.method == "POST":
        if form.is_valid() and request.POST.get("who"):
            user = form.save()
            person = who[request.POST.get("who")]
            person(user=user).save()
            return render(request, 'school/index.html')
    context['form'] = form
    return render(request, 'registration/sign_up.html', context)

def sign_in(request):
    context = {}
    form = UserLoginForm(request.POST or None)
    context['form'] = form
    if form.is_valid():
        print('test1')
        username = form.cleaned_data.get('username')
        password = form.cleaned_data.get('password')
        user = authenticate(username=username, password=password)
        login(request, user)
        return redirect('/')

    else:
        print('test3')
        attempt = request.session.get('attempt') or 0
        request.session['attempt'] = attempt + 1
        return render(request, 'registration/sign_in.html', context)

    return render(request, 'registration/sign_in.html', context)