updated api, filterset and graphql by ownership_type, added method get color to ownership type

Author: jirivrany

cesnet_service_path_plugin/api/serializers/segment.py

@@ -55,6 +55,7 @@ class Meta:
             "name",
             "segment_type",
             "status",
+            "ownership_type",
             "network_label",
             "install_date",
             "termination_date",
@@ -82,6 +83,7 @@ class Meta:
             "name",
             "segment_type",
             "status",
+            "ownership_type",
             "has_path_data",
             "tags",
         )
@@ -197,6 +199,7 @@ class Meta:
             "name",
             "segment_type",
             "status",
+            "ownership_type",
             "network_label",
             "install_date",
             "termination_date",
@@ -225,6 +228,7 @@ class Meta:
             "name",
             "segment_type",
             "status",
+            "ownership_type",
             "has_path_data",
             "tags",
         )

cesnet_service_path_plugin/filtersets/segment.py

@@ -8,7 +8,7 @@
 from netbox.filtersets import NetBoxModelFilterSet
 
 from cesnet_service_path_plugin.models import Segment
-from cesnet_service_path_plugin.models.custom_choices import StatusChoices
+from cesnet_service_path_plugin.models.custom_choices import StatusChoices, OwnershipTypeChoices
 from cesnet_service_path_plugin.models.segment_types import SegmentTypeChoices
 
 logger = logging.getLogger(__name__)
@@ -23,6 +23,7 @@ class SegmentFilterSet(NetBoxModelFilterSet):
     name = django_filters.CharFilter(lookup_expr="icontains")
     network_label = django_filters.CharFilter(lookup_expr="icontains")
     status = django_filters.MultipleChoiceFilter(choices=StatusChoices, null_value=None)
+    ownership_type = django_filters.MultipleChoiceFilter(choices=OwnershipTypeChoices, null_value=None)
 
     # Basic segment type filter
     segment_type = django_filters.MultipleChoiceFilter(
@@ -521,6 +522,7 @@ def search(self, queryset, name, value):
         network_label = Q(network_label__icontains=value)
         provider_segment_id = Q(provider_segment_id__icontains=value)
         status = Q(status__iexact=value)
+        ownership_type = Q(ownership_type__iexact=value)
         segment_type = Q(segment_type__iexact=value)
 
         return queryset.filter(
@@ -532,5 +534,6 @@ def search(self, queryset, name, value):
             | network_label
             | provider_segment_id
             | status
+            | ownership_type
             | segment_type
         )

cesnet_service_path_plugin/graphql/filters.py

@@ -39,6 +39,7 @@ class SegmentFilter(NetBoxModelFilterMixin):
     install_date: FilterLookup[str] | None = strawberry_django.filter_field()  # Date fields as string
     termination_date: FilterLookup[str] | None = strawberry_django.filter_field()
     status: FilterLookup[str] | None = strawberry_django.filter_field()
+    ownership_type: FilterLookup[str] | None = strawberry_django.filter_field()
     provider_segment_id: FilterLookup[str] | None = strawberry_django.filter_field()
     comments: FilterLookup[str] | None = strawberry_django.filter_field()
 

cesnet_service_path_plugin/graphql/types.py

@@ -81,6 +81,7 @@ class SegmentType(NetBoxObjectType):
     install_date: auto
     termination_date: auto
     status: auto
+    ownership_type: auto
 
     # Segment type fields
     segment_type: auto

cesnet_service_path_plugin/models/segment.py

@@ -178,6 +178,9 @@ def save(self, *args, **kwargs):
     def get_status_color(self):
         return StatusChoices.colors.get(self.status, "gray")
 
+    def get_ownership_type_color(self):
+        return OwnershipTypeChoices.colors.get(self.ownership_type, "gray")
+
     def get_segment_type_color(self):
         """Get color for segment type badge"""
         return SegmentTypeChoices.colors.get(self.segment_type, "gray")