from django import forms
from .models import *


class CompanyEntityForm(forms.ModelForm):
    class Meta:
        model = CompanyEntity
        exclude = ['historical_evolution', 'related_bank_accounts_info']
        widgets = {
            'company_name': forms.TextInput(attrs={'class': 'form-control'}),
            'company_type': forms.TextInput(attrs={'class': 'form-control'}),
            'registration_address': forms.TextInput(attrs={'class': 'form-control'}),
            'registered_capital': forms.NumberInput(attrs={'class': 'form-control'}),
            'capital_paid_time': forms.DateInput(attrs={'type': 'date', 'class': 'form-control'}),
            'capital_paid': forms.NumberInput(attrs={'class': 'form-control'}),
            'establishment_time': forms.DateInput(attrs={'type': 'date', 'class': 'form-control'}),
            'operation_period': forms.TextInput(attrs={'class': 'form-control'}),
            'taxpayer_identification_number': forms.TextInput(attrs={'class': 'form-control'}),
            'business_status': forms.Select(attrs={'class': 'form-control'}),
            'scope_of_business': forms.Textarea(attrs={'class': 'form-control'}),
            'purpose_of_company': forms.TextInput(attrs={'class': 'form-control'}),
            'chairman': forms.TextInput(attrs={'class': 'form-control'}),
            'directors': forms.TextInput(attrs={'class': 'form-control'}),
            'supervisor_chairman': forms.TextInput(attrs={'class': 'form-control'}),
            'employee_supervisor': forms.TextInput(attrs={'class': 'form-control'}),
            'supervisors': forms.TextInput(attrs={'class': 'form-control'}),
            'general_manager': forms.TextInput(attrs={'class': 'form-control'}),
            'financial_officer': forms.TextInput(attrs={'class': 'form-control'}),
        }

    def __init__(self, *args, **kwargs):
        super(CompanyEntityForm, self).__init__(*args, **kwargs)

        for field_name in self.fields:
            if field_name != 'company_name':
                self.fields[field_name].required = False

        self.fields['shareholders_and_stakes'].widget.attrs.update({
            'placeholder': '请按正确格式输入，例如：张三50%；李四50%'
        })


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}),
        }