from django import forms
from .models import *
from ..org_mgnt.models import PrimaryDepartment, CompanyEntity


class FixedAssetsInventoryForm(forms.ModelForm):
    class Meta:
        model = FixedAssetsInventory
        fields = '__all__'
        widgets = {
            'asset_id': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入资产编号'}),
            'asset_category': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入资产类别'}),
            'brand': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入品牌'}),
            'model': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入规格型号'}),
            'unit': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入单位'}),
            'quantity': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '请输入数量'}),
            'location': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入存放地点'}),
            'user': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入使用人'}),
            'purchase_date': forms.DateInput(attrs={'class': 'form-control', 'type': 'date'}),
            'recorded_date': forms.DateInput(attrs={'class': 'form-control', 'type': 'date'}),
            'original_value': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '请输入资产原值'}),
            'residual_value': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '请输入残值'}),
            'depreciation_method': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入折旧方法'}),
            'depreciation_years': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '请输入折旧年限'}),
            'depreciation_months': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '请输入折旧月数'}),
            'book_value': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '请输入账面价值'}),
            'status': forms.Select(attrs={'class': 'form-control'}),
        }

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

        self.fields['department'] = forms.ChoiceField(
            choices=[('', '---------')] + [(dept.department_name, dept.department_name) for dept in PrimaryDepartment.objects.all()],
            widget=forms.Select(attrs={'class': 'form-control'}),
            label="所属部门",
            required=False
        )

        self.fields['affiliated_entity'] = forms.ChoiceField(
            choices=[('', '---------')] + [(dept.company_name, dept.company_name) for dept in CompanyEntity.objects.all()],
            widget=forms.Select(attrs={'class': 'form-control'}),
            label="归属主体"
        )

        self.fields['depreciation_method'] = forms.ChoiceField(
            choices=[('', '---------')] + [('年限平均法', '年限平均法'), ('双倍余额递减法', '双倍余额递减法'), ('年数总和法', '年数总和法'), ('工作量法', '工作量法')],
            widget=forms.Select(attrs={'class': 'form-control'}),
            label="折旧方法"
        )

        for field_name in self.fields:
            self.fields[field_name].required = False


class IntangibleAssetsInventoryForm(forms.ModelForm):
    class Meta:
        model = IntangibleAssetsInventory
        fields = '__all__'
        widgets = {
            'asset_id': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '资产编号'}),
            'asset_category': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '资产类别'}),
            'brand': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '品牌'}),
            'model': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '规格型号'}),
            'unit': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '单位'}),
            'quantity': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '数量'}),
            'user': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '使用人'}),
            'purchase_date': forms.DateInput(attrs={'class': 'form-control', 'type': 'date'}),
            'recorded_date': forms.DateInput(attrs={'class': 'form-control', 'type': 'date'}),
            'original_value': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '资产原值'}),
            'residual_value': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '残值'}),
            'amortization_method': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '摊销方法'}),
            'amortization_years': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '摊销年限'}),
            'amortization_months': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '摊销月数'}),
            'book_value': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '账面价值'}),
            'status': forms.Select(attrs={'class': 'form-control'}, choices=IntangibleAssetsInventory.STATUS_CHOICES),
        }

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

        self.fields['department'] = forms.ChoiceField(
            choices=[('', '---------')] + [(dept.department_name, dept.department_name) for dept in PrimaryDepartment.objects.all()],
            widget=forms.Select(attrs={'class': 'form-control'}),
            label="所属部门",
            required=False
        )

        self.fields['affiliated_entity'] = forms.ChoiceField(
            choices=[('', '---------')] + [(dept.company_name, dept.company_name) for dept in CompanyEntity.objects.all()],
            widget=forms.Select(attrs={'class': 'form-control'}),
            label="归属主体"
        )

        for field_name in self.fields:
            self.fields[field_name].required = False