~lutostag/ubuntu/trusty/maas/1.5.4+keystone

« back to all changes in this revision

Viewing changes to src/maasserver/dj14/genericipaddressfield.py

  • Committer: Package Import Robot
  • Author(s): Andres Rodriguez
  • Date: 2013-03-04 11:49:44 UTC
  • mto: This revision was merged to the branch mainline in revision 25.
  • Revision ID: package-import@ubuntu.com-20130304114944-azcvu9anlf8mizpa
Tags: upstream-1.3+bzr1452+dfsg
ImportĀ upstreamĀ versionĀ 1.3+bzr1452+dfsg

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# flake8: noqa
2
 
# Extract of Django 1.4's db/models/fields/__init__.py file with modified
3
 
# imports.
4
 
from django.core import exceptions
5
 
from django.db.models.fields import Field
6
 
from django.utils.translation import ugettext_lazy as _
7
 
from maasserver.dj14.forms import GenericIPAddressFormField
8
 
from maasserver.dj14.ipv6 import clean_ipv6_address
9
 
from maasserver.dj14.validators import ip_address_validators
10
 
 
11
 
 
12
 
class GenericIPAddressField(Field):
13
 
    empty_strings_allowed = True
14
 
    description = _("IP address")
15
 
    default_error_messages = {}
16
 
 
17
 
    def __init__(self, protocol='both', unpack_ipv4=False, *args, **kwargs):
18
 
        self.unpack_ipv4 = unpack_ipv4
19
 
        self.default_validators, invalid_error_message = \
20
 
            ip_address_validators(protocol, unpack_ipv4)
21
 
        self.default_error_messages['invalid'] = invalid_error_message
22
 
        kwargs['max_length'] = 39
23
 
        Field.__init__(self, *args, **kwargs)
24
 
 
25
 
    def get_internal_type(self):
26
 
        return "GenericIPAddressField"
27
 
 
28
 
    def to_python(self, value):
29
 
        if value and ':' in value:
30
 
            return clean_ipv6_address(value,
31
 
                self.unpack_ipv4, self.error_messages['invalid'])
32
 
        return value
33
 
 
34
 
    def get_db_prep_value(self, value, connection, prepared=False):
35
 
        if not prepared:
36
 
            value = self.get_prep_value(value)
37
 
        return value or None
38
 
 
39
 
    def get_prep_value(self, value):
40
 
        if value and ':' in value:
41
 
            try:
42
 
                return clean_ipv6_address(value, self.unpack_ipv4)
43
 
            except exceptions.ValidationError:
44
 
                pass
45
 
        return value
46
 
 
47
 
    def formfield(self, **kwargs):
48
 
        defaults = {'form_class': GenericIPAddressFormField}
49
 
        defaults.update(kwargs)
50
 
        return super(GenericIPAddressField, self).formfield(**defaults)