from django import forms
from .models import *


class CompanyEntityForm(forms.ModelForm):
    class Meta:
        model = CompanyEntity
        fields = [field.name for field in CompanyEntity._meta.fields if
                  field.name not in ['historical_evolution', 'related_bank_accounts_info']]
        widgets = {
            'capital_paid_time': forms.DateInput(attrs={'type': 'date', 'class': 'form-control'}),
            'establishment_time': forms.DateInput(attrs={'type': 'date', 'class': 'form-control'}),
            'business_status': forms.Select(attrs={'class': 'form-control'})
        }


class EntityChangeRecordForm(forms.ModelForm):
    class Meta:
        model = EntityChangeRecord
        fields = '__all__'
        widgets = {
            'change_date': forms.DateInput(attrs={'type': 'date', 'class': 'form-control'}),
            'change_type': forms.TextInput(attrs={'class': 'form-control'}),
            'content_before_change': forms.Textarea(attrs={'class': 'form-control'}),
            'content_after_change': forms.Textarea(attrs={'class': 'form-control'}),
            'change_reason': forms.Textarea(attrs={'class': 'form-control'}),
            'change_executed_by': forms.TextInput(attrs={'class': 'form-control'}),
        }


class CompanyBankAccountForm(forms.ModelForm):
    class Meta:
        model = CompanyBankAccount
        fields = '__all__'
        widgets = {
            'bank_name': forms.TextInput(attrs={'class': 'form-control'}),
            'account_number': forms.TextInput(attrs={'class': 'form-control'}),
            'account_type': forms.Select(attrs={'class': 'form-control'}),
        }


class PrimaryDepartmentForm(forms.ModelForm):
    class Meta:
        model = PrimaryDepartment
        fields = '__all__'
        widgets = {
            'department_name': forms.TextInput(attrs={'class': 'form-control'}),
            'description': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}),
        }


class SecondaryDepartmentForm(forms.ModelForm):
    class Meta:
        model = SecondaryDepartment
        fields = '__all__'
        widgets = {
            'secondary_department_name': forms.TextInput(attrs={'class': 'form-control'}),
            'primary_department': forms.Select(attrs={'class': 'form-control'}),
            'description': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}),
        }