Add SoftwareLicense (#20)

netbox_slm/api/serializers.py

@@ -1,7 +1,7 @@
 from rest_framework import serializers
 
 from netbox.api.serializers import NetBoxModelSerializer
-from netbox_slm.models import SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation
+from netbox_slm.models import SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation, SoftwareLicense
 
 
 class SoftwareProductSerializer(NetBoxModelSerializer):
@@ -13,7 +13,8 @@ class SoftwareProductSerializer(NetBoxModelSerializer):
     class Meta:
         model = SoftwareProduct
         fields = [
-            'id', 'display', 'url', 'name', 'tags', 'custom_field_data', 'created', 'last_updated',
+            'id', 'display', 'url', 'name',
+            'tags', 'custom_field_data', 'created', 'last_updated',
         ]
 
     def get_display(self, obj):
@@ -29,7 +30,8 @@ class SoftwareProductVersionSerializer(NetBoxModelSerializer):
     class Meta:
         model = SoftwareProductVersion
         fields = [
-            'id', 'display', 'url', 'name', 'software_product', 'tags', 'custom_field_data', 'created', 'last_updated',
+            'id', 'display', 'url', 'name', 'software_product',
+            'tags', 'custom_field_data', 'created', 'last_updated',
         ]
 
     def get_display(self, obj):
@@ -45,8 +47,26 @@ class SoftwareProductInstallationSerializer(NetBoxModelSerializer):
     class Meta:
         model = SoftwareProductInstallation
         fields = [
-            'id', 'display', 'url', 'device', 'virtualmachine', 'software_product', 'version', 'tags',
-            'custom_field_data', 'created', 'last_updated',
+            'id', 'display', 'url', 'device', 'virtualmachine', 'software_product', 'version',
+            'tags', 'custom_field_data', 'created', 'last_updated',
+        ]
+
+    def get_display(self, obj):
+        return f"{obj}"
+
+
+class SoftwareLicenseSerializer(NetBoxModelSerializer):
+    display = serializers.SerializerMethodField()
+    url = serializers.HyperlinkedIdentityField(
+        view_name="plugins-api:netbox_slm-api:softwarelicense-detail"
+    )
+
+    class Meta:
+        model = SoftwareLicense
+        fields = [
+            'id', 'display', 'url', 'name', 'description', 'type', 'stored_location',
+            'start_date', 'expiration_date', 'software_product', 'version', 'installation',
+            'tags', 'custom_field_data', 'created', 'last_updated',
         ]
 
     def get_display(self, obj):

netbox_slm/api/urls.py

@@ -4,6 +4,7 @@ from netbox_slm.api.views import (
     SoftwareProductViewSet,
     SoftwareProductVersionViewSet,
     SoftwareProductInstallationViewSet,
+    SoftwareLicenseViewSet,
 )
 
 router = NetBoxRouter()
@@ -12,4 +13,5 @@ router.APIRootView = NetboxSLMRootView
 router.register("softwareproducts", SoftwareProductViewSet)
 router.register("softwareproductversions", SoftwareProductVersionViewSet)
 router.register("softwareproductinstallations", SoftwareProductInstallationViewSet)
+router.register("softwarelicenses", SoftwareLicenseViewSet)
 urlpatterns = router.urls

netbox_slm/api/views.py

@@ -2,12 +2,18 @@ from rest_framework.routers import APIRootView
 
 from netbox.api.viewsets import NetBoxModelViewSet
 from netbox_slm.api.serializers import (
-    SoftwareProductSerializer, SoftwareProductVersionSerializer, SoftwareProductInstallationSerializer,
+    SoftwareProductSerializer,
+    SoftwareProductVersionSerializer,
+    SoftwareProductInstallationSerializer,
+    SoftwareLicenseSerializer,
 )
 from netbox_slm.filtersets import (
-    SoftwareProductFilterSet, SoftwareProductVersionFilterSet, SoftwareProductInstallationFilterSet,
+    SoftwareProductFilterSet,
+    SoftwareProductVersionFilterSet,
+    SoftwareProductInstallationFilterSet,
+    SoftwareLicenseFilterSet,
 )
-from netbox_slm.models import SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation
+from netbox_slm.models import SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation, SoftwareLicense
 
 
 class NetboxSLMRootView(APIRootView):
@@ -35,3 +41,9 @@ class SoftwareProductInstallationViewSet(NetBoxModelViewSet):
     queryset = SoftwareProductInstallation.objects.all()
     serializer_class = SoftwareProductInstallationSerializer
     filterset_class = SoftwareProductInstallationFilterSet
+
+
+class SoftwareLicenseViewSet(NetBoxModelViewSet):
+    queryset = SoftwareLicense.objects.all()
+    serializer_class = SoftwareLicenseSerializer
+    filterset_class = SoftwareLicenseFilterSet