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

netbox_dns/forms/zone_template.py

@@ -0,0 +1,463 @@
+from django import forms
+from django.utils.translation import gettext_lazy as _
+from django.contrib.postgres.forms import SimpleArrayField
+
+from netbox.forms import (
+    NetBoxModelBulkEditForm,
+    NetBoxModelFilterSetForm,
+    NetBoxModelImportForm,
+    NetBoxModelForm,
+)
+from utilities.forms.fields import (
+    DynamicModelMultipleChoiceField,
+    TagFilterField,
+    CSVModelChoiceField,
+    CSVModelMultipleChoiceField,
+    DynamicModelChoiceField,
+)
+from utilities.forms.rendering import FieldSet
+from tenancy.models import Tenant, TenantGroup
+from tenancy.forms import TenancyForm, TenancyFilterForm
+
+from netbox_dns.models import (
+    ZoneTemplate,
+    RecordTemplate,
+    NameServer,
+    Registrar,
+    RegistrationContact,
+    DNSSECPolicy,
+)
+
+
+__all__ = (
+    "ZoneTemplateForm",
+    "ZoneTemplateFilterForm",
+    "ZoneTemplateImportForm",
+    "ZoneTemplateBulkEditForm",
+)
+
+
+class ZoneTemplateForm(TenancyForm, NetBoxModelForm):
+    class Meta:
+        model = ZoneTemplate
+
+        fields = (
+            "name",
+            "nameservers",
+            "soa_mname",
+            "soa_rname",
+            "dnssec_policy",
+            "parental_agents",
+            "record_templates",
+            "description",
+            "registrar",
+            "registrant",
+            "admin_c",
+            "tech_c",
+            "billing_c",
+            "tenant_group",
+            "tenant",
+            "tags",
+        )
+
+        labels = {
+            "soa_rname": _("RName"),
+        }
+
+    fieldsets = (
+        FieldSet(
+            "name",
+            "description",
+            "nameservers",
+            name=_("Zone Template"),
+        ),
+        FieldSet(
+            "soa_mname",
+            "soa_rname",
+            name=_("SOA"),
+        ),
+        FieldSet(
+            "record_templates",
+            name=_("Record Templates"),
+        ),
+        FieldSet(
+            "dnssec_policy",
+            "parental_agents",
+            name=_("DNSSEC"),
+        ),
+        FieldSet(
+            "registrar",
+            "registrant",
+            "admin_c",
+            "tech_c",
+            "billing_c",
+            name=_("Domain Registration"),
+        ),
+        FieldSet(
+            "tenant_group",
+            "tenant",
+            name=_("Tenancy"),
+        ),
+        FieldSet(
+            "tags",
+            name=_("Tags"),
+        ),
+    )
+
+    nameservers = DynamicModelMultipleChoiceField(
+        queryset=NameServer.objects.all(),
+        required=False,
+        quick_add=True,
+    )
+    soa_mname = DynamicModelChoiceField(
+        queryset=NameServer.objects.all(),
+        required=False,
+        label=_("MName"),
+        quick_add=True,
+    )
+    record_templates = DynamicModelMultipleChoiceField(
+        queryset=RecordTemplate.objects.all(),
+        required=False,
+        quick_add=True,
+    )
+
+
+class ZoneTemplateFilterForm(TenancyFilterForm, NetBoxModelFilterSetForm):
+    model = ZoneTemplate
+
+    fieldsets = (
+        FieldSet(
+            "q",
+            "filter_id",
+            "tag",
+        ),
+        FieldSet(
+            "name",
+            "nameserver_id",
+            "description",
+            name=_("Attributes"),
+        ),
+        FieldSet(
+            "soa_mname_id",
+            "soa_rname",
+            name=_("SOA"),
+        ),
+        FieldSet(
+            "record_template_id",
+            name=_("Record Templates"),
+        ),
+        FieldSet(
+            "dnssec_policy",
+            "parental_agents",
+            name=_("DNSSEC"),
+        ),
+        FieldSet(
+            "registrar_id",
+            "registrant_id",
+            "admin_c_id",
+            "tech_c_id",
+            "billing_c_id",
+            name=_("Registration"),
+        ),
+        FieldSet(
+            "tenant_group_id",
+            "tenant_id",
+            name=_("Tenancy"),
+        ),
+    )
+
+    name = forms.CharField(
+        required=False,
+        label=_("Template Name"),
+    )
+    nameserver_id = DynamicModelMultipleChoiceField(
+        queryset=NameServer.objects.all(),
+        required=False,
+        null_option=_("None"),
+        label=_("Nameservers"),
+    )
+    soa_mname_id = DynamicModelMultipleChoiceField(
+        queryset=NameServer.objects.all(),
+        required=False,
+        null_option=_("None"),
+        label=_("MName"),
+    )
+    soa_rname = forms.CharField(
+        required=False,
+        label=_("RName"),
+    )
+    record_template_id = DynamicModelMultipleChoiceField(
+        queryset=RecordTemplate.objects.all(),
+        required=False,
+        null_option=_("None"),
+        label=_("Record Templates"),
+    )
+    description = forms.CharField(
+        required=False,
+    )
+    dnssec_policy_id = DynamicModelMultipleChoiceField(
+        queryset=DNSSECPolicy.objects.all(),
+        required=False,
+        null_option=_("None"),
+        label=_("DNSSEC Policy ID"),
+    )
+    parental_agents = forms.GenericIPAddressField(
+        required=False,
+        label=_("Parental Agents"),
+    )
+    registrar_id = DynamicModelMultipleChoiceField(
+        queryset=Registrar.objects.all(),
+        required=False,
+        null_option=_("None"),
+        label=_("Registrar"),
+    )
+    registrant_id = DynamicModelMultipleChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        null_option=_("None"),
+        label=_("Registrant"),
+    )
+    admin_c_id = DynamicModelMultipleChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        null_option=_("None"),
+        label=_("Administrative Contact"),
+    )
+    tech_c_id = DynamicModelMultipleChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        null_option=_("None"),
+        label=_("Technical Contact"),
+    )
+    billing_c_id = DynamicModelMultipleChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        null_option=_("None"),
+        label=_("Billing Contact"),
+    )
+    tag = TagFilterField(ZoneTemplate)
+
+
+class ZoneTemplateImportForm(NetBoxModelImportForm):
+    class Meta:
+        model = ZoneTemplate
+
+        fields = (
+            "name",
+            "nameservers",
+            "soa_mname",
+            "soa_rname",
+            "record_templates",
+            "dnssec_policy",
+            "parental_agents",
+            "description",
+            "registrar",
+            "registrant",
+            "admin_c",
+            "tech_c",
+            "billing_c",
+            "tenant",
+            "tags",
+        )
+
+    nameservers = CSVModelMultipleChoiceField(
+        queryset=NameServer.objects.all(),
+        to_field_name="name",
+        required=False,
+        label=_("Nameservers"),
+    )
+    soa_mname = CSVModelChoiceField(
+        queryset=NameServer.objects.all(),
+        to_field_name="name",
+        required=False,
+        label=_("SOA MName"),
+    )
+    record_templates = CSVModelMultipleChoiceField(
+        queryset=RecordTemplate.objects.all(),
+        to_field_name="name",
+        required=False,
+        label=_("Record Templates"),
+    )
+    dnssec_policy = CSVModelChoiceField(
+        queryset=DNSSECPolicy.objects.all(),
+        required=False,
+        to_field_name="name",
+        error_messages={
+            "invalid_choice": _("DNSSEC policy %(value)s not found"),
+        },
+        label=_("DNSSEC Policy"),
+    )
+    registrar = CSVModelChoiceField(
+        queryset=Registrar.objects.all(),
+        required=False,
+        to_field_name="name",
+        error_messages={
+            "invalid_choice": _("Registrar %(value)s not found"),
+        },
+        label=_("Registrar"),
+    )
+    registrant = CSVModelChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        to_field_name="contact_id",
+        error_messages={
+            "invalid_choice": _("Registrant contact ID %(value)s not found"),
+        },
+        label=_("Registrant"),
+    )
+    admin_c = CSVModelChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        to_field_name="contact_id",
+        error_messages={
+            "invalid_choice": _("Administrative contact ID %(value)s not found"),
+        },
+        label=_("Administrative Contact"),
+    )
+    tech_c = CSVModelChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        to_field_name="contact_id",
+        error_messages={
+            "invalid_choice": _("Technical contact ID %(value)s not found"),
+        },
+        label=_("Technical Contact"),
+    )
+    billing_c = CSVModelChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        to_field_name="contact_id",
+        error_messages={
+            "invalid_choice": _("Billing contact ID %(value)s not found"),
+        },
+        label=_("Billing Contact"),
+    )
+    tenant = CSVModelChoiceField(
+        queryset=Tenant.objects.all(),
+        required=False,
+        to_field_name="name",
+        label=_("Tenant"),
+    )
+
+
+class ZoneTemplateBulkEditForm(NetBoxModelBulkEditForm):
+    model = ZoneTemplate
+
+    fieldsets = (
+        FieldSet(
+            "nameservers",
+            "description",
+            name=_("Attributes"),
+        ),
+        FieldSet(
+            "soa_mname",
+            "soa_rname",
+            name=_("SOA"),
+        ),
+        FieldSet(
+            "record_templates",
+            name=_("Record Templates"),
+        ),
+        FieldSet(
+            "dnssec_policy",
+            "parental_agents",
+            name=_("DNSSEC"),
+        ),
+        FieldSet(
+            "registrar",
+            "registrant",
+            "admin_c",
+            "tech_c",
+            "billing_c",
+            name=_("Domain Registration"),
+        ),
+        FieldSet(
+            "tenant_group",
+            "tenant",
+            name=_("Tenancy"),
+        ),
+    )
+
+    nullable_fields = (
+        "description",
+        "nameservers",
+        "soa_mname",
+        "soa_rname",
+        "record_templates",
+        "dnssec_policy",
+        "registrar",
+        "registrant",
+        "admin_c",
+        "tech_c",
+        "billing_c",
+        "tenant",
+    )
+
+    nameservers = DynamicModelMultipleChoiceField(
+        queryset=NameServer.objects.all(),
+        required=False,
+        label=_("Nameservers"),
+    )
+    soa_mname = DynamicModelChoiceField(
+        queryset=NameServer.objects.all(),
+        required=False,
+        label=_("MName"),
+    )
+    soa_rname = forms.CharField(
+        max_length=255,
+        required=False,
+        label=_("RName"),
+    )
+    record_templates = DynamicModelMultipleChoiceField(
+        queryset=RecordTemplate.objects.all(),
+        required=False,
+        label=_("Record Templates"),
+    )
+    description = forms.CharField(
+        max_length=200, required=False, label=_("Description")
+    )
+    dnssec_policy = DynamicModelChoiceField(
+        queryset=DNSSECPolicy.objects.all(),
+        required=False,
+        label=_("DNSSEC Policy"),
+    )
+    parental_agents = SimpleArrayField(
+        required=False,
+        base_field=forms.GenericIPAddressField(),
+        label=_("Parental Agents"),
+    )
+    registrar = DynamicModelChoiceField(
+        queryset=Registrar.objects.all(),
+        required=False,
+        label=_("Registrar"),
+    )
+    registrant = DynamicModelChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        label=_("Registrant"),
+    )
+    admin_c = DynamicModelChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        label=_("Administrative Contact"),
+    )
+    tech_c = DynamicModelChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        label=_("Technical Contact"),
+    )
+    billing_c = DynamicModelChoiceField(
+        queryset=RegistrationContact.objects.all(),
+        required=False,
+        label=_("Billing Contact"),
+    )
+    tenant_group = DynamicModelChoiceField(
+        queryset=TenantGroup.objects.all(),
+        required=False,
+        label=_("Tenant Group"),
+    )
+    tenant = DynamicModelChoiceField(
+        queryset=Tenant.objects.all(),
+        required=False,
+        label=_("Tenant"),
+    )

netbox_dns/graphql/__init__.py

@@ -1,22 +1,25 @@
-import graphene
-from graphene_django.converter import convert_django_field
-
-from netbox_dns.fields import NetworkField, AddressField
-
-
-@convert_django_field.register(NetworkField)
-def convert_field_to_string(field, registry=None):
-    return graphene.String(description=field.help_text, required=not field.null)
-
-
-@convert_django_field.register(AddressField)
-def convert_field_to_string(field, registry=None):
-    return graphene.String(description=field.help_text, required=not field.null)
-
-
-from .schema import *
-
-from .view import *
-from .zone import *
-from .nameserver import *
-from .record import *
+from .schema import (
+    NetBoxDNSNameServerQuery,
+    NetBoxDNSViewQuery,
+    NetBoxDNSZoneQuery,
+    NetBoxDNSRecordQuery,
+    NetBoxDNSDNSSECKeyTemplateQuery,
+    NetBoxDNSDNSSECPolicyQuery,
+    NetBoxDNSRegistrationContactQuery,
+    NetBoxDNSRegistrarQuery,
+    NetBoxDNSZoneTemplateQuery,
+    NetBoxDNSRecordTemplateQuery,
+)
+
+schema = [
+    NetBoxDNSNameServerQuery,
+    NetBoxDNSViewQuery,
+    NetBoxDNSZoneQuery,
+    NetBoxDNSRecordQuery,
+    NetBoxDNSDNSSECKeyTemplateQuery,
+    NetBoxDNSDNSSECPolicyQuery,
+    NetBoxDNSRegistrationContactQuery,
+    NetBoxDNSRegistrarQuery,
+    NetBoxDNSZoneTemplateQuery,
+    NetBoxDNSRecordTemplateQuery,
+]

netbox_dns/graphql/enums.py

@@ -0,0 +1,41 @@
+import strawberry
+
+from netbox_dns.choices import (
+    RecordTypeChoices,
+    RecordClassChoices,
+    RecordStatusChoices,
+    ZoneStatusChoices,
+    DNSSECPolicyDigestChoices,
+    DNSSECPolicyStatusChoices,
+    DNSSECKeyTemplateTypeChoices,
+    DNSSECKeyTemplateAlgorithmChoices,
+    DNSSECKeyTemplateKeySizeChoices,
+)
+
+__all__ = (
+    "NetBoxDNSRecordTypeEnum",
+    "NetBoxDNSRecordClassEnum",
+    "NetBoxDNSRecordStatusEnum",
+    "NetBoxDNSZoneStatusEnum",
+    "NetBoxDNSDNSSECPolicyDigestEnum",
+    "NetBoxDNSDNSSECPolicyStatusEnum",
+    "NetBoxDNSDNSSECKeyTemplateTypeEnum",
+    "NetBoxDNSDNSSECKeyTemplateAlgorithmEnum",
+    "NetBoxDNSDNSSECKeyTemplateKeySizeEnum",
+)
+
+NetBoxDNSRecordTypeEnum = strawberry.enum(RecordTypeChoices.as_enum())
+NetBoxDNSRecordClassEnum = strawberry.enum(RecordClassChoices.as_enum())
+NetBoxDNSRecordStatusEnum = strawberry.enum(RecordStatusChoices.as_enum())
+NetBoxDNSZoneStatusEnum = strawberry.enum(ZoneStatusChoices.as_enum())
+NetBoxDNSDNSSECPolicyDigestEnum = strawberry.enum(DNSSECPolicyDigestChoices.as_enum())
+NetBoxDNSDNSSECPolicyStatusEnum = strawberry.enum(DNSSECPolicyStatusChoices.as_enum())
+NetBoxDNSDNSSECKeyTemplateTypeEnum = strawberry.enum(
+    DNSSECKeyTemplateTypeChoices.as_enum()
+)
+NetBoxDNSDNSSECKeyTemplateAlgorithmEnum = strawberry.enum(
+    DNSSECKeyTemplateAlgorithmChoices.as_enum()
+)
+NetBoxDNSDNSSECKeyTemplateKeySizeEnum = strawberry.enum(
+    DNSSECKeyTemplateKeySizeChoices.as_enum()
+)

netbox_dns/graphql/filter_lookups.py

@@ -0,0 +1,13 @@
+import strawberry
+
+from netbox.graphql.filter_lookups import ArrayLookup
+
+from netbox_dns.graphql.enums import NetBoxDNSDNSSECPolicyDigestEnum
+
+
+@strawberry.input(
+    one_of=True,
+    description="Lookup for Array fields. Only one of the lookup fields can be set.",
+)
+class PolicyDigestArrayLookup(ArrayLookup[NetBoxDNSDNSSECPolicyDigestEnum]):
+    pass

netbox_dns/graphql/filters/__init__.py

@@ -0,0 +1,12 @@
+from .view import *
+from .zone import *
+from .nameserver import *
+from .record 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/graphql/filters/dnssec_key_template.py

@@ -0,0 +1,63 @@
+from typing import Annotated, TYPE_CHECKING
+
+import strawberry
+import strawberry_django
+from strawberry.scalars import ID
+from strawberry_django import FilterLookup
+
+from netbox.graphql.filter_mixins import NetBoxModelFilterMixin
+from tenancy.graphql.filter_mixins import ContactFilterMixin, TenancyFilterMixin
+
+if TYPE_CHECKING:
+    from netbox.graphql.filter_lookups import IntegerLookup
+    from .enums import (
+        NetBoxDNSDNSSECKeyTemplateTypeEnum,
+        NetBoxDNSDNSSECKeyTemplateAlgorithmEnum,
+        NetBoxDNSDNSSECKeyTemplateKeySizeEnum,
+    )
+    from .dnssec_policy import NetBoxDNSDNSSECPolicyFilter
+
+from netbox_dns.models import DNSSECKeyTemplate
+
+
+__all__ = ("NetBoxDNSDNSSECKeyTemplateFilter",)
+
+
+@strawberry_django.filter_type(DNSSECKeyTemplate, lookups=True)
+class NetBoxDNSDNSSECKeyTemplateFilter(
+    ContactFilterMixin, TenancyFilterMixin, NetBoxModelFilterMixin
+):
+    name: FilterLookup[str] | None = strawberry_django.filter_field()
+    description: FilterLookup[str] | None = strawberry_django.filter_field()
+    type: (
+        Annotated[
+            "NetBoxDNSDNSSECKeyTemplateTypeEnum",
+            strawberry.lazy("netbox_dns.graphql.enums"),
+        ]
+        | None
+    ) = strawberry_django.filter_field()
+    algorithm: (
+        Annotated[
+            "NetBoxDNSDNSSECKeyTemplateAlgorithmEnum",
+            strawberry.lazy("netbox_dns.graphql.enums"),
+        ]
+        | None
+    ) = strawberry_django.filter_field()
+    lifetime: (
+        Annotated["IntegerLookup", strawberry.lazy("netbox.graphql.filter_lookups")]
+        | None
+    ) = strawberry_django.filter_field()
+    key_size: (
+        Annotated[
+            "NetBoxDNSDNSSECKeyTemplateKeySizeEnum",
+            strawberry.lazy("netbox_dns.graphql.enums"),
+        ]
+        | None
+    ) = strawberry_django.filter_field()
+    policy: (
+        Annotated[
+            "NetBoxDNSDNSSECPolicyFilter", strawberry.lazy("netbox_dns.graphql.filters")
+        ]
+        | None
+    ) = strawberry_django.filter_field()
+    policy_id: ID | None = strawberry_django.filter_field()