normalized project structure

2022-04-22 13:52:26 +02:00
parent 7a24221444
commit 9d76bb421d
26 changed files with 0 additions and 13 deletions
@@ -0,0 +1,55 @@
+from rest_framework import serializers
+
+from netbox.api.serializers import NetBoxModelSerializer
+from netbox_slm.models import (
+    SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation,
+)
+
+
+class SoftwareProductSerializer(NetBoxModelSerializer):
+    display = serializers.SerializerMethodField()
+    url = serializers.HyperlinkedIdentityField(
+        view_name="plugins-api:netbox_slm-api:softwareproduct-detail"
+    )
+
+    class Meta:
+        model = SoftwareProduct
+        fields = [
+            'id', 'display', 'url', 'name', 'tags', 'custom_field_data', 'created', 'last_updated',
+        ]
+
+    def get_display(self, obj):
+        return f"{obj.manufacturer.name} - {obj.name}"
+
+
+class SoftwareProductVersionSerializer(NetBoxModelSerializer):
+    display = serializers.SerializerMethodField()
+    url = serializers.HyperlinkedIdentityField(
+        view_name="plugins-api:netbox_slm-api:softwareproductversion-detail"
+    )
+
+    class Meta:
+        model = SoftwareProductVersion
+        fields = [
+            'id', 'display', 'url', 'name', 'software_product', 'tags', 'custom_field_data', 'created', 'last_updated',
+        ]
+
+    def get_display(self, obj):
+        return obj.name
+
+
+class SoftwareProductInstallationSerializer(NetBoxModelSerializer):
+    display = serializers.SerializerMethodField()
+    url = serializers.HyperlinkedIdentityField(
+        view_name="plugins-api:netbox_slm-api:softwareproductinstallation-detail"
+    )
+
+    class Meta:
+        model = SoftwareProductInstallation
+        fields = [
+            'id', 'display', 'url', 'device', 'virtualmachine', 'software_product', 'version', 'tags',
+            'custom_field_data', 'created', 'last_updated',
+        ]
+
+    def get_display(self, obj):
+        return obj
@@ -0,0 +1,15 @@
+from netbox.api.routers import NetBoxRouter
+from netbox_slm.api.views import (
+    NetboxSLMRootView,
+    SoftwareProductViewSet,
+    SoftwareProductVersionViewSet,
+    SoftwareProductInstallationViewSet,
+)
+
+router = NetBoxRouter()
+router.APIRootView = NetboxSLMRootView
+
+router.register("softwareproducts", SoftwareProductViewSet)
+router.register("softwareproductversions", SoftwareProductVersionViewSet)
+router.register("softwareproductinstallations", SoftwareProductInstallationViewSet)
+urlpatterns = router.urls
@@ -0,0 +1,39 @@
+from netbox.api.viewsets import NetBoxModelViewSet
+from rest_framework.routers import APIRootView
+
+from netbox_slm.api.serializers import (
+    SoftwareProductSerializer, SoftwareProductVersionSerializer, SoftwareProductInstallationSerializer,
+)
+from netbox_slm.filtersets import (
+    SoftwareProductFilterSet, SoftwareProductVersionFilterSet, SoftwareProductInstallationFilterSet,
+)
+from netbox_slm.models import (
+    SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation,
+)
+
+
+class NetboxSLMRootView(APIRootView):
+    """
+    NetboxSLM API root view
+    """
+
+    def get_view_name(self):
+        return "NetboxSLM"
+
+
+class SoftwareProductViewSet(NetBoxModelViewSet):
+    queryset = SoftwareProduct.objects.all()
+    serializer_class = SoftwareProductSerializer
+    filterset_class = SoftwareProductFilterSet
+
+
+class SoftwareProductVersionViewSet(NetBoxModelViewSet):
+    queryset = SoftwareProductVersion.objects.all()
+    serializer_class = SoftwareProductVersionSerializer
+    filterset_class = SoftwareProductVersionFilterSet
+
+
+class SoftwareProductInstallationViewSet(NetBoxModelViewSet):
+    queryset = SoftwareProductInstallation.objects.all()
+    serializer_class = SoftwareProductInstallationSerializer
+    filterset_class = SoftwareProductInstallationFilterSet