HH/apps/rca/forms.py

"""
RCA (Root Cause Analysis) forms
"""
RCA (Root Cause Analysis) forms
"""
from django import forms
from django.utils import timezone
from apps.core.models import PriorityChoices
from apps.core.form_mixins import HospitalFieldMixin
from .models import (
    RCAActionStatus,
    RCAActionType,
    RCACorrectiveAction,
    RCAStatus,
    RCASeverity,
    RCAAttachment,
    RCANote,
    RCARootCause,
    RootCauseAnalysis,
)


class RootCauseAnalysisForm(HospitalFieldMixin, forms.ModelForm):
    """Form for creating and editing RootCauseAnalysis"""

    class Meta:
        model = RootCauseAnalysis
        fields = [
            'title',
            'description',
            'background',
            'hospital',
            'department',
            'status',
            'severity',
            'priority',
            'assigned_to',
            'target_completion_date',
            'root_cause_summary',
        ]
        widgets = {
            'title': forms.TextInput(attrs={
                'class': 'form-control',
                'placeholder': 'Enter RCA title'
            }),
            'description': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 4,
                'placeholder': 'Describe the incident or issue'
            }),
            'background': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 3,
                'placeholder': 'Provide background information and context'
            }),
            'hospital': forms.Select(attrs={'class': 'form-select'}),
            'department': forms.Select(attrs={'class': 'form-select'}),
            'status': forms.Select(attrs={'class': 'form-select'}),
            'severity': forms.Select(attrs={'class': 'form-select'}),
            'priority': forms.Select(attrs={'class': 'form-select'}),
            'assigned_to': forms.Select(attrs={'class': 'form-select'}),
            'target_completion_date': forms.DateInput(attrs={
                'class': 'form-control',
                'type': 'date'
            }),
            'root_cause_summary': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 4,
                'placeholder': 'Summary of root cause analysis findings'
            }),
        }

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

        # Filter assigned_to to show only active users
        if 'assigned_to' in self.fields:
            from apps.accounts.models import User
            self.fields['assigned_to'].queryset = User.objects.filter(
                is_active=True
            ).order_by('email')


class RCARootCauseForm(forms.ModelForm):
    """Form for creating and editing RCARootCause"""

    class Meta:
        model = RCARootCause
        fields = [
            'description',
            'category',
            'contributing_factors',
            'likelihood',
            'impact',
            'evidence',
        ]
        widgets = {
            'description': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 3,
                'placeholder': 'Describe the root cause'
            }),
            'category': forms.Select(attrs={'class': 'form-select'}),
            'contributing_factors': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 2,
                'placeholder': 'Factors that contributed to this root cause'
            }),
            'likelihood': forms.NumberInput(attrs={
                'class': 'form-control',
                'min': 1,
                'max': 5,
                'placeholder': '1-5'
            }),
            'impact': forms.NumberInput(attrs={
                'class': 'form-control',
                'min': 1,
                'max': 5,
                'placeholder': '1-5'
            }),
            'evidence': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 2,
                'placeholder': 'Evidence supporting this root cause'
            }),
        }

    def clean_likelihood(self):
        likelihood = self.cleaned_data.get('likelihood')
        if likelihood and (likelihood < 1 or likelihood > 5):
            raise forms.ValidationError('Likelihood must be between 1 and 5')
        return likelihood

    def clean_impact(self):
        impact = self.cleaned_data.get('impact')
        if impact and (impact < 1 or impact > 5):
            raise forms.ValidationError('Impact must be between 1 and 5')
        return impact


class RCACorrectiveActionForm(forms.ModelForm):
    """Form for creating and editing RCACorrectiveAction"""

    class Meta:
        model = RCACorrectiveAction
        fields = [
            'description',
            'action_type',
            'root_cause',
            'responsible_person',
            'target_date',
            'completion_date',
            'status',
            'effectiveness_measure',
            'effectiveness_assessment',
            'effectiveness_score',
            'obstacles',
        ]
        widgets = {
            'description': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 3,
                'placeholder': 'Describe the corrective action'
            }),
            'action_type': forms.Select(attrs={'class': 'form-select'}),
            'root_cause': forms.Select(attrs={'class': 'form-select'}),
            'responsible_person': forms.Select(attrs={'class': 'form-select'}),
            'target_date': forms.DateInput(attrs={
                'class': 'form-control',
                'type': 'date'
            }),
            'completion_date': forms.DateInput(attrs={
                'class': 'form-control',
                'type': 'date'
            }),
            'status': forms.Select(attrs={'class': 'form-select'}),
            'effectiveness_measure': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 2,
                'placeholder': 'How will effectiveness be measured?'
            }),
            'effectiveness_assessment': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 2,
                'placeholder': 'Assessment of action effectiveness'
            }),
            'effectiveness_score': forms.NumberInput(attrs={
                'class': 'form-control',
                'min': 1,
                'max': 5,
                'placeholder': '1-5'
            }),
            'obstacles': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 2,
                'placeholder': 'Obstacles encountered during implementation'
            }),
        }

    def __init__(self, *args, **kwargs):
        rca = kwargs.pop('rca', None)
        super().__init__(*args, **kwargs)

        # Filter root_cause to show only for this RCA
        if 'root_cause' in self.fields and rca:
            self.fields['root_cause'].queryset = rca.root_causes.all()

    def clean_effectiveness_score(self):
        score = self.cleaned_data.get('effectiveness_score')
        if score and (score < 1 or score > 5):
            raise forms.ValidationError('Effectiveness score must be between 1 and 5')
        return score


class RCAFilterForm(forms.Form):
    """Form for filtering RCA list"""
    status = forms.ChoiceField(
        required=False,
        choices=[('', 'All Statuses')] + list(RCAStatus.choices),
        widget=forms.Select(attrs={'class': 'form-select'})
    )
    severity = forms.ChoiceField(
        required=False,
        choices=[('', 'All Severities')] + list(RCASeverity.choices),
        widget=forms.Select(attrs={'class': 'form-select'})
    )
    priority = forms.ChoiceField(
        required=False,
        choices=[('', 'All Priorities')] + list(PriorityChoices.choices),
        widget=forms.Select(attrs={'class': 'form-select'})
    )
    hospital = forms.ChoiceField(
        required=False,
        choices=[('', 'All Hospitals')],
        widget=forms.Select(attrs={'class': 'form-select'})
    )
    department = forms.ChoiceField(
        required=False,
        choices=[('', 'All Departments')],
        widget=forms.Select(attrs={'class': 'form-select'})
    )
    search = forms.CharField(
        required=False,
        widget=forms.TextInput(attrs={
            'class': 'form-control',
            'placeholder': 'Search RCAs...'
        })
    )
    date_from = forms.DateField(
        required=False,
        widget=forms.DateInput(attrs={'class': 'form-control', 'type': 'date'})
    )
    date_to = forms.DateField(
        required=False,
        widget=forms.DateInput(attrs={'class': 'form-control', 'type': 'date'})
    )

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

        # Populate hospital choices
        from apps.organizations.models import Hospital
        hospital_choices = [('', 'All Hospitals')]
        hospital_choices.extend([
            (h.id, h.name)
            for h in Hospital.objects.all()
        ])
        self.fields['hospital'].choices = hospital_choices


class RCAStatusChangeForm(forms.Form):
    """Form for changing RCA status"""
    new_status = forms.ChoiceField(
        choices=RCAStatus.choices,
        widget=forms.Select(attrs={'class': 'form-select'})
    )
    notes = forms.CharField(
        required=False,
        widget=forms.Textarea(attrs={
            'class': 'form-control',
            'rows': 3,
            'placeholder': 'Add notes about this status change'
        })
    )


class RCAApprovalForm(forms.Form):
    """Form for approving RCA"""
    approval_notes = forms.CharField(
        required=False,
        widget=forms.Textarea(attrs={
            'class': 'form-control',
            'rows': 3,
            'placeholder': 'Add approval notes'
        })
    )


class RCAClosureForm(forms.Form):
    """Form for closing RCA"""
    closure_notes = forms.CharField(
        required=True,
        widget=forms.Textarea(attrs={
            'class': 'form-control',
            'rows': 3,
            'placeholder': 'Provide closure notes and summary'
        })
    )
    actual_completion_date = forms.DateField(
        required=True,
        widget=forms.DateInput(attrs={
            'class': 'form-control',
            'type': 'date'
        })
    )

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # Set default to today
        self.fields['actual_completion_date'].initial = timezone.now().date()


class RCAAttachmentForm(forms.ModelForm):
    """Form for uploading RCA attachments"""
    class Meta:
        model = RCAAttachment
        fields = ['file', 'description']
        widgets = {
            'file': forms.FileInput(attrs={'class': 'form-control'}),
            'description': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 2,
                'placeholder': 'Describe this attachment'
            }),
        }


class RCANoteForm(forms.ModelForm):
    """Form for adding RCA notes"""
    class Meta:
        model = RCANote
        fields = ['note', 'is_internal']
        widgets = {
            'note': forms.Textarea(attrs={
                'class': 'form-control',
                'rows': 3,
                'placeholder': 'Add a note'
            }),
            'is_internal': forms.CheckboxInput(attrs={
                'class': 'form-check-input'
            }),
        }