sharix-open-webapp-base/sharix_admin/forms/order.py

from django import forms
from dbsynce.models import Orders
from dbsynce.models import Client
from dbsynce.models import Company
from dbsynce.models import Service
from dbsynce.models import ServiceCategory
from dbsynce.models import Provider
from tickets.models import Ticket
from django.contrib.auth import get_user_model

class OrderForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super(OrderForm, 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()