infrastructure-manager-sync 0.5.5__py3-none-any.whl

infrastructure_manager_sync/__init__.py

@@ -0,0 +1,14 @@
+from netbox.plugins import PluginConfig
+
+
+class InfrastructureManagerSync(PluginConfig):
+    name = "infrastructure_manager_sync"
+    verbose_name = "Infrastructure Manager Sync"
+    description = "Plugin that configures the vcenter import for VM, clusters & hosts"
+    version = "0.5.5"
+    author = "Mattijs Vanhaverbeke"
+    base_url = "infrastructure-manager-sync"
+    min_version = "3.4.0"
+
+
+config = InfrastructureManagerSync

infrastructure_manager_sync/api/serializers.py

@@ -0,0 +1,89 @@
+from rest_framework import serializers
+
+from netbox.api.serializers import NetBoxModelSerializer
+from ..models import (
+    InfrastructureManagerSync,
+    InfrastructureManagerSyncVMInfo,
+    InfrastructureManagerSyncHostInfo,
+)
+
+
+#
+# Regular serializers
+#
+
+
+class InfrastructureManagerSyncSerializer(NetBoxModelSerializer):
+    url = serializers.HyperlinkedIdentityField(
+        view_name="plugins-api:infrastructure_manager_sync-api:infrastructuremanagersync-detail"
+    )
+
+    class Meta:
+        model = InfrastructureManagerSync
+        fields = (
+            "id",
+            "fqdn",
+            "name",
+            "username",
+            "password",
+            "cluster_tenant",
+            "entry_type",
+            "url",
+            "primary_site",
+            "enabled",
+            "build_number",
+            "version",
+            "last_successful_sync",
+            "update_prio",
+            "assign_by_default_to_cluster_tenant",
+        )
+
+
+class InfrastructureManagerSyncVMInfoSerializer(NetBoxModelSerializer):
+    url = serializers.HyperlinkedIdentityField(
+        view_name="plugins-api:infrastructure_manager_sync-api:infrastructuremanagersyncvminfo-detail"
+    )
+
+    class Meta:
+        model = InfrastructureManagerSyncVMInfo
+        fields = (
+            "id",
+            "vm",
+            "backup_plan",
+            "backup_type",
+            "criticality",
+            "environment",
+            "financial_info",
+            "licensing",
+            "owner",
+            "service_info",
+            "url",
+            "ims",
+            "backup_status",
+            "last_backup",
+            "deployed_on",
+            "deployed_by",
+            "billing_reference",
+            "vmware_tools_version",
+            "vm_hardware_compatibility",
+        )
+
+
+class InfrastructureManagerSyncHostInfoSerializer(NetBoxModelSerializer):
+    url = serializers.HyperlinkedIdentityField(
+        view_name="plugins-api:infrastructure_manager_sync-api:infrastructuremanagersynchostinfo-detail"
+    )
+
+    class Meta:
+        model = InfrastructureManagerSyncHostInfo
+        fields = (
+            "id",
+            "host",
+            "url",
+            "ims",
+            "physical_cpu_count",
+            "core_per_cpu",
+            "cpu_type",
+            "build_number",
+            "memory",
+        )

infrastructure_manager_sync/api/urls.py

@@ -0,0 +1,16 @@
+from netbox.api.routers import NetBoxRouter
+from . import views
+
+
+app_name = "infrastructure_manager_sync"
+
+router = NetBoxRouter()
+router.register("infrastructuremanagersync", views.InfrastructureManagerSyncViewSet)
+router.register(
+    "infrastructuremanagersyncvminfo", views.InfrastructureManagerSyncVMInfoViewSet
+)
+router.register(
+    "infrastructuremanagersynchostinfo", views.InfrastructureManagerSyncHostInfoViewSet
+)
+
+urlpatterns = router.urls

infrastructure_manager_sync/api/views.py

@@ -0,0 +1,26 @@
+from netbox.api.viewsets import NetBoxModelViewSet
+
+from .. import filtersets, models
+from .serializers import (
+    InfrastructureManagerSyncSerializer,
+    InfrastructureManagerSyncVMInfoSerializer,
+    InfrastructureManagerSyncHostInfoSerializer,
+)
+
+
+class InfrastructureManagerSyncViewSet(NetBoxModelViewSet):
+    # permission_required = 'netbox_vcenter_tag_sync.view_vcenters' #TODO
+    queryset = models.InfrastructureManagerSync.objects.all()
+    serializer_class = InfrastructureManagerSyncSerializer
+
+
+class InfrastructureManagerSyncVMInfoViewSet(NetBoxModelViewSet):
+    # permission_required = 'netbox_vcenter_tag_sync.view_vcenters' #TODO
+    queryset = models.InfrastructureManagerSyncVMInfo.objects.all()
+    serializer_class = InfrastructureManagerSyncVMInfoSerializer
+
+
+class InfrastructureManagerSyncHostInfoViewSet(NetBoxModelViewSet):
+    # permission_required = 'netbox_vcenter_tag_sync.view_vcenters' #TODO
+    queryset = models.InfrastructureManagerSyncHostInfo.objects.all()
+    serializer_class = InfrastructureManagerSyncHostInfoSerializer

infrastructure_manager_sync/filtersets.py

@@ -0,0 +1,108 @@
+import django_filters
+from netbox.filtersets import NetBoxModelFilterSet
+from .models import (
+    InfrastructureManagerSync,
+    InfrastructureManagerSyncVMInfo,
+    InfrastructureManagerSyncHostInfo,
+)
+from tenancy.models import Tenant
+from virtualization.models import Cluster
+from django.db.models import Q
+
+
+class InfrastructureManagerSyncFilterSet(NetBoxModelFilterSet):
+    class Meta:
+        model = InfrastructureManagerSync
+        fields = (
+            "name",
+            "fqdn",
+            "username",
+            "primary_site_id",
+            "enabled",
+            "update_prio",
+            "cluster_tenant_id",
+        )
+
+    assign_by_default_to_cluster_tenant = django_filters.BooleanFilter(
+        label="Assigned tenant by default",
+    )
+
+    enabled = django_filters.BooleanFilter(
+        label="Enabled",
+    )
+
+    def search(self, queryset, name, value):
+        """Perform the filtered search."""
+        if not value.strip():
+            return queryset
+        qs_filter = Q(name__icontains=value) | Q(fqdn__icontains=value)
+        return queryset.filter(qs_filter)
+
+
+class InfrastructureManagerSyncVMInfoFilterSet(NetBoxModelFilterSet):
+
+    tenant_id = django_filters.ModelMultipleChoiceFilter(
+        field_name="vm__tenant",
+        queryset=Tenant.objects.all(),
+        label="Tenant (ID)",
+    )
+
+    cluster_id = django_filters.ModelMultipleChoiceFilter(
+        field_name="vm__cluster",
+        queryset=Cluster.objects.all(),
+        label="Cluster (ID)",
+    )
+
+    def search(self, queryset, name, value):
+        """Perform the filtered search."""
+        if not value.strip():
+            return queryset
+        qs_filter = Q(vm__name__icontains=value)
+        return queryset.filter(qs_filter)
+
+    class Meta:
+        model = InfrastructureManagerSyncVMInfo
+        fields = (
+            "environment",
+            "criticality",
+            "financial_info",
+            "licensing",
+            "owner",
+            "backup_status",
+            "backup_plan",
+            "backup_type",
+            "deployed_by",
+            "billing_reference",
+            "ims",
+            "vmware_tools_version",
+            "vm_hardware_compatibility",
+        )
+
+
+class InfrastructureManagerSyncHostInfoFilterSet(NetBoxModelFilterSet):
+    class Meta:
+        model = InfrastructureManagerSyncHostInfo
+        fields = (
+            "memory",
+            "build_number",
+            "ims",
+        )
+
+    tenant_id = django_filters.ModelMultipleChoiceFilter(
+        field_name="host__tenant",
+        queryset=Tenant.objects.all(),
+        label="Tenant (ID)",
+    )
+
+    cluster_id = django_filters.ModelMultipleChoiceFilter(
+        field_name="host__cluster",
+        queryset=Cluster.objects.all(),
+        label="Cluster (ID)",
+    )
+
+    def search(self, queryset, name, value):
+        """Perform the filtered search."""
+        if not value.strip():
+            return queryset
+        qs_filter = Q(host__name__icontains=value)
+        return queryset.filter(qs_filter)

infrastructure_manager_sync/forms.py

@@ -0,0 +1,204 @@
+from django import forms
+
+from netbox.forms import NetBoxModelForm, NetBoxModelFilterSetForm
+from utilities.forms.fields import (
+    CommentField,
+    TagFilterField,
+    DynamicModelMultipleChoiceField,
+)
+from .models import (
+    CriticalityChoise,
+    EnvironmentChoise,
+    FinancialInfoChoise,
+    InfrastructureManagerSync,
+    InfrastructureManagerSyncVMInfo,
+    LicensingChoise,
+    SyncType,
+    UpdatePrioChoise,
+    InfrastructureManagerSyncHostInfo,
+)
+from utilities.forms.widgets import DatePicker, DateTimePicker
+from utilities.forms import BOOLEAN_WITH_BLANK_CHOICES, add_blank_choice
+from django.utils.translation import gettext as _
+from tenancy.models import Tenant
+from virtualization.models import Cluster
+from utilities.forms.rendering import FieldSet
+
+
+class InfrastructureManagerSyncForm(NetBoxModelForm):
+    comments = CommentField()
+
+    assign_by_default_to_cluster_tenant = forms.BooleanField(
+        label=("Assign untagged vms to cluster tenant"),
+        required=False,
+        initial=False,
+        help_text=_(
+            "Assign VMs without tenant tag automatically to the Cluster tenant"
+        ),
+    )
+
+    class Meta:
+        model = InfrastructureManagerSync
+        fields = (
+            "name",
+            "fqdn",
+            "username",
+            "password",
+            "cluster_tenant",
+            "entry_type",
+            "primary_site",
+            "enabled",
+            "update_prio",
+            "assign_by_default_to_cluster_tenant",
+            "comments",
+            "tags",
+        )
+
+
+class InfrastructureManagerSyncFilterForm(NetBoxModelFilterSetForm):
+    model = InfrastructureManagerSync
+
+    fieldsets = (
+        FieldSet("q", "filter_id", "tag"),
+        FieldSet(
+            "entry_type",
+            "update_prio",
+            "assign_by_default_to_cluster_tenant",
+            "enabled",
+            name=_("IMS"),
+        ),
+        FieldSet(
+            "cluster_tenant_id",
+            name=_("Tenancy"),
+        ),
+    )
+
+    entry_type = forms.MultipleChoiceField(choices=SyncType, required=False)
+    update_prio = forms.MultipleChoiceField(choices=UpdatePrioChoise, required=False)
+
+    assign_by_default_to_cluster_tenant = forms.NullBooleanField(
+        required=False,
+        label=_("Assign default tenant"),
+        widget=forms.Select(choices=BOOLEAN_WITH_BLANK_CHOICES),
+    )
+
+    enabled = forms.NullBooleanField(
+        required=False,
+        label=_("Enabled"),
+        widget=forms.Select(choices=BOOLEAN_WITH_BLANK_CHOICES),
+    )
+
+    cluster_tenant_id = DynamicModelMultipleChoiceField(
+        queryset=Tenant.objects.all(),
+        required=False,
+        null_option="None",
+        label=_("Cluster Tenant"),
+    )
+
+    tag = TagFilterField(model)
+
+
+class InfrastructureManagerSyncVMInfoSyncForm(NetBoxModelForm):
+    class Meta:
+        model = InfrastructureManagerSyncVMInfo
+        fields = (
+            "vm",
+            "backup_plan",
+            "backup_type",
+            "criticality",
+            "environment",
+            "financial_info",
+            "licensing",
+            "owner",
+            "service_info",
+            "backup_status",
+            "last_backup",
+            "deployed_on",
+            "deployed_by",
+            "billing_reference",
+            "ims",
+            "vmware_tools_version",
+            "vm_hardware_compatibility",
+        )
+        widgets = {
+            "deployed_on": DatePicker(),
+            "last_backup": DateTimePicker(),
+        }
+
+
+class InfrastructureManagerSyncVMInfoFilterForm(NetBoxModelFilterSetForm):
+    model = InfrastructureManagerSyncVMInfo
+
+    fieldsets = (
+        FieldSet("q", "filter_id", "tag"),
+        FieldSet(
+            "environment",
+            "criticality",
+            "financial_info",
+            "licensing",
+            "ims",
+            "owner",
+            "backup_plan",
+            "backup_type",
+            name=_("Attributes"),
+        ),
+        FieldSet("tenant_id", name=_("Tenant")),
+        FieldSet("cluster_id", name=_("Cluster")),
+    )
+
+    environment = forms.MultipleChoiceField(choices=EnvironmentChoise, required=False)
+    criticality = forms.MultipleChoiceField(choices=CriticalityChoise, required=False)
+    financial_info = forms.MultipleChoiceField(
+        choices=FinancialInfoChoise, required=False
+    )
+    licensing = forms.MultipleChoiceField(choices=LicensingChoise, required=False)
+
+    ims = forms.ModelChoiceField(
+        queryset=InfrastructureManagerSync.objects.all(),
+        required=False,
+        label=_("ims"),
+    )
+
+    owner = forms.CharField(max_length=100, required=False)
+
+    backup_plan = forms.CharField(max_length=100, required=False)
+
+    backup_type = forms.CharField(max_length=100, required=False)
+
+    tenant_id = DynamicModelMultipleChoiceField(
+        queryset=Tenant.objects.all(), required=False, label=_("Tenants")
+    )
+    cluster_id = DynamicModelMultipleChoiceField(
+        queryset=Cluster.objects.all(), required=False, label=_("Clusters")
+    )
+
+
+class InfrastructureManagerSyncHostInfoSyncForm(NetBoxModelForm):
+    class Meta:
+        model = InfrastructureManagerSyncHostInfo
+        fields = ("host", "build_number")
+
+
+class InfrastructureManagerSyncHostInfoFilterForm(NetBoxModelFilterSetForm):
+    model = InfrastructureManagerSyncHostInfo
+
+    fieldsets = (
+        FieldSet("q", "filter_id", "tag"),
+        FieldSet("ims", name=_("Attributes")),
+        FieldSet("tenant_id", name=_("Tenant")),
+        FieldSet("cluster_id", name=_("Cluster")),
+    )
+
+    ims = forms.ModelChoiceField(
+        queryset=InfrastructureManagerSync.objects.all(),
+        required=False,
+        label=_("ims"),
+    )
+
+    tenant_id = DynamicModelMultipleChoiceField(
+        queryset=Tenant.objects.all(), required=False, label=_("Tenants")
+    )
+
+    cluster_id = DynamicModelMultipleChoiceField(
+        queryset=Cluster.objects.all(), required=False, label=_("Clusters")
+    )

infrastructure_manager_sync/graphql.py

@@ -0,0 +1,36 @@
+from typing import List
+import strawberry
+import strawberry_django
+
+from . import models
+
+
+@strawberry_django.type(
+    models.InfrastructureManagerSync,
+    fields=["name",
+            "fqdn",
+            "cluster_tenant",
+            "entry_type",
+            "primary_site",
+            "enabled",
+            "update_prio",
+            "assign_by_default_to_cluster_tenant",
+            "last_successful_sync",
+            "build_number",
+            "version",
+            "id"
+        ],
+)
+class InfrastructureManagerSyncType:
+    pass
+
+
+@strawberry.type(name="Query")
+class DummyQuery:
+    infrastructure_manager_sync: InfrastructureManagerSyncType = strawberry_django.field()
+    infrastructure_manager_sync_list: List[InfrastructureManagerSyncType] = strawberry_django.field()
+
+
+schema = [
+    DummyQuery,
+]

infrastructure_manager_sync/migrations/0001_initial.py

@@ -0,0 +1,123 @@
+# Generated by Django 4.1.8 on 2023-05-12 13:20
+
+from django.db import migrations, models
+import django.db.models.deletion
+import taggit.managers
+import utilities.json
+
+
+class Migration(migrations.Migration):
+    initial = True
+
+    dependencies = [
+        ("virtualization", "0034_standardize_description_comments"),
+        ("extras", "0092_delete_jobresult"),
+        ("tenancy", "0010_tenant_relax_uniqueness"),
+        ("dcim", "0171_cabletermination_change_logging"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="InfrastructureManagerSyncVMInfo",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True, primary_key=True, serialize=False
+                    ),
+                ),
+                ("created", models.DateTimeField(auto_now_add=True, null=True)),
+                ("last_updated", models.DateTimeField(auto_now=True, null=True)),
+                (
+                    "custom_field_data",
+                    models.JSONField(
+                        blank=True,
+                        default=dict,
+                        encoder=utilities.json.CustomFieldJSONEncoder,
+                    ),
+                ),
+                ("backup_plan", models.CharField(blank=True, max_length=100)),
+                ("backup_type", models.CharField(blank=True, max_length=100)),
+                ("criticality", models.CharField(blank=True, max_length=30)),
+                ("environment", models.CharField(blank=True, max_length=30)),
+                ("financial_info", models.CharField(blank=True, max_length=30)),
+                ("licensing", models.CharField(blank=True, max_length=30)),
+                ("owner", models.CharField(blank=True, max_length=100)),
+                ("service_info", models.CharField(blank=True, max_length=30)),
+                (
+                    "tags",
+                    taggit.managers.TaggableManager(
+                        through="extras.TaggedItem", to="extras.Tag"
+                    ),
+                ),
+                (
+                    "vm",
+                    models.OneToOneField(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="InfrastructureManagerSyncVMInfo",
+                        to="virtualization.virtualmachine",
+                    ),
+                ),
+            ],
+            options={
+                "ordering": ("pk",),
+            },
+        ),
+        migrations.CreateModel(
+            name="InfrastructureManagerSync",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True, primary_key=True, serialize=False
+                    ),
+                ),
+                ("created", models.DateTimeField(auto_now_add=True, null=True)),
+                ("last_updated", models.DateTimeField(auto_now=True, null=True)),
+                (
+                    "custom_field_data",
+                    models.JSONField(
+                        blank=True,
+                        default=dict,
+                        encoder=utilities.json.CustomFieldJSONEncoder,
+                    ),
+                ),
+                ("name", models.CharField(max_length=100)),
+                ("fqdn", models.CharField(max_length=100)),
+                ("username", models.CharField(max_length=100)),
+                ("password", models.CharField(default="ChangeMe", max_length=100)),
+                ("entry_type", models.CharField(blank=True, max_length=30)),
+                ("enabled", models.BooleanField(default=True)),
+                ("comments", models.TextField(blank=True)),
+                (
+                    "cluster_tenant",
+                    models.ForeignKey(
+                        blank=True,
+                        null=True,
+                        on_delete=django.db.models.deletion.PROTECT,
+                        related_name="infrastructuremanagersync",
+                        to="tenancy.tenant",
+                    ),
+                ),
+                (
+                    "primary_site",
+                    models.ForeignKey(
+                        blank=True,
+                        null=True,
+                        on_delete=django.db.models.deletion.PROTECT,
+                        related_name="infrastructuremanagersync_sync",
+                        to="dcim.site",
+                    ),
+                ),
+                (
+                    "tags",
+                    taggit.managers.TaggableManager(
+                        through="extras.TaggedItem", to="extras.Tag"
+                    ),
+                ),
+            ],
+            options={
+                "ordering": ("name",),
+            },
+        ),
+    ]

infrastructure_manager_sync/migrations/0002_infrastructuremanagersyncvminfo_ims.py

@@ -0,0 +1,23 @@
+# Generated by Django 4.1.8 on 2023-06-19 08:39
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("infrastructure_manager_sync", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="infrastructuremanagersyncvminfo",
+            name="ims",
+            field=models.OneToOneField(
+                null=True,
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name="InfrastructureManagerSync",
+                to="infrastructure_manager_sync.infrastructuremanagersync",
+            ),
+        ),
+    ]

infrastructure_manager_sync/migrations/0003_alter_infrastructuremanagersyncvminfo_ims.py

@@ -0,0 +1,23 @@
+# Generated by Django 4.1.8 on 2023-06-19 09:27
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("infrastructure_manager_sync", "0002_infrastructuremanagersyncvminfo_ims"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="infrastructuremanagersyncvminfo",
+            name="ims",
+            field=models.ForeignKey(
+                null=True,
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name="InfrastructureManagerSync",
+                to="infrastructure_manager_sync.infrastructuremanagersync",
+            ),
+        ),
+    ]