netbox-plugin-dns 0.22.9__py3-none-any.whl → 1.0.0__py3-none-any.whl

netbox_dns/filtersets/zone.py

@@ -0,0 +1,205 @@
+import netaddr
+
+import django_filters
+from django.db.models import Q
+
+from netbox.filtersets import NetBoxModelFilterSet
+from tenancy.filtersets import TenancyFilterSet
+from utilities.filters import MultiValueCharFilter
+
+from netbox_dns.models import (
+    View,
+    Zone,
+    ZoneStatusChoices,
+    Registrar,
+    Contact,
+    NameServer,
+)
+
+
+class ZoneFilterSet(TenancyFilterSet, NetBoxModelFilterSet):
+    status = django_filters.MultipleChoiceFilter(
+        choices=ZoneStatusChoices,
+    )
+    view_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=View.objects.all(),
+        label="View ID",
+    )
+    view = django_filters.ModelMultipleChoiceFilter(
+        queryset=View.objects.all(),
+        field_name="view__name",
+        to_field_name="name",
+        label="View",
+    )
+    # DEPRECATED: Remove in 1.1
+    name_server_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=NameServer.objects.all(),
+        field_name="nameservers",
+        to_field_name="id",
+        label="Nameserver IDs",
+    )
+    # DEPRECATED: Remove in 1.1
+    name_server = django_filters.ModelMultipleChoiceFilter(
+        queryset=NameServer.objects.all(),
+        field_name="nameservers__name",
+        to_field_name="name",
+        label="Nameservers",
+    )
+    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(),
+        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",
+    )
+    arpa_network = MultiValueCharFilter(
+        method="filter_arpa_network",
+        label="ARPA Network",
+    )
+    rfc2317_prefix = MultiValueCharFilter(
+        method="filter_rfc2317_prefix",
+        label="RFC2317 Prefix",
+    )
+    rfc2317_parent_zone_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=Zone.objects.all(),
+        field_name="rfc2317_parent_zone",
+        to_field_name="id",
+        label="RFC2317 Parent Zone",
+    )
+    rfc2317_parent_zone = django_filters.ModelMultipleChoiceFilter(
+        queryset=Zone.objects.all(),
+        field_name="rfc2317_parent_zone__name",
+        to_field_name="name",
+        label="RFC2317 Parent Zone",
+    )
+    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=Contact.objects.all(),
+        label="Registrant ID",
+    )
+    registrant = django_filters.ModelMultipleChoiceFilter(
+        queryset=Contact.objects.all(),
+        field_name="registrant__contact_id",
+        to_field_name="contact_id",
+        label="Registrant",
+    )
+    admin_c_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=Contact.objects.all(),
+        label="Administrative Contact ID",
+    )
+    admin_c = django_filters.ModelMultipleChoiceFilter(
+        queryset=Contact.objects.all(),
+        field_name="admin_c__contact_id",
+        to_field_name="contact_id",
+        label="Administrative Contact",
+    )
+    tech_c_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=Contact.objects.all(),
+        label="Technical Contact ID",
+    )
+    tech_c = django_filters.ModelMultipleChoiceFilter(
+        queryset=Contact.objects.all(),
+        field_name="tech_c__contact_id",
+        to_field_name="contact_id",
+        label="Technical Contact",
+    )
+    billing_c_id = django_filters.ModelMultipleChoiceFilter(
+        queryset=Contact.objects.all(),
+        label="Billing Contact ID",
+    )
+    billing_c = django_filters.ModelMultipleChoiceFilter(
+        queryset=Contact.objects.all(),
+        field_name="billing_c__contact_id",
+        to_field_name="contact_id",
+        label="Billing Contact",
+    )
+    active = django_filters.BooleanFilter(
+        label="Zone is active",
+    )
+
+    class Meta:
+        model = Zone
+        fields = (
+            "id",
+            "name",
+            "description",
+            "default_ttl",
+            "soa_ttl",
+            "soa_rname",
+            "soa_serial",
+            "soa_refresh",
+            "soa_retry",
+            "soa_expire",
+            "soa_minimum",
+            "soa_serial_auto",
+            "rfc2317_parent_managed",
+            "registry_domain_id",
+        )
+
+    def filter_arpa_network(self, queryset, name, value):
+        if not value:
+            return queryset
+        try:
+            arpa_networks = [
+                str(netaddr.IPNetwork(item)) for item in value if item.strip()
+            ]
+            if not arpa_networks:
+                return queryset
+            return queryset.filter(arpa_network__in=arpa_networks)
+
+        except (netaddr.AddrFormatError, ValueError):
+            return queryset.none()
+
+    def filter_rfc2317_prefix(self, queryset, name, value):
+        if not value:
+            return queryset
+        try:
+            rfc2317_prefixes = [
+                str(netaddr.IPNetwork(item)) for item in value if item.strip()
+            ]
+            if not rfc2317_prefixes:
+                return queryset
+            return queryset.filter(rfc2317_prefix__in=rfc2317_prefixes)
+
+        except (netaddr.AddrFormatError, ValueError):
+            return queryset.none()
+
+    def search(self, queryset, name, value):
+        if not value.strip():
+            return queryset
+        qs_filter = (
+            Q(name__icontains=value)
+            | Q(status__icontains=value)
+            | Q(view__name__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/contact.py

@@ -7,15 +7,38 @@ from netbox.forms import (
     NetBoxModelForm,
 )
 from utilities.forms.fields import TagFilterField
+from utilities.forms.rendering import FieldSet
 
 from netbox_dns.models import Contact
 
 
 class ContactForm(NetBoxModelForm):
+    fieldsets = (
+        FieldSet(
+            "name",
+            "description",
+            "contact_id",
+            "organization",
+            "street",
+            "city",
+            "state_province",
+            "postal_code",
+            "country",
+            "phone",
+            "phone_ext",
+            "fax",
+            "fax_ext",
+            "email",
+            name="Contact",
+        ),
+        FieldSet("tags", name="Tags"),
+    )
+
     class Meta:
         model = Contact
         fields = (
             "name",
+            "description",
             "contact_id",
             "organization",
             "street",
@@ -34,25 +57,28 @@ class ContactForm(NetBoxModelForm):
 
 class ContactFilterForm(NetBoxModelFilterSetForm):
     model = Contact
+
     fieldsets = (
-        (None, ("q", "name", "tags", "contact_id")),
-        (
-            "Address",
-            (
-                "organization",
-                "street",
-                "city",
-                "state_province",
-                "postal_code",
-                "country",
-            ),
+        FieldSet("q", "filter_id", "tag"),
+        FieldSet("name", "contact_id", "description", name="Attributes"),
+        FieldSet(
+            "organization",
+            "street",
+            "city",
+            "state_province",
+            "postal_code",
+            "country",
+            name="Address",
         ),
-        ("Communication", ("phone", "phone_ext", "fax", "fax_ext", "email")),
+        FieldSet("phone", "phone_ext", "fax", "fax_ext", "email", name="Communication"),
     )
 
     name = forms.CharField(
         required=False,
     )
+    description = forms.CharField(
+        required=False,
+    )
     contact_id = forms.CharField(
         required=False,
         label="Contact ID",
@@ -94,7 +120,7 @@ class ContactFilterForm(NetBoxModelFilterSetForm):
     email = forms.CharField(
         required=False,
     )
-    tags = TagFilterField(Contact)
+    tag = TagFilterField(Contact)
 
 
 class ContactImportForm(NetBoxModelImportForm):
@@ -102,6 +128,7 @@ class ContactImportForm(NetBoxModelImportForm):
         model = Contact
         fields = (
             "name",
+            "description",
             "contact_id",
             "organization",
             "street",
@@ -125,6 +152,10 @@ class ContactBulkEditForm(NetBoxModelBulkEditForm):
         required=False,
         label="Name",
     )
+    description = forms.CharField(
+        required=False,
+        label="Description",
+    )
     organization = forms.CharField(
         required=False,
         label="Organization",
@@ -171,32 +202,29 @@ class ContactBulkEditForm(NetBoxModelBulkEditForm):
     )
 
     fieldsets = (
-        (None, ("name",)),
-        (
-            "Address",
-            (
-                "organization",
-                "street",
-                "city",
-                "state_province",
-                "postal_code",
-                "country",
-            ),
+        FieldSet("name", "description", name="Attributes"),
+        FieldSet(
+            "organization",
+            "street",
+            "city",
+            "state_province",
+            "postal_code",
+            "country",
+            name="Address",
         ),
-        (
-            "Communication",
-            (
-                "phone",
-                "phone_ext",
-                "fax",
-                "fax_ext",
-                "email",
-            ),
+        FieldSet(
+            "phone",
+            "phone_ext",
+            "fax",
+            "fax_ext",
+            "email",
+            name="Communication",
         ),
     )
 
     nullable_fields = (
         "name",
+        "description",
         "organization",
         "street",
         "city",

netbox_dns/forms/nameserver.py

@@ -12,6 +12,7 @@ from utilities.forms.fields import (
     CSVModelChoiceField,
     DynamicModelChoiceField,
 )
+from utilities.forms.rendering import FieldSet
 from tenancy.models import Tenant
 from tenancy.forms import TenancyForm, TenancyFilterForm
 
@@ -28,8 +29,9 @@ class NameServerForm(TenancyForm, NetBoxModelForm):
             self.initial["name"] = name_to_unicode(initial_name)
 
     fieldsets = (
-        ("Nameserver", ("name", "description", "tags")),
-        ("Tenancy", ("tenant_group", "tenant")),
+        FieldSet("name", "description", name="Nameserver"),
+        FieldSet("tenant_group", "tenant", name="Tenancy"),
+        FieldSet("tags", name="Tags"),
     )
 
     class Meta:
@@ -42,14 +44,16 @@ class NameServerFilterForm(TenancyFilterForm, NetBoxModelFilterSetForm):
 
     name = forms.CharField(
         required=False,
-        label="Name",
+    )
+    description = forms.CharField(
+        required=False,
     )
     tag = TagFilterField(NameServer)
 
     fieldsets = (
-        (None, ("q", "filter_id", "tag")),
-        ("Attributes", ("name",)),
-        ("Tenant", ("tenant_group_id", "tenant_id")),
+        FieldSet("q", "filter_id", "tag"),
+        FieldSet("name", "description", name="Attributes"),
+        FieldSet("tenant_group_id", "tenant_id", name="Tenancy"),
     )
 
 
@@ -78,5 +82,14 @@ class NameServerBulkEditForm(NetBoxModelBulkEditForm):
     description = forms.CharField(max_length=200, required=False)
     tenant = DynamicModelChoiceField(queryset=Tenant.objects.all(), required=False)
 
-    class Meta:
-        nullable_fields = ("description", "tenant")
+    fieldsets = (
+        FieldSet(
+            "name",
+            "description",
+            "tenant",
+            "tags",
+            name="Attributes",
+        ),
+    )
+
+    nullable_fields = ("description", "tenant")

netbox_dns/forms/record.py

@@ -16,6 +16,7 @@ from utilities.forms.fields import (
 )
 from utilities.forms.widgets import BulkEditNullBooleanSelect, APISelect
 from utilities.forms import add_blank_choice
+from utilities.forms.rendering import FieldSet
 from tenancy.models import Tenant
 from tenancy.forms import TenancyForm, TenancyFilterForm
 
@@ -27,10 +28,30 @@ class RecordForm(TenancyForm, NetBoxModelForm):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
+        initial_zone_id = self.initial.get("zone")
+        if initial_zone_id is not None:
+            self.initial["view"] = Zone.objects.get(pk=initial_zone_id).view
+        else:
+            self.initial["view"] = View.get_default_view()
+
         initial_name = self.initial.get("name")
         if initial_name:
             self.initial["name"] = name_to_unicode(initial_name)
 
+    view = DynamicModelChoiceField(
+        queryset=View.objects.all(),
+        required=False,
+        label="View",
+    )
+    zone = DynamicModelChoiceField(
+        queryset=Zone.objects.all(),
+        required=True,
+        query_params={
+            "view_id": "$view",
+        },
+        label="Zone",
+    )
+
     disable_ptr = forms.BooleanField(
         label="Disable PTR",
         required=False,
@@ -41,21 +62,20 @@ class RecordForm(TenancyForm, NetBoxModelForm):
     )
 
     fieldsets = (
-        (
-            "Record",
-            (
-                "name",
-                "zone",
-                "type",
-                "value",
-                "status",
-                "ttl",
-                "disable_ptr",
-                "description",
-                "tags",
-            ),
+        FieldSet(
+            "name",
+            "view",
+            "zone",
+            "type",
+            "value",
+            "status",
+            "ttl",
+            "disable_ptr",
+            "description",
+            name="Record",
         ),
-        ("Tenancy", ("tenant_group", "tenant")),
+        FieldSet("tenant_group", "tenant", name="Tenancy"),
+        FieldSet("tags", name="Tags"),
     )
 
     class Meta:
@@ -78,12 +98,19 @@ class RecordForm(TenancyForm, NetBoxModelForm):
 class RecordFilterForm(TenancyFilterForm, NetBoxModelFilterSetForm):
     model = Record
     fieldsets = (
-        (None, ("q", "filter_id", "tag")),
-        (
-            "Attributes",
-            ("view_id", "zone_id", "name", "fqdn", "type", "value", "status"),
+        FieldSet("q", "filter_id", "tag"),
+        FieldSet(
+            "zone_id",
+            "name",
+            "fqdn",
+            "type",
+            "value",
+            "disable_ptr",
+            "status",
+            "description",
+            name="Attributes",
         ),
-        ("Tenant", ("tenant_group_id", "tenant_id")),
+        FieldSet("tenant_group_id", "tenant_id", name="Tenancy"),
     )
 
     type = forms.MultipleChoiceField(
@@ -92,7 +119,6 @@ class RecordFilterForm(TenancyFilterForm, NetBoxModelFilterSetForm):
     )
     name = forms.CharField(
         required=False,
-        label="Name",
     )
     fqdn = forms.CharField(
         required=False,
@@ -100,7 +126,10 @@ class RecordFilterForm(TenancyFilterForm, NetBoxModelFilterSetForm):
     )
     value = forms.CharField(
         required=False,
-        label="Value",
+    )
+    disable_ptr = forms.NullBooleanField(
+        required=False,
+        label="Disable PTR",
     )
     status = forms.MultipleChoiceField(
         choices=RecordStatusChoices,
@@ -111,10 +140,8 @@ class RecordFilterForm(TenancyFilterForm, NetBoxModelFilterSetForm):
         required=False,
         label="Zone",
     )
-    view_id = DynamicModelMultipleChoiceField(
-        queryset=View.objects.all(),
+    description = forms.CharField(
         required=False,
-        label="View",
     )
     tag = TagFilterField(Record)
 
@@ -130,10 +157,12 @@ class RecordImportForm(NetBoxModelImportForm):
             except forms.ValidationError:
                 pass
 
-        if view:
+        if view is not None:
             self.fields["zone"].queryset = Zone.objects.filter(view=view)
         else:
-            self.fields["zone"].queryset = Zone.objects.filter(view__isnull=True)
+            self.fields["zone"].queryset = Zone.objects.filter(
+                view=View.get_default_view()
+            )
 
     zone = CSVModelChoiceField(
         queryset=Zone.objects.all(),
@@ -231,18 +260,16 @@ class RecordBulkEditForm(NetBoxModelBulkEditForm):
     tenant = DynamicModelChoiceField(queryset=Tenant.objects.all(), required=False)
 
     fieldsets = (
-        (
-            None,
-            (
-                "zone",
-                "type",
-                "value",
-                "status",
-                "ttl",
-                "disable_ptr",
-                "description",
-                "tenant",
-            ),
+        FieldSet(
+            "zone",
+            "type",
+            "value",
+            "status",
+            "ttl",
+            "disable_ptr",
+            "description",
+            name="Attributes",
         ),
+        FieldSet("tenant_group", "tenant", name="Tenancy"),
     )
     nullable_fields = ("description", "ttl", "tenant")