import datetime

from django import forms

from .models import *
from ..org_mgnt.models import PrimaryDepartment


class GroupBusinessTargetForm(forms.ModelForm):
    current_year = datetime.datetime.now().year

    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="一级部门"
    )
    year = forms.ChoiceField(
        choices=[('', '---------')] + [(year, year) for year in range(current_year-1, current_year+2)],
        widget=forms.Select(attrs={'class': 'form-control'}),
        label="年份",
        required=False
    )

    class Meta:
        model = GroupBusinessTarget
        fields = '__all__'
        widgets = {
            'project_nature': forms.Select(attrs={'class': 'form-control'}),
            'sales': forms.NumberInput(attrs={'class': 'form-control'}),
            'total_revenue_target': forms.NumberInput(attrs={'class': 'form-control'}),
            'new_revenue_target': forms.NumberInput(attrs={'class': 'form-control'}),
            'existing_revenue_target': forms.NumberInput(attrs={'class': 'form-control'}),
            'cost_limit': forms.NumberInput(attrs={'class': 'form-control'}),
            'gross_profit': forms.NumberInput(attrs={'class': 'form-control'}),
            'expense_limit': forms.NumberInput(attrs={'class': 'form-control'}),
            'operating_profit': forms.NumberInput(attrs={'class': 'form-control'}),
        }

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


class EmployeePerformanceTargetForm(forms.ModelForm):
    current_year = datetime.datetime.now().year

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

    year = forms.ChoiceField(
        choices=[('', '---------')] + [(year, year) for year in range(current_year-1, current_year+2)],
        widget=forms.Select(attrs={'class': 'form-control'}),
        label="年份",
        required=False
    )

    class Meta:
        model = EmployeePerformanceTarget
        fields = '__all__'
        widgets = {
            'name': forms.TextInput(attrs={'class': 'form-control'}),
            'year': forms.NumberInput(attrs={'class': 'form-control'}),
            'sales_target': forms.NumberInput(attrs={'class': 'form-control'}),
            'total_revenue_target': forms.NumberInput(attrs={'class': 'form-control'}),
            'new_revenue_target': forms.NumberInput(attrs={'class': 'form-control'}),
            'existing_revenue_target': forms.NumberInput(attrs={'class': 'form-control'}),
        }

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