update

osint/forms.py

@@ -0,0 +1,97 @@
+"""
+Forms for OSINT app.
+"""
+import json
+from django import forms
+from django.core.exceptions import ValidationError
+from .models import SeedWebsite, OSINTKeyword
+
+
+class SeedWebsiteForm(forms.ModelForm):
+    """Form for creating/editing seed websites."""
+    allowed_domains_text = forms.CharField(
+        required=False,
+        widget=forms.Textarea(attrs={
+            'class': 'form-control',
+            'rows': 3,
+            'placeholder': 'Enter domains separated by commas or as JSON array, e.g. example.com, subdomain.example.com\nOr: ["example.com", "subdomain.example.com"]'
+        }),
+        help_text='Enter domains separated by commas or as JSON array. Leave empty for same domain only.'
+    )
+    
+    class Meta:
+        model = SeedWebsite
+        fields = [
+            'name', 'url', 'description', 'is_active', 'priority',
+            'crawl_depth', 'crawl_interval_hours', 'user_agent'
+        ]
+        widgets = {
+            'name': forms.TextInput(attrs={'class': 'form-control'}),
+            'url': forms.URLInput(attrs={'class': 'form-control'}),
+            'description': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}),
+            'is_active': forms.CheckboxInput(attrs={'class': 'form-check-input'}),
+            'priority': forms.Select(attrs={'class': 'form-control'}),
+            'crawl_depth': forms.NumberInput(attrs={'class': 'form-control', 'min': 0, 'max': 5}),
+            'crawl_interval_hours': forms.NumberInput(attrs={'class': 'form-control', 'min': 1}),
+            'user_agent': forms.TextInput(attrs={'class': 'form-control'}),
+        }
+    
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        if self.instance and self.instance.pk and self.instance.allowed_domains:
+            # Convert list to text representation
+            if isinstance(self.instance.allowed_domains, list):
+                self.fields['allowed_domains_text'].initial = ', '.join(self.instance.allowed_domains)
+            else:
+                self.fields['allowed_domains_text'].initial = str(self.instance.allowed_domains)
+    
+    def clean_allowed_domains_text(self):
+        text = self.cleaned_data.get('allowed_domains_text', '').strip()
+        if not text:
+            return []
+        
+        # Try to parse as JSON first
+        try:
+            domains = json.loads(text)
+            if isinstance(domains, list):
+                return [str(d).strip() for d in domains if d]
+        except (json.JSONDecodeError, ValueError):
+            pass
+        
+        # Otherwise, treat as comma-separated
+        domains = [d.strip() for d in text.split(',') if d.strip()]
+        return domains
+    
+    def save(self, commit=True):
+        instance = super().save(commit=False)
+        instance.allowed_domains = self.cleaned_data.get('allowed_domains_text', [])
+        if commit:
+            instance.save()
+        return instance
+
+
+class OSINTKeywordForm(forms.ModelForm):
+    """Form for creating/editing OSINT keywords."""
+    
+    class Meta:
+        model = OSINTKeyword
+        fields = [
+            'name', 'keyword', 'description', 'keyword_type', 'is_active',
+            'case_sensitive', 'confidence_score', 'auto_approve'
+        ]
+        widgets = {
+            'name': forms.TextInput(attrs={'class': 'form-control'}),
+            'keyword': forms.Textarea(attrs={'class': 'form-control', 'rows': 2}),
+            'description': forms.Textarea(attrs={'class': 'form-control', 'rows': 2}),
+            'keyword_type': forms.Select(attrs={'class': 'form-control'}),
+            'is_active': forms.CheckboxInput(attrs={'class': 'form-check-input'}),
+            'case_sensitive': forms.CheckboxInput(attrs={'class': 'form-check-input'}),
+            'confidence_score': forms.NumberInput(attrs={
+                'class': 'form-control',
+                'min': 0,
+                'max': 100,
+                'step': 1
+            }),
+            'auto_approve': forms.CheckboxInput(attrs={'class': 'form-check-input'}),
+        }
+