123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131 |
- from netaddr import IPAddress, IPNetwork, EUI
- from netaddr.core import AddrFormatError
- from django.db import models
- from django.core.exceptions import ValidationError
- from netfields.managers import NET_OPERATORS, NET_TEXT_OPERATORS
- from netfields.forms import InetAddressFormField, CidrAddressFormField, MACAddressFormField
- from netfields.mac import mac_unix_common
- class _NetAddressField(models.Field):
- empty_strings_allowed = False
- def __init__(self, *args, **kwargs):
- kwargs['max_length'] = self.max_length
- super(_NetAddressField, self).__init__(*args, **kwargs)
- def to_python(self, value):
- if not value:
- return value
- try:
- return self.python_type()(value)
- except AddrFormatError as e:
- raise ValidationError(e)
- def get_prep_lookup(self, lookup_type, value):
- if not value:
- return None
- if (lookup_type in NET_OPERATORS and
- NET_OPERATORS[lookup_type] not in NET_TEXT_OPERATORS):
- if lookup_type.startswith('net_contained') and value is not None:
-
- return unicode(value)
- return self.get_prep_value(value)
- return super(_NetAddressField, self).get_prep_lookup(
- lookup_type, value)
- def get_prep_value(self, value):
- if not value:
- return None
- return unicode(self.to_python(value))
- def get_db_prep_lookup(self, lookup_type, value, connection,
- prepared=False):
- if not value:
- return []
- if (lookup_type in NET_OPERATORS and
- NET_OPERATORS[lookup_type] not in NET_TEXT_OPERATORS):
- return [value] if prepared else [self.get_prep_value(value)]
- return super(_NetAddressField, self).get_db_prep_lookup(
- lookup_type, value, connection=connection, prepared=prepared)
- def formfield(self, **kwargs):
- defaults = {'form_class': self.form_class()}
- defaults.update(kwargs)
- return super(_NetAddressField, self).formfield(**defaults)
- class InetAddressField(_NetAddressField):
- description = "PostgreSQL INET field"
- max_length = 39
- __metaclass__ = models.SubfieldBase
- def db_type(self, connection):
- return 'inet'
- def python_type(self):
- return IPAddress
- def form_class(self):
- return InetAddressFormField
- class CidrAddressField(_NetAddressField):
- description = "PostgreSQL CIDR field"
- max_length = 43
- __metaclass__ = models.SubfieldBase
- def db_type(self, connection):
- return 'cidr'
- def python_type(self):
- return IPNetwork
- def form_class(self):
- return CidrAddressFormField
- class MACAddressField(models.Field):
- description = "PostgreSQL MACADDR field"
- def db_type(self, connection):
- return 'macaddr'
- def to_python(self, value):
- if not value:
- return value
- try:
- return EUI(value, dialect=mac_unix_common)
- except AddrFormatError as e:
- raise ValidationError(e)
- def get_prep_value(self, value):
- if not value:
- return None
- return unicode(self.to_python(value))
- def formfield(self, **kwargs):
- defaults = {'form_class': MACAddressFormField}
- defaults.update(kwargs)
- return super(MACAddressField, self).formfield(**defaults)
- try:
- from south.modelsinspector import add_introspection_rules
- add_introspection_rules([], [
- "^netfields\.fields\.InetAddressField",
- "^netfields\.fields\.CidrAddressField",
- "^netfields\.fields\.MACAddressField",
- ])
- except ImportError:
- pass
|