from django import forms
from .models import Organization

class OrganizationForm(forms.ModelForm):
    class Meta:
        model = Organization
        fields = [
            'name', 'email', 'employee_headcount', 'annual_revenue', 'critical_applications',
            'compliance_frameworks', 'industry_sector', 'it_dependency', 'data_sensitivity',
            'network_infrastructure', 'remote_workforce_percentage', 'third_party_vendor_access',
            'internal_software_development', 'geographic_scope', 'customer_base', 'customer_type',
            'product_portfolio', 'supplier_base', 'it_infrastructure', 'intellectual_property',
            'sensitive_data', 'integration_level'
        ]
        widgets = {
            'compliance_frameworks': forms.CheckboxSelectMultiple(),
            'it_infrastructure': forms.CheckboxSelectMultiple(),
            'intellectual_property': forms.CheckboxSelectMultiple(),
            'sensitive_data': forms.CheckboxSelectMultiple(),
        }
