from django.contrib import messages
from django.shortcuts import render, redirect
from django.contrib.auth.decorators import login_required
from .forms import *
from api.models import *


# Create your views here.
def register_view(request):
    if request.method == 'POST':
        form = UserRegisterForm(request.POST)
        if form.is_valid():
            user = form.save()

            prof = Profile(user=user)
            prof.save()

            messages.success(request, f'Your account has been created! You are now able to log in')
            return redirect('login')
    else:
        form = UserRegisterForm()
    return render(request, 'frontend/register.html', {'form': form})

@login_required
def home_view(request):
    return render(request, 'frontend/temp.html', {'classes': Classroom.objects.filter(student=Profile.objects.get(user=request.user))})

@login_required
def class_form_view(request):
    if request.method == "POST":
        form = ClassroomForm(request.POST)
        if form.is_valid():
            class_i = Classroom(
                student=Profile.objects.get(user=request.user),
                name=form.cleaned_data.get('name'),
                teacher=form.cleaned_data.get('teacher'),
                link=form.cleaned_data.get('link'),
                period=form.cleaned_data.get('period'),
            )
            class_i.save()
            return redirect('home')

    else:
        form = ClassroomForm

    return render(request, 'frontend/classroomForm.html', {'form': form})