|
|
@@ -2,11 +2,30 @@ from django.contrib.postgres.fields import ArrayField
|
|
|
from django.core.exceptions import ValidationError
|
|
|
from django.core.validators import MinValueValidator, MaxValueValidator
|
|
|
from django.db import models
|
|
|
-from netaddr import AddrFormatError, EUI, mac_unix_expanded
|
|
|
+from netaddr import AddrFormatError, EUI, eui64_unix_expanded, mac_unix_expanded
|
|
|
|
|
|
from ipam.constants import BGP_ASN_MAX, BGP_ASN_MIN
|
|
|
from .lookups import PathContains
|
|
|
|
|
|
+__all__ = (
|
|
|
+ 'ASNField',
|
|
|
+ 'MACAddressField',
|
|
|
+ 'PathField',
|
|
|
+ 'WWNField',
|
|
|
+)
|
|
|
+
|
|
|
+
|
|
|
+class mac_unix_expanded_uppercase(mac_unix_expanded):
|
|
|
+ word_fmt = '%.2X'
|
|
|
+
|
|
|
+
|
|
|
+class eui64_unix_expanded_uppercase(eui64_unix_expanded):
|
|
|
+ word_fmt = '%.2X'
|
|
|
+
|
|
|
+
|
|
|
+#
|
|
|
+# Fields
|
|
|
+#
|
|
|
|
|
|
class ASNField(models.BigIntegerField):
|
|
|
description = "32-bit ASN field"
|
|
|
@@ -24,10 +43,6 @@ class ASNField(models.BigIntegerField):
|
|
|
return super().formfield(**defaults)
|
|
|
|
|
|
|
|
|
-class mac_unix_expanded_uppercase(mac_unix_expanded):
|
|
|
- word_fmt = '%.2X'
|
|
|
-
|
|
|
-
|
|
|
class MACAddressField(models.Field):
|
|
|
description = "PostgreSQL MAC Address field"
|
|
|
|
|
|
@@ -42,8 +57,8 @@ class MACAddressField(models.Field):
|
|
|
return value
|
|
|
try:
|
|
|
return EUI(value, version=48, dialect=mac_unix_expanded_uppercase)
|
|
|
- except AddrFormatError as e:
|
|
|
- raise ValidationError("Invalid MAC address format: {}".format(value))
|
|
|
+ except AddrFormatError:
|
|
|
+ raise ValidationError(f"Invalid MAC address format: {value}")
|
|
|
|
|
|
def db_type(self, connection):
|
|
|
return 'macaddr'
|
|
|
@@ -54,6 +69,32 @@ class MACAddressField(models.Field):
|
|
|
return str(self.to_python(value))
|
|
|
|
|
|
|
|
|
+class WWNField(models.Field):
|
|
|
+ description = "World Wide Name field"
|
|
|
+
|
|
|
+ def python_type(self):
|
|
|
+ return EUI
|
|
|
+
|
|
|
+ def from_db_value(self, value, expression, connection):
|
|
|
+ return self.to_python(value)
|
|
|
+
|
|
|
+ def to_python(self, value):
|
|
|
+ if value is None:
|
|
|
+ return value
|
|
|
+ try:
|
|
|
+ return EUI(value, version=64, dialect=eui64_unix_expanded_uppercase)
|
|
|
+ except AddrFormatError:
|
|
|
+ raise ValidationError(f"Invalid WWN format: {value}")
|
|
|
+
|
|
|
+ def db_type(self, connection):
|
|
|
+ return 'macaddr8'
|
|
|
+
|
|
|
+ def get_prep_value(self, value):
|
|
|
+ if not value:
|
|
|
+ return None
|
|
|
+ return str(self.to_python(value))
|
|
|
+
|
|
|
+
|
|
|
class PathField(ArrayField):
|
|
|
"""
|
|
|
An ArrayField which holds a set of objects, each identified by a (type, ID) tuple.
|