XH_Digital_Management/application/perf_mgnt/forms.py

73 lines
3.0 KiB
Python

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)