from django import forms
from django.forms import DateInput

from ugssim.models import *
from ugssim.models import Address
from django import forms


class UGSModelForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super(forms.ModelForm, self).__init__(*args, **kwargs)
        # 'description should be the first field
        # value = self.fields.pop('description')
        # copy = self.fields.copy()
        # new_pos = self.fields.keyOrder.index('b')
        # self.fields = {'description': value}
        # self.fields.update(copy)
        # set 'form-control' as standard calls
        for name, field in self.fields.items():
            # if field.widget.__class__ == forms.widgets.TextInput:
            if 'class' in field.widget.attrs:
                field.widget.attrs['class'] += 'form-control'
            else:
                field.widget.attrs.update({'class': 'form-control'})


class AddressForm(UGSModelForm):
    class Meta:
        model = Address
        fields = '__all__'
        widgets = {
            'geburtstag': forms.DateInput(

                format='%Y-%m-%d',
                attrs={  # 'class': 'form-control',
                    'type': 'date'
                }
            ),

        }


class SummaryForm(UGSModelForm):
    class Meta:
        model = Summary
        fields = '__all__'
        widgets = {
            'text': forms.Textarea(attrs={'rows': 10})
        }


class SalesAreaTypeForm(UGSModelForm):
    class Meta:
        model = SalesAreaType
        fields = '__all__'


class SalesAreaForm(UGSModelForm):
    class Meta:
        model = SalesArea
        fields = '__all__'


class CompanyDataForm(UGSModelForm):
    class Meta:
        model = CompanyData
        fields = '__all__'