from django import forms
from .models import Employee


class EmployeeForm(forms.ModelForm):
    class Meta:
        model = Employee
        fields = [
            "name",
            "username",
            "role",
            "office_number",
            "email",
            "directie",
            "serviciu",
            "compartiment",
            "observations",
        ]
        widgets = {
            "name": forms.TextInput(attrs={"class": "form-control"}),
            "username": forms.TextInput(attrs={"class": "form-control"}),
            "role": forms.TextInput(attrs={"class": "form-control"}),
            "office_number": forms.TextInput(attrs={"class": "form-control"}),
            "email": forms.EmailInput(attrs={"class": "form-control"}),
            "directie": forms.Select(attrs={"class": "form-select"}),
            "serviciu": forms.Select(attrs={"class": "form-select"}),
            "compartiment": forms.Select(attrs={"class": "form-select"}),
            "observations": forms.Textarea(attrs={"class": "form-control", "rows": 2}),
        }