from django import forms
from .models import *


class CompanyEntityForm(forms.ModelForm):
    class Meta:
        model = CompanyEntity
        fields = '__all__'
        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'}),
            'scope_of_business': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}),
            'historical_evolution': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}),
        }


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