netbox-plugin-dns 0.21.4__py3-none-any.whl → 1.4.7__py3-none-any.whl

netbox_dns/models/dnssec_policy.py

@@ -0,0 +1,203 @@
+from django.db import models
+from django.utils.translation import gettext_lazy as _
+
+from netbox.models import NetBoxModel
+from netbox.search import SearchIndex, register_search
+from netbox.models.features import ContactsMixin
+from netbox.plugins.utils import get_plugin_config
+
+from netbox_dns.choices import DNSSECPolicyDigestChoices, DNSSECPolicyStatusChoices
+from netbox_dns.fields import ChoiceArrayField
+
+
+__all__ = (
+    "DNSSECPolicy",
+    "DNSSECPolicyIndex",
+)
+
+
+class DNSSECPolicy(ContactsMixin, NetBoxModel):
+    class Meta:
+        verbose_name = _("DNSSEC Policy")
+        verbose_name_plural = _("DNSSEC Policies")
+
+        ordering = ("name",)
+
+    clone_fields = (
+        "name",
+        "key_templates",
+        "description",
+        "tenant",
+    )
+
+    def __str__(self):
+        return str(self.name)
+
+    name = models.CharField(
+        verbose_name=_("Name"),
+        max_length=255,
+        unique=True,
+    )
+    description = models.CharField(
+        verbose_name=_("Description"),
+        max_length=200,
+        blank=True,
+    )
+    status = models.CharField(
+        verbose_name=_("Status"),
+        choices=DNSSECPolicyStatusChoices,
+        default=DNSSECPolicyStatusChoices.STATUS_ACTIVE,
+        blank=False,
+        null=False,
+    )
+
+    inline_signing = models.BooleanField(
+        verbose_name=_("Inline Signing"),
+        help_text=_("Use inline signing"),
+        default=True,
+    )
+    key_templates = models.ManyToManyField(
+        verbose_name=_("Key Templates"),
+        to="DNSSECKeyTemplate",
+        related_name="policies",
+        blank=True,
+    )
+    dnskey_ttl = models.PositiveIntegerField(
+        verbose_name=_("DNSKEY TTL"),
+        blank=True,
+        null=True,
+    )
+    purge_keys = models.PositiveIntegerField(
+        verbose_name=_("Purge Keys"),
+        blank=True,
+        null=True,
+    )
+    publish_safety = models.PositiveIntegerField(
+        verbose_name=_("Publish Safety"),
+        blank=True,
+        null=True,
+    )
+    retire_safety = models.PositiveIntegerField(
+        verbose_name=_("Retire Safety"),
+        blank=True,
+        null=True,
+    )
+    signatures_jitter = models.PositiveIntegerField(
+        verbose_name=_("Signatures Jitter"),
+        blank=True,
+        null=True,
+    )
+    signatures_refresh = models.PositiveIntegerField(
+        verbose_name=_("Signatures Refresh"),
+        blank=True,
+        null=True,
+    )
+    signatures_validity = models.PositiveIntegerField(
+        verbose_name=_("Signatures Validity"),
+        blank=True,
+        null=True,
+    )
+    signatures_validity_dnskey = models.PositiveIntegerField(
+        verbose_name=_("Signatures Validity for DNSKEY"),
+        blank=True,
+        null=True,
+    )
+    max_zone_ttl = models.PositiveIntegerField(
+        verbose_name=_("Max Zone TTL"),
+        blank=True,
+        null=True,
+    )
+    zone_propagation_delay = models.PositiveIntegerField(
+        verbose_name=_("Zone Propagation Delay"),
+        blank=True,
+        null=True,
+    )
+
+    create_cdnskey = models.BooleanField(
+        verbose_name=_("Create CDNSKEY"),
+        null=False,
+        default=True,
+    )
+    cds_digest_types = ChoiceArrayField(
+        base_field=models.CharField(
+            choices=DNSSECPolicyDigestChoices,
+        ),
+        verbose_name=_("CDS Digest Types"),
+        blank=True,
+        null=True,
+        default=list,
+    )
+    parent_ds_ttl = models.PositiveIntegerField(
+        verbose_name=_("Parent DS TTL"),
+        blank=True,
+        null=True,
+    )
+    parent_propagation_delay = models.PositiveIntegerField(
+        verbose_name=_("Parent Propagation Delay"),
+        blank=True,
+        null=True,
+    )
+
+    use_nsec3 = models.BooleanField(
+        verbose_name=_("Use NSEC3"),
+        null=False,
+        default=True,
+    )
+    nsec3_iterations = models.PositiveIntegerField(
+        verbose_name=_("NSEC3 Iterations"),
+        blank=True,
+        null=True,
+    )
+    nsec3_opt_out = models.BooleanField(
+        verbose_name=_("NSEC3 Opt-Out"),
+        blank=True,
+        null=True,
+    )
+    nsec3_salt_size = models.PositiveIntegerField(
+        verbose_name=_("NSEC3 Salt Size"),
+        blank=True,
+        null=True,
+    )
+
+    tenant = models.ForeignKey(
+        verbose_name=_("Tenant"),
+        to="tenancy.Tenant",
+        on_delete=models.PROTECT,
+        related_name="netbox_dns_dnssec_policy",
+        blank=True,
+        null=True,
+    )
+
+    def get_status_color(self):
+        return DNSSECPolicyStatusChoices.colors.get(self.status)
+
+    @property
+    def purge_keys_value(self):
+        return self.purge_keys if self.purge_keys is not None else 776000
+
+    @property
+    def publish_safety_value(self):
+        return self.publish_safety if self.publish_safety is not None else 3600
+
+    def get_effective_value(self, attribute):
+        if not hasattr(self, attribute):
+            raise AttributeError(f"DNSSECPolicy does not have attribute {attribute}")
+
+        if (value := getattr(self, attribute)) is None:
+            return self.get_fallback_setting(attribute)
+
+        return value
+
+    @classmethod
+    def get_fallback_setting(cls, attribute):
+        return get_plugin_config("netbox_dns", f"dnssec_{attribute}")
+
+
+@register_search
+class DNSSECPolicyIndex(SearchIndex):
+    model = DNSSECPolicy
+
+    fields = (
+        ("name", 100),
+        ("description", 500),
+    )

netbox_dns/models/nameserver.py

@@ -2,32 +2,72 @@ from dns import name as dns_name
 
 from django.core.exceptions import ValidationError
 from django.db import models, transaction
-from django.db.models import Q
-from django.urls import reverse
+from django.db.models import Q, UniqueConstraint
+from django.db.models.functions import Lower
+from django.utils.translation import gettext_lazy as _
 
 from netbox.models import NetBoxModel
 from netbox.search import SearchIndex, register_search
+from netbox.models.features import ContactsMixin
+from netbox.plugins.utils import get_plugin_config
 
 from netbox_dns.utilities import (
     name_to_unicode,
     normalize_name,
     NameFormatError,
 )
+from netbox_dns.choices import RecordTypeChoices
 from netbox_dns.validators import validate_fqdn
+from netbox_dns.mixins import ObjectModificationMixin
 
-from .record import Record, RecordTypeChoices
 
+__all__ = (
+    "NameServer",
+    "NameServerIndex",
+)
+
+
+class NameServer(ObjectModificationMixin, ContactsMixin, NetBoxModel):
+    class Meta:
+        verbose_name = _("Nameserver")
+        verbose_name_plural = _("Nameservers")
+
+        ordering = ("name",)
+
+        constraints = [
+            UniqueConstraint(
+                Lower("name"),
+                name="name_unique_ci",
+                violation_error_message=_(
+                    "There is already a nameserver with this name"
+                ),
+            ),
+        ]
+
+    clone_fields = (
+        "name",
+        "description",
+        "tenant",
+    )
+
+    def __str__(self):
+        try:
+            return dns_name.from_text(self.name, origin=None).to_unicode()
+        except dns_name.IDNAException:
+            return self.name
 
-class NameServer(NetBoxModel):
     name = models.CharField(
-        unique=True,
+        verbose_name=_("Name"),
         max_length=255,
+        db_collation="natural_sort",
     )
     description = models.CharField(
+        verbose_name=_("Description"),
         max_length=200,
         blank=True,
     )
     tenant = models.ForeignKey(
+        verbose_name=_("Tenant"),
         to="tenancy.Tenant",
         on_delete=models.PROTECT,
         related_name="netbox_dns_nameservers",
@@ -35,25 +75,17 @@ class NameServer(NetBoxModel):
         null=True,
     )
 
-    clone_fields = ["name", "description"]
-
-    class Meta:
-        ordering = ("name",)
-
-    def __str__(self):
-        try:
-            return dns_name.from_text(self.name, origin=None).to_unicode()
-        except dns_name.IDNAException:
-            return self.name
-
     @property
     def display_name(self):
         return name_to_unicode(self.name)
 
-    def get_absolute_url(self):
-        return reverse("plugins:netbox_dns:nameserver", kwargs={"pk": self.pk})
+    def clean_fields(self, exclude=None):
+        if get_plugin_config("netbox_dns", "convert_names_to_lowercase", False):
+            self.name = self.name.lower()
 
-    def clean(self):
+        super().clean_fields(exclude=exclude)
+
+    def clean(self, *args, **kwargs):
         try:
             self.name = normalize_name(self.name)
         except NameFormatError as exc:
@@ -61,7 +93,7 @@ class NameServer(NetBoxModel):
                 {
                     "name": str(exc),
                 }
-            ) from None
+            )
 
         try:
             validate_fqdn(self.name)
@@ -70,20 +102,20 @@ class NameServer(NetBoxModel):
                 {
                     "name": exc,
                 }
-            ) from None
+            )
+
+        super().clean(*args, **kwargs)
 
     def save(self, *args, **kwargs):
         self.full_clean()
 
-        name_changed = (
-            self.pk is not None and self.name != NameServer.objects.get(pk=self.pk).name
-        )
+        changed_fields = self.changed_fields
 
         with transaction.atomic():
             super().save(*args, **kwargs)
 
-            if name_changed:
-                soa_zones = self.zones_soa.all()
+            if changed_fields is not None and "name" in changed_fields:
+                soa_zones = self.soa_zones.all()
                 for soa_zone in soa_zones:
                     soa_zone.update_soa_record()
 
@@ -93,10 +125,8 @@ class NameServer(NetBoxModel):
 
     def delete(self, *args, **kwargs):
         with transaction.atomic():
-            zones = self.zones.all()
-            for zone in zones:
-                Record.objects.filter(
-                    Q(zone=zone),
+            for zone in self.zones.all():
+                zone.records.filter(
                     Q(managed=True),
                     Q(value=f"{self.name}."),
                     Q(type=RecordTypeChoices.NS),
@@ -108,6 +138,7 @@ class NameServer(NetBoxModel):
 @register_search
 class NameServerIndex(SearchIndex):
     model = NameServer
+
     fields = (
         ("name", 100),
         ("description", 500),