from django.shortcuts import render, redirect
from .forms import MultipleShiftForm
from .models import Shift, Employee


def create_multiple_shifts(request):
    if request.method == 'POST':
        form = MultipleShiftForm(request.POST)
        if form.is_valid():
            employees = form.cleaned_data['employees']
            date = form.cleaned_data['date']
            start = form.cleaned_data['start']
            end = form.cleaned_data['end']
            shifttype = form.cleaned_data['shifttype']

            for employee in employees:
                Shift.objects.create(
                    employee=employee,
                    date=date,
                    start=start,
                    end=end,
                    shifttype=shifttype
                )
            return redirect('shift_success')  # Annahme, dass Sie eine Erfolgsmeldung anzeigen möchten
    else:
        form = MultipleShiftForm()

    return render(request, 'create_multiple_shifts.html', {'form': form})