from django.shortcuts import render, get_object_or_404, redirect
from django.contrib.auth.decorators import login_required
from django.contrib import messages
from .models import Student
from .forms import StudentForm

@login_required
def student_list(request):
    query = request.GET.get('q', '')
    students = Student.objects.select_related('school_class').all()
    if query:
        students = students.filter(first_name__icontains=query) | students.filter(last_name__icontains=query) | students.filter(student_id__icontains=query)
    return render(request, 'students/list.html', {'students': students, 'query': query})

@login_required
def student_detail(request, pk):
    student = get_object_or_404(Student, pk=pk)
    return render(request, 'students/detail.html', {'student': student})

@login_required
def student_add(request):
    form = StudentForm(request.POST or None, request.FILES or None)
    if form.is_valid():
        form.save()
        messages.success(request, 'Student added successfully!')
        return redirect('student_list')
    return render(request, 'students/form.html', {'form': form, 'title': 'Add Student'})

@login_required
def student_edit(request, pk):
    student = get_object_or_404(Student, pk=pk)
    form = StudentForm(request.POST or None, request.FILES or None, instance=student)
    if form.is_valid():
        form.save()
        messages.success(request, 'Student updated successfully!')
        return redirect('student_detail', pk=pk)
    return render(request, 'students/form.html', {'form': form, 'title': 'Edit Student', 'student': student})

@login_required
def student_delete(request, pk):
    student = get_object_or_404(Student, pk=pk)
    if request.method == 'POST':
        student.delete()
        messages.success(request, 'Student deleted.')
        return redirect('student_list')
    return render(request, 'students/confirm_delete.html', {'student': student})