from django import forms
from dbsynce.models import Orders

class OrderForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super(ResourceForm, self).__init__(*args, **kwargs)
        # Добавляем класс .form-control для всех полей формы
        for field in iter(self.fields):
            if 'class' not in self.fields[field].widget.attrs:
                self.fields[field].widget.attrs.update({'class': 'form-control'})
                
    class Meta:
        model = Orders
        fields = [
            'status', 'title', 'note', 'time_placed', 'time_start', 'time_start_real',
            'time_start_predicted', 'time_finish_real', 'time_finish_predicted',
            'real_price', 'predicted_price', 'asap', 'ticket', 'service',
            'service_category', 'client', 'provider', 'receiver', 'company'
        ]
        widgets = {
            'time_placed': forms.DateTimeInput(attrs={'type': 'datetime-local'}),
            'time_start': forms.DateTimeInput(attrs={'type': 'datetime-local'}),
            'time_start_real': forms.DateTimeInput(attrs={'type': 'datetime-local'}),
            'time_start_predicted': forms.DateTimeInput(attrs={'type': 'datetime-local'}),
            'time_finish_real': forms.DateTimeInput(attrs={'type': 'datetime-local'}),
            'time_finish_predicted': forms.DateTimeInput(attrs={'type': 'datetime-local'}),
        }

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['receiver'].queryset = get_user_model().objects.all()
        self.fields['client'].queryset = Client.objects.all()
        self.fields['company'].queryset = Company.objects.all()
        self.fields['provider'].queryset = Provider.objects.all()
        self.fields['service'].queryset = Service.objects.all()
        self.fields['service_category'].queryset = ServiceCategory.objects.all()
        self.fields['ticket'].queryset = Ticket.objects.all()