upload

2026-07-10 09:15:15 +02:00
parent b80f501d5a
commit 9d5ea53b5c
18 changed files with 2122 additions and 0 deletions
@@ -0,0 +1,140 @@
+from django import forms
+from django.core.exceptions import ValidationError
+from django.utils.translation import gettext_lazy as _
+from dcim.models import Site
+from netbox.forms import NetBoxModelFilterSetForm, NetBoxModelForm
+from tenancy.models import Tenant
+from utilities.forms.fields import (
+    DynamicModelChoiceField,
+    DynamicModelMultipleChoiceField,
+    MultipleChoiceField,
+    SlugField,
+)
+from utilities.forms.rendering import FieldSet
+from virtualization.models import Cluster
+
+from .choices import SyncStatusChoices
+from .models import VCenterEndpoint
+
+
+class VCenterEndpointForm(NetBoxModelForm):
+    slug = SlugField()
+    tenant = DynamicModelChoiceField(
+        queryset=Tenant.objects.all(),
+        required=False,
+    )
+    site = DynamicModelChoiceField(
+        queryset=Site.objects.all(),
+        required=False,
+    )
+    cluster = DynamicModelChoiceField(
+        queryset=Cluster.objects.all(),
+    )
+    password = forms.CharField(
+        label=_("Password"),
+        required=False,
+        widget=forms.PasswordInput(render_value=False),
+        help_text=_("Leave blank to keep the currently stored password."),
+    )
+
+    fieldsets = (
+        FieldSet("name", "slug", "enabled", "comments", name=_("Endpoint")),
+        FieldSet("host", "port", "username", "password", "validate_ssl", name=_("VMware connection")),
+        FieldSet("tenant", "site", "cluster", name=_("NetBox target")),
+        FieldSet(
+            "include_name_regex",
+            "exclude_name_regex",
+            "sync_powered_off",
+            "sync_interfaces",
+            "sync_ip_addresses",
+            "sync_primary_ips",
+            "update_existing",
+            name=_("Sync behavior"),
+        ),
+        FieldSet("default_ipv4_prefix_length", "default_ipv6_prefix_length", "sync_interval_minutes", name=_("Automation")),
+        FieldSet("tags", name=_("Tags")),
+    )
+
+    class Meta:
+        model = VCenterEndpoint
+        fields = (
+            "name",
+            "slug",
+            "enabled",
+            "host",
+            "port",
+            "username",
+            "password",
+            "validate_ssl",
+            "tenant",
+            "site",
+            "cluster",
+            "include_name_regex",
+            "exclude_name_regex",
+            "sync_powered_off",
+            "sync_interfaces",
+            "sync_ip_addresses",
+            "sync_primary_ips",
+            "update_existing",
+            "default_ipv4_prefix_length",
+            "default_ipv6_prefix_length",
+            "sync_interval_minutes",
+            "comments",
+            "tags",
+        )
+
+    def clean(self):
+        cleaned_data = super().clean()
+        password = cleaned_data.get("password")
+
+        if not self.instance.pk and not password:
+            raise ValidationError({"password": _("A password is required for new endpoints.")})
+
+        return cleaned_data
+
+    def save(self, commit=True):
+        obj = super().save(commit=False)
+        password = self.cleaned_data.get("password")
+
+        if password:
+            obj.set_password(password)
+
+        if commit:
+            obj.save()
+            self.save_m2m()
+
+        return obj
+
+
+class VCenterEndpointFilterForm(NetBoxModelFilterSetForm):
+    model = VCenterEndpoint
+
+    enabled = forms.NullBooleanField(
+        required=False,
+        label=_("Enabled"),
+    )
+
+    tenant_id = DynamicModelMultipleChoiceField(
+        queryset=Tenant.objects.all(),
+        required=False,
+        label=_("Tenant"),
+    )
+    site_id = DynamicModelMultipleChoiceField(
+        queryset=Site.objects.all(),
+        required=False,
+        label=_("Site"),
+    )
+    cluster_id = DynamicModelMultipleChoiceField(
+        queryset=Cluster.objects.all(),
+        required=False,
+        label=_("Cluster"),
+    )
+    last_status = MultipleChoiceField(
+        choices=SyncStatusChoices,
+        required=False,
+        label=_("Last status"),
+    )
+
+    fieldsets = (
+        FieldSet("q", "enabled", "tenant_id", "site_id", "cluster_id", "last_status", name=_("Endpoint")),
+    )