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

netbox_dns/filtersets/zone_template.py

@@ -0,0 +1,165 @@
+import netaddr
+from netaddr.core import AddrFormatError
+
+import django_filters
+
+from django.db.models import Q
+from django.utils.translation import gettext as _
+
+from netbox.filtersets import NetBoxModelFilterSet
+from tenancy.filtersets import TenancyFilterSet
+from utilities.filters import MultiValueCharFilter
+
+from netbox_dns.models import (
+    ZoneTemplate,
+    RecordTemplate,
+    Registrar,
+    RegistrationContact,
+    NameServer,
+    DNSSECPolicy,
+)
+
+
+__all__ = ("ZoneTemplateFilterSet",)
+
+
+class ZoneTemplateFilterSet(TenancyFilterSet, NetBoxModelFilterSet):
+    class Meta:
+        model = ZoneTemplate
+
+        fields = (
+            "id",
+            "name",
+            "description",
+            "soa_rname",
+        )
+
+    record_template_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=RecordTemplate.objects.all(),
+        field_name="record_templates",
+        to_field_name="id",
+        label=_("Record Template ID"),
+    )
+    record_template = django_filters.ModelMultipleChoiceFilter(
+        queryset=RecordTemplate.objects.all(),
+        field_name="record_templates__name",
+        to_field_name="name",
+        label=_("Record Template"),
+    )
+    nameserver_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=NameServer.objects.all(),
+        field_name="nameservers",
+        to_field_name="id",
+        label=_("Nameservers ID"),
+    )
+    nameserver = django_filters.ModelMultipleChoiceFilter(
+        queryset=NameServer.objects.all(),
+        field_name="nameservers__name",
+        to_field_name="name",
+        label=_("Nameserver"),
+    )
+    soa_mname_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=NameServer.objects.all(),
+        field_name="soa_mname",
+        to_field_name="id",
+        label=_("SOA MName ID"),
+    )
+    soa_mname = django_filters.ModelMultipleChoiceFilter(
+        queryset=NameServer.objects.all(),
+        field_name="soa_mname__name",
+        to_field_name="name",
+        label=_("SOA MName"),
+    )
+    dnssec_policy_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=DNSSECPolicy.objects.all(),
+        label=_("DNSSEC Policy ID"),
+    )
+    dnssec_policy = django_filters.ModelMultipleChoiceFilter(
+        queryset=DNSSECPolicy.objects.all(),
+        field_name="dnssec_policy__name",
+        to_field_name="name",
+        label=_("DNSSEC Policy"),
+    )
+    parental_agents = MultiValueCharFilter(
+        method="filter_parental_agents",
+        label=_("Parental Agents"),
+    )
+    registrar_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=Registrar.objects.all(),
+        label=_("Registrar ID"),
+    )
+    registrar = django_filters.ModelMultipleChoiceFilter(
+        queryset=Registrar.objects.all(),
+        field_name="registrar__name",
+        to_field_name="name",
+        label=_("Registrar"),
+    )
+    registrant_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=RegistrationContact.objects.all(),
+        label=_("Registrant ID"),
+    )
+    registrant = django_filters.ModelMultipleChoiceFilter(
+        queryset=RegistrationContact.objects.all(),
+        field_name="registrant__contact_id",
+        to_field_name="contact_id",
+        label=_("Registrant"),
+    )
+    admin_c_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=RegistrationContact.objects.all(),
+        label=_("Administrative Contact ID"),
+    )
+    admin_c = django_filters.ModelMultipleChoiceFilter(
+        queryset=RegistrationContact.objects.all(),
+        field_name="admin_c__contact_id",
+        to_field_name="contact_id",
+        label=_("Administrative Contact"),
+    )
+    tech_c_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=RegistrationContact.objects.all(),
+        label=_("Technical Contact ID"),
+    )
+    tech_c = django_filters.ModelMultipleChoiceFilter(
+        queryset=RegistrationContact.objects.all(),
+        field_name="tech_c__contact_id",
+        to_field_name="contact_id",
+        label=_("Technical Contact"),
+    )
+    billing_c_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=RegistrationContact.objects.all(),
+        label=_("Billing Contact ID"),
+    )
+    billing_c = django_filters.ModelMultipleChoiceFilter(
+        queryset=RegistrationContact.objects.all(),
+        field_name="billing_c__contact_id",
+        to_field_name="contact_id",
+        label=_("Billing Contact"),
+    )
+
+    def filter_parental_agents(self, queryset, name, value):
+        if not value:
+            return queryset
+
+        query_values = []
+        for v in value:
+            try:
+                query_values.append(str(netaddr.IPAddress(v)))
+            except (AddrFormatError, ValueError):
+                pass
+
+        return queryset.filter(parental_agents__overlap=query_values)
+
+    def search(self, queryset, name, value):
+        if not value.strip():
+            return queryset
+        qs_filter = (
+            Q(name__icontains=value)
+            | Q(description__icontains=value)
+            | Q(soa_rname__icontains=value)
+            | Q(registrar__name__icontains=value)
+            | Q(registry_domain_id__icontains=value)
+            | Q(registrant__name__icontains=value)
+            | Q(admin_c__name__icontains=value)
+            | Q(tech_c__name__icontains=value)
+            | Q(billing_c__name__icontains=value)
+        )
+        return queryset.filter(qs_filter)

netbox_dns/forms/__init__.py

@@ -2,5 +2,9 @@ from .view import *
 from .zone import *
 from .nameserver import *
 from .record import *
-from .contact import *
+from .registration_contact import *
 from .registrar import *
+from .zone_template import *
+from .record_template import *
+from .dnssec_key_template import *
+from .dnssec_policy import *

netbox_dns/forms/dnssec_key_template.py

@@ -0,0 +1,250 @@
+from django import forms
+from django.contrib.postgres.forms import SimpleArrayField
+from django.utils.translation import gettext_lazy as _
+
+from netbox.forms import (
+    NetBoxModelBulkEditForm,
+    NetBoxModelFilterSetForm,
+    NetBoxModelImportForm,
+    NetBoxModelForm,
+)
+from utilities.forms.fields import (
+    TagFilterField,
+    CSVModelChoiceField,
+    DynamicModelChoiceField,
+    DynamicModelMultipleChoiceField,
+)
+from utilities.forms.rendering import FieldSet
+from utilities.forms.widgets import HTMXSelect
+from utilities.forms import add_blank_choice, get_field_value
+from tenancy.models import Tenant, TenantGroup
+from tenancy.forms import TenancyForm, TenancyFilterForm
+
+from netbox_dns.models import DNSSECKeyTemplate, DNSSECPolicy
+from netbox_dns.choices import (
+    DNSSECKeyTemplateTypeChoices,
+    DNSSECKeyTemplateAlgorithmChoices,
+    DNSSECKeyTemplateKeySizeChoices,
+)
+from netbox_dns.fields import TimePeriodField
+
+
+__all__ = (
+    "DNSSECKeyTemplateForm",
+    "DNSSECKeyTemplateFilterForm",
+    "DNSSECKeyTemplateImportForm",
+    "DNSSECKeyTemplateBulkEditForm",
+)
+
+
+class DNSSECKeyTemplateForm(TenancyForm, NetBoxModelForm):
+    class Meta:
+        model = DNSSECKeyTemplate
+
+        fields = (
+            "name",
+            "description",
+            "type",
+            "lifetime",
+            "key_size",
+            "algorithm",
+            "tenant_group",
+            "tenant",
+            "tags",
+        )
+
+        widgets = {
+            "algorithm": HTMXSelect(),
+        }
+
+    fieldsets = (
+        FieldSet(
+            "name",
+            "description",
+            name=_("Attributes"),
+        ),
+        FieldSet(
+            "type",
+            "lifetime",
+            "algorithm",
+            "key_size",
+            name=_("Key Properties"),
+        ),
+        FieldSet(
+            "tenant_group",
+            "tenant",
+            name=_("Tenancy"),
+        ),
+        FieldSet(
+            "tags",
+            name=_("Tags"),
+        ),
+    )
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        algorithm = get_field_value(self, "algorithm")
+        if algorithm != DNSSECKeyTemplateAlgorithmChoices.RSASHA256:
+            del self.fields["key_size"]
+
+    lifetime = TimePeriodField(
+        required=False,
+    )
+
+
+class DNSSECKeyTemplateFilterForm(TenancyFilterForm, NetBoxModelFilterSetForm):
+    model = DNSSECKeyTemplate
+
+    fieldsets = (
+        FieldSet(
+            "q",
+            "filter_id",
+            "tag",
+        ),
+        FieldSet(
+            "name",
+            "description",
+            name=_("Attributes"),
+        ),
+        FieldSet(
+            "policy_id",
+            name=_("Policies"),
+        ),
+        FieldSet(
+            "type",
+            "lifetime",
+            "algorithm",
+            "key_size",
+            name=_("Key Properties"),
+        ),
+        FieldSet(
+            "tenant_group_id",
+            "tenant_id",
+            name=_("Tenancy"),
+        ),
+    )
+
+    name = forms.CharField(
+        required=False,
+    )
+    description = forms.CharField(
+        required=False,
+    )
+    policy_id = DynamicModelMultipleChoiceField(
+        queryset=DNSSECPolicy.objects.all(),
+        required=False,
+        null_option=_("None"),
+        label=_("Policy"),
+    )
+    type = forms.MultipleChoiceField(
+        choices=DNSSECKeyTemplateTypeChoices,
+        required=False,
+    )
+    lifetime = TimePeriodField(
+        label=_("Lifetime"),
+    )
+    algorithm = forms.MultipleChoiceField(
+        choices=DNSSECKeyTemplateAlgorithmChoices,
+        required=False,
+    )
+    key_size = SimpleArrayField(
+        base_field=forms.ChoiceField(
+            choices=DNSSECKeyTemplateKeySizeChoices,
+        ),
+        required=False,
+        help_text=_("Enter a list of integer key sizes, separated by comma (,)"),
+    )
+    tag = TagFilterField(DNSSECKeyTemplate)
+
+
+class DNSSECKeyTemplateImportForm(NetBoxModelImportForm):
+    class Meta:
+        model = DNSSECKeyTemplate
+
+        fields = (
+            "name",
+            "description",
+            "type",
+            "lifetime",
+            "algorithm",
+            "key_size",
+            "tenant",
+            "tags",
+        )
+
+    lifetime = TimePeriodField(
+        required=False,
+        label=_("Lifetime"),
+    )
+    tenant = CSVModelChoiceField(
+        queryset=Tenant.objects.all(),
+        to_field_name="name",
+        required=False,
+        label=_("Tenant"),
+    )
+
+
+class DNSSECKeyTemplateBulkEditForm(NetBoxModelBulkEditForm):
+    model = DNSSECKeyTemplate
+
+    fieldsets = (
+        FieldSet(
+            "description",
+            name=_("Attributes"),
+        ),
+        FieldSet(
+            "type",
+            "lifetime",
+            "algorithm",
+            "key_size",
+            name=_("Key Properties"),
+        ),
+        FieldSet(
+            "tenant_group",
+            "tenant",
+            name=_("Tenancy"),
+        ),
+    )
+
+    nullable_fields = (
+        "description",
+        "tenant",
+        "lifetime",
+        "key_size",
+    )
+
+    description = forms.CharField(
+        max_length=200,
+        required=False,
+        label=_("Description"),
+    )
+    type = forms.ChoiceField(
+        choices=add_blank_choice(DNSSECKeyTemplateTypeChoices),
+        required=False,
+        label=_("Key Type"),
+    )
+    lifetime = TimePeriodField(
+        required=False,
+        label=_("Lifetime"),
+    )
+    algorithm = forms.ChoiceField(
+        choices=add_blank_choice(DNSSECKeyTemplateAlgorithmChoices),
+        required=False,
+        label=_("Algorithm"),
+    )
+    key_size = forms.ChoiceField(
+        choices=add_blank_choice(DNSSECKeyTemplateKeySizeChoices),
+        required=False,
+        label=_("Key Size"),
+    )
+    tenant_group = DynamicModelChoiceField(
+        queryset=TenantGroup.objects.all(),
+        required=False,
+        label=_("Tenant Group"),
+    )
+    tenant = DynamicModelChoiceField(
+        queryset=Tenant.objects.all(),
+        required=False,
+        label=_("Tenant"),
+    )