from django import forms
from .models import *


class GroupBusinessTargetForm(forms.ModelForm):
    class Meta:
        model = GroupBusinessTarget
        fields = '__all__'
        widgets = {
            'year': forms.NumberInput(attrs={'class': 'form-control'}),
            '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'}),
        }


class EmployeePerformanceTargetForm(forms.ModelForm):
    class Meta:
        model = EmployeePerformanceTarget
        fields = '__all__'
        widgets = {
            'name': forms.TextInput(attrs={'class': 'form-control'}),
            'department': forms.Select(attrs={'class': 'form-control'}),
            'year': forms.NumberInput(attrs={'class': 'form-control'}),
            'project_nature': forms.Select(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'}),
        }