69 lines
3.1 KiB
Python
69 lines
3.1 KiB
Python
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',
|
|
'sensitive_data_types', 'integration_level', 'change_rate', 'threat_actors', 'expert_analysis'
|
|
]
|
|
widgets = {
|
|
'compliance_frameworks': forms.CheckboxSelectMultiple(),
|
|
'it_infrastructure': forms.CheckboxSelectMultiple(),
|
|
'threat_actors': forms.CheckboxSelectMultiple(),
|
|
'sensitive_data_types': forms.CheckboxSelectMultiple(),
|
|
}
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
|
|
# Handle compliance_frameworks "Other"
|
|
frameworks = cleaned_data.get('compliance_frameworks', [])
|
|
other_framework = self.data.get('compliance_frameworks_other', '').strip()
|
|
if 'other' in frameworks and other_framework:
|
|
frameworks = [fw for fw in frameworks if fw != 'other']
|
|
frameworks.append(other_framework)
|
|
cleaned_data['compliance_frameworks'] = frameworks
|
|
|
|
# Handle industry_sector "Other"
|
|
sector = cleaned_data.get('industry_sector')
|
|
sector_other = self.data.get('industry_sector_other', '').strip()
|
|
if sector == 'other' and sector_other:
|
|
cleaned_data['industry_sector'] = sector_other
|
|
|
|
# Handle Sensitive Data Types & Business Impact
|
|
sensitive_data_types = {}
|
|
data_types = [
|
|
('personal', 'personal_applicable', 'personal_impact'),
|
|
('financial', 'financial_applicable', 'financial_impact'),
|
|
('ip', 'ip_applicable', 'ip_impact'),
|
|
('operational', 'operational_applicable', 'operational_impact'),
|
|
('government', 'government_applicable', 'government_impact'),
|
|
('none', 'none_applicable', None)
|
|
]
|
|
for key, applicable_name, impact_name in data_types:
|
|
applicable = self.data.get(applicable_name) == 'on'
|
|
entry = {'applicable': applicable}
|
|
if impact_name:
|
|
impact = self.data.get(impact_name)
|
|
entry['impact'] = int(impact) if impact and impact.isdigit() else None
|
|
sensitive_data_types[key] = entry
|
|
cleaned_data['sensitive_data_types'] = sensitive_data_types
|
|
|
|
return cleaned_data
|
|
|
|
|
|
class GenerateCodesForm(forms.Form):
|
|
count = forms.IntegerField(label="How many codes to generate?", min_value=1, max_value=1000)
|
|
|
|
|
|
class ContactForm(forms.Form):
|
|
name = forms.CharField(label="Name", max_length=100)
|
|
email = forms.EmailField(label="Email")
|
|
message = forms.CharField(label="Message", widget=forms.Textarea(attrs={"rows": 6}), max_length=5000)
|