from django import forms
from .models import Shift, Employee, Event, Helper, Reinigung
from django.utils.translation import gettext_lazy as _

class MultipleShiftForm(forms.Form):
    employees = forms.ModelMultipleChoiceField(queryset=Employee.objects.all(), widget=forms.CheckboxSelectMultiple)
    date = forms.DateField(widget=forms.DateInput)
    start = forms.TimeField(required=False, widget=forms.TimeInput(format='%H:%M'))
    end = forms.TimeField(required=False, widget=forms.TimeInput(format='%H:%M'))
    shifttype = forms.ChoiceField(choices=Shift.ShiftType.choices)


class ShiftForm(forms.ModelForm):
    class Meta:
        model = Shift
        fields = ['date', 'start', 'end', 'shifttype', 'employee']


class EventForm(forms.ModelForm):
    class Meta:
        model = Event
        fields = ['date', 'name', 'event_type', 'location', 'cvd', 'cvt']

class ReinigungForm(forms.ModelForm):
    class Meta:
        model = Reinigung
        fields = ['date', 'auftrag', 'info']

class HelperForm(forms.ModelForm):

    class Meta:
        model = Helper
        fields = ['date', 'ben', 'best', 'info']
        widgets = {
            'date': forms.widgets.DateInput(attrs={'type': 'date'})
        }
        labels = {
            'date': _('Datum'),
            'ben': _('Benötigt'),
            'best': _('Bestätigt'),
            'info': _('Info')
        }