Merge branch 'archweb' into archweb-generic2

1 files changed, 49 insertions, 0 deletions

diff --git a/mirrors/fields.py b/mirrors/fields.py
new file mode 100644
index 00000000..206c9d7d
--- /dev/null
+++ b/mirrors/fields.py
@@ -0,0 +1,49 @@
+from IPy import IP
+
+from django import forms
+from django.core import validators
+from django.core.exceptions import ValidationError
+from django.db import models
+from south.modelsinspector import add_introspection_rules
+
+
+class IPNetworkFormField(forms.Field):
+    def to_python(self, value):
+        if value in validators.EMPTY_VALUES:
+            return None
+        try:
+            value = IP(value)
+        except ValueError as e:
+            raise ValidationError(str(e))
+        return value
+
+
+class IPNetworkField(models.Field):
+    __metaclass__ = models.SubfieldBase
+    description = "IPv4 or IPv6 address or subnet"
+
+    def __init__(self, *args, **kwargs):
+        kwargs['max_length'] = 44
+        super(IPNetworkField, self).__init__(*args, **kwargs)
+
+    def get_internal_type(self):
+        return "IPAddressField"
+
+    def to_python(self, value):
+        if not value:
+            return None
+        return IP(value)
+
+    def get_prep_value(self, value):
+        value = self.to_python(value)
+        if not value:
+            return None
+        return str(value)
+
+    def formfield(self, **kwargs):
+        defaults = {'form_class': IPNetworkFormField}
+        defaults.update(kwargs)
+        return super(IPNetworkField, self).formfield(**defaults)
+
+
+add_introspection_rules([], ["^mirrors\.fields\.IPNetworkField"])