Fix form handling

Author: adamcik

manager.py

@@ -1,10 +1,12 @@
 import re
 from IPy import IP
 
+from django import forms
 from django.db import models, connection
 from django.db.models import sql, query
 from django.db.models.query_utils import QueryWrapper
-from django import forms
+from django.utils.encoding import force_unicode
+from django.utils.safestring import mark_safe
 
 NET_OPERATORS = {
     'lt': '<',
@@ -40,6 +42,7 @@ def add_filter(self, (filter_string, value), *args, **kwargs):
         return super(NetQuery, self).add_filter(
             (filter_string, value), *args, **kwargs)
 
+
 class NetWhere(sql.where.WhereNode):
     def make_atom(self, child, qn):
         table_alias, name, db_type, lookup_type, value_annot, params = child
@@ -78,20 +81,35 @@ def make_atom(self, child, qn):
 
         raise ValueError('Invalid lookup type "%s"' % lookup_type)
 
+
 class NetManger(models.Manager):
     use_for_related_fields = True
 
     def get_query_set(self):
         q = NetQuery(self.model, connection, NetWhere)
         return query.QuerySet(self.model, q)
 
+
+class NetInput(forms.Widget):
+    input_type = 'text'
+
+    def render(self, name, value, attrs=None):
+        # Default forms.Widget compares value != '' which breaks IP...
+        if value is None: value = ''
+        final_attrs = self.build_attrs(attrs, type=self.input_type, name=name)
+        if value:
+            final_attrs['value'] = force_unicode(value)
+        return mark_safe(u'<input%s />' % forms.util.flatatt(final_attrs))
+
+
 class NetAddressFormField(forms.Field):
+    widget = NetInput
     default_error_messages = {
         'invalid': u'Enter a valid IP Address.',
     }
 
     def __init__(self, *args, **kwargs):
-        super(DateTimeField, self).__init__(*args, **kwargs)
+        super(NetAddressFormField, self).__init__(*args, **kwargs)
 
     def clean(self, value):
         super(NetAddressFormField, self).clean(value)
@@ -102,18 +120,20 @@ def clean(self, value):
             return value
         try:
             return IP(value)
-        except ValueError:
-            raise forms.ValidationError(self.error_messages['invalid'])
+        except ValueError, e:
+            raise forms.ValidationError(e)
+
 
-mac_re = re.compile(r'^(([A-F0-9]:){5}[A-F0-9])$')
+mac_re = re.compile(r'^(([A-F0-9]{2}:){5}[A-F0-9]{2})$')
 
 class MACAddressFormField(forms.RegexField):
     default_error_messages = {
         'invalid': u'Enter a valid MAC address.',
     }
 
     def __init__(self, *args, **kwargs):
-        super(IPAddressField, self).__init__(mac_re, *args, **kwargs)
+        super(MACAddressFormField, self).__init__(mac_re, *args, **kwargs)
+
 
 class _NetAddressField(models.Field):
     empty_strings_allowed = False
@@ -153,6 +173,7 @@ def formfield(self, **kwargs):
         defaults.update(kwargs)
         return super(_NetAddressField, self).formfield(**defaults)
 
+
 class InetAddressField(_NetAddressField):
     description = "PostgreSQL INET field"
     max_length = 39
@@ -161,6 +182,7 @@ class InetAddressField(_NetAddressField):
     def db_type(self):
         return 'inet'
 
+
 class CidrAddressField(_NetAddressField):
     description = "PostgreSQL CIDR field"
     max_length = 43
@@ -169,6 +191,7 @@ class CidrAddressField(_NetAddressField):
     def db_type(self):
         return 'cidr'
 
+
 class MACAddressField(models.Field):
     description = "PostgreSQL MACADDR field"
 
@@ -184,6 +207,8 @@ def formfield(self, **kwargs):
         defaults.update(kwargs)
         return super(MACAddressField, self).formfield(**defaults)
 
+
+# ---- TESTS ----
 class InetTestModel(models.Model):
     '''
     >>> cursor = connection.cursor()