from django import forms
from django.utils.translation import gettext_lazy as _
from .models import Risk, Control, Incident, ResidualRisk

class RiskStatusForm(forms.ModelForm):
    class Meta:
        model = Risk
        fields = ["status"]
        labels = {"status": _("Status")}
        widgets = {"status": forms.Select(attrs={"class": "select"})}

class ControlStatusForm(forms.ModelForm):
    class Meta:
        model = Control
        fields = ["status"]
        labels = {"status": _("Status")}
        widgets = {"status": forms.Select(attrs={"class": "select"})}

class IncidentStatusForm(forms.ModelForm):
    class Meta:
        model = Incident
        fields = ["status"]
        labels = {"status": _("Status")}
        widgets = {"status": forms.Select(attrs={"class": "select"})}

class ResidualReviewForm(forms.ModelForm):
    class Meta:
        model = ResidualRisk
        fields = ["review_required"]
        labels = {"review_required": _("Review required")}
        widgets = {"review_required": forms.CheckboxInput(attrs={"class": "checkbox"})}