from django import forms
from .models import *


class ProjectLedgerForm(forms.ModelForm):
    class Meta:
        model = ProjectLedger
        fields = '__all__'
        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'}),
            'primary_department': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '一级部门'}),
            '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.NumberInput(attrs={'class': 'form-control', 'placeholder': '合同费率'}),
            'revenue': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '收入'}),
            'cost_rate': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '成本费率'}),
            'cost': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '成本'}),
            'net_income': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '净收入'}),
            'total_amount_including_tax': forms.NumberInput(
                attrs={'class': 'form-control', 'placeholder': '价税合计金额'}),
            'repayment_amount': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '回款金额'}),
            'receivable_net_income': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '应收净收入'}),
            'actual_net_income': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '实收净收入'}),
            'outstanding_net_income': forms.NumberInput(attrs={'class': 'form-control', 'placeholder': '待收净收入'}),
            'notes': forms.Textarea(attrs={'class': 'form-control', 'placeholder': '备注'}),
        }


class EmployeeProjectIncomeSettlementForm(forms.ModelForm):
    class Meta:
        model = EmployeeProjectIncomeSettlement
        fields = '__all__'
        widgets = {
            'project_name': forms.Select(attrs={'class': 'form-control', 'placeholder': '请选择项目名称'}),
            'year_month': forms.TextInput(attrs={'class': 'form-control', 'placeholder': 'YYYY-MM'}),
        }