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


class ProjectLedgerForm(forms.ModelForm):
    primary_department = forms.ChoiceField(
        choices=[('', '---------')] + [(dept.department_name, dept.department_name) for dept in
                                       PrimaryDepartment.objects.all()],
        widget=forms.Select(attrs={'class': 'form-control'}),
        label="一级部门"
    )

    class Meta:
        model = ProjectLedger
        exclude = ['net_income', 'total_amount_including_tax', 'repayment_amount', 'receivable_net_income',
                   'actual_net_income', 'outstanding_net_income']
        widgets = {
            'project_name': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '项目名称'}),
            'start_date': forms.DateInput(attrs={'class': 'form-control', 'type': 'date'}),
            'end_date': forms.DateInput(attrs={'class': 'form-control', 'type': 'date'}),
            'customer_name': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '客户名称'}),
            'province': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '省'}),
            'city': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '市'}),
            'district': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '区县'}),
            'project_leader': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '负责人'}),
            'project_members': forms.Textarea(attrs={'class': 'form-control', 'placeholder': '项目组员'}),
            'project_status': forms.Select(attrs={'class': 'form-control'}),
            'resource_type': forms.Select(attrs={'class': 'form-control'}),
            'project_nature': forms.Select(attrs={'class': 'form-control'}),
            'project_progress': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '项目进度'}),
            'contract_date': forms.DateInput(attrs={'class': 'form-control', 'type': 'date'}),
            'contract_amount': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '标的金额（元）'}),
            'contract_rate': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '合同费率'}),
            'revenue': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '收入（元）'}),
            'cost_rate': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '成本费率（%）'}),
            'cost': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '成本（元）'}),
            'notes': forms.Textarea(attrs={'class': 'form-control', 'placeholder': '备注'})
        }

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


class EmployeeProjectIncomeSettlementForm(forms.ModelForm):
    class Meta:
        model = EmployeeProjectIncomeSettlement
        fields = '__all__'
        widgets = {
            'project_name': forms.Select(attrs={'class': 'form-control'}),
            'year_month': forms.DateInput(attrs={'class': 'form-control', 'type': 'month'}),
            'total_amount_including_tax': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '价税合计金额（元）'}),
            'name': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '姓名'}),
            'contribution_rate': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '贡献率'}),
            'sales_income': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '销售收入（元）'}),
        }