from django import forms
from .models import ProcessInstance, StepInstance

class ProcessInstanceForm(forms.ModelForm):
    class Meta:
        model = ProcessInstance
        fields = ['description', 'process', 'well', 'representative', 'requester', 'priority', 'status', 'current_step']
        widgets = {
            'description': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}),
            'process': forms.Select(attrs={'class': 'form-control'}),
            'well': forms.Select(attrs={'class': 'form-control'}),
            'representative': forms.Select(attrs={'class': 'form-control'}),
            'requester': forms.Select(attrs={'class': 'form-control'}),
            'priority': forms.Select(attrs={'class': 'form-control'}),
            'status': forms.Select(attrs={'class': 'form-control'}),
            'current_step': forms.Select(attrs={'class': 'form-control'}),
        }

class StepInstanceForm(forms.ModelForm):
    class Meta:
        model = StepInstance
        fields = ['status', 'notes']
        widgets = {
            'status': forms.Select(attrs={'class': 'form-control'}),
            'notes': forms.Textarea(attrs={'class': 'form-control', 'rows': 3})
        }