Monkey patched User model. Poller js is templated. Plus minor changes

[flowspy] / flowspec / forms.py

diff --git a/flowspec/forms.py b/flowspec/forms.py
index bdd78a6..4daee84 100644 (file)
--- a/flowspec/forms.py
+++ b/flowspec/forms.py
@@ -3,9 +3,9 @@ from django.utils.safestring import mark_safe
 from django.utils.translation import ugettext as _
 from django.utils.translation import ugettext_lazy
 from django.template.defaultfilters import filesizeformat
-
 from flowspy.flowspec.models import * 
 from ipaddr import *
+from django.contrib.auth.models import User
 
 class RouteForm(forms.ModelForm):
 #    name = forms.CharField(help_text=ugettext_lazy("A unique route name,"
@@ -44,6 +44,16 @@ class RouteForm(forms.ModelForm):
         ports = self.cleaned_data.get('port', None)
         destination = self.cleaned_data.get('destination', None)
         destinationports = self.cleaned_data.get('destinationport', None)
+        user = self.cleaned_data.get('applier', None)
+        networks = user.get_profile().peer.networks.all()
+        mynetwork = False
+        if destination:
+            for network in networks:
+                net = IPNetwork(network.network)
+                if IPNetwork(destination) in net:
+                    mynetwork = True
+            if not mynetwork:
+                 raise forms.ValidationError('Destination address/network should belong to your administrative address space. Check My Profile to review your networks')
         if (sourceports and ports):
             raise forms.ValidationError('Cannot create rule for source ports and ports at the same time. Select either ports or source ports')
         if (destinationports and ports):
@@ -54,4 +64,45 @@ class RouteForm(forms.ModelForm):
             raise forms.ValidationError('Once destination port is matched, destination has to be filled as well. Either deselect destination port or fill destination address')
         if not (source or sourceports or ports or destination or destinationports):
             raise forms.ValidationError('Fill at least a Route Match Condition')
-        return self.cleaned_data
\ No newline at end of file
+        return self.cleaned_data
+
+class ThenPlainForm(forms.ModelForm):
+#    action = forms.CharField(initial='rate-limit')
+    class Meta:
+        model = ThenAction
+    
+    def clean_action_value(self):
+        action_value = self.cleaned_data['action_value']
+        if action_value:
+            try:
+                assert(int(action_value))
+                return "%s" %self.cleaned_data["action_value"]
+            except:
+                raise forms.ValidationError('Rate-limiting should be an integer')
+            if int(action_value) < 50:
+                raise forms.ValidationError('Rate-limiting cannot be < 50kbps')
+        else:
+            raise forms.ValidationError('Cannot be empty')
+
+    def clean_action(self):
+        action = self.cleaned_data['action']
+        if action != 'rate-limit':
+            raise forms.ValidationError('Cannot select something other than rate-limit')
+        else:
+            return self.cleaned_data["action"]
+
+class PortPlainForm(forms.ModelForm):
+#    action = forms.CharField(initial='rate-limit')
+    class Meta:
+        model = MatchPort
+    
+    def clean_port(self):
+        port = self.cleaned_data['port']
+        if port:
+            try:
+                assert(int(port))
+                return "%s" %self.cleaned_data["port"]
+            except:
+                raise forms.ValidationError('Port should be an integer')
+        else:
+            raise forms.ValidationError('Cannot be empty')