mirror of
https://github.com/lucaspalomodevelop/netbox-acls.git
synced 2026-03-12 23:27:23 +00:00
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
"""
|
|
Create views to handle the API logic.
|
|
A view set is a single class that can handle the view, add, change,
|
|
and delete operations which each require dedicated views under the UI.
|
|
"""
|
|
|
|
from django.db.models import Count
|
|
from netbox.api.viewsets import NetBoxModelViewSet
|
|
|
|
from .. import filtersets, models
|
|
from .serializers import (
|
|
AccessListSerializer,
|
|
ACLExtendedRuleSerializer,
|
|
ACLInterfaceAssignmentSerializer,
|
|
ACLStandardRuleSerializer,
|
|
)
|
|
|
|
__all__ = [
|
|
"AccessListViewSet",
|
|
"ACLStandardRuleViewSet",
|
|
"ACLInterfaceAssignmentViewSet" "ACLExtendedRuleViewSet",
|
|
]
|
|
|
|
|
|
class AccessListViewSet(NetBoxModelViewSet):
|
|
"""
|
|
Defines the view set for the django AccessList model & associates it to a view.
|
|
"""
|
|
|
|
queryset = models.AccessList.objects.prefetch_related("tags").annotate(
|
|
rule_count=Count("aclextendedrules") + Count("aclstandardrules"),
|
|
)
|
|
serializer_class = AccessListSerializer
|
|
filterset_class = filtersets.AccessListFilterSet
|
|
|
|
|
|
class ACLInterfaceAssignmentViewSet(NetBoxModelViewSet):
|
|
"""
|
|
Defines the view set for the django ACLInterfaceAssignment model & associates it to a view.
|
|
"""
|
|
|
|
queryset = models.ACLInterfaceAssignment.objects.prefetch_related(
|
|
"access_list", "tags"
|
|
)
|
|
serializer_class = ACLInterfaceAssignmentSerializer
|
|
filterset_class = filtersets.ACLInterfaceAssignmentFilterSet
|
|
|
|
|
|
class ACLStandardRuleViewSet(NetBoxModelViewSet):
|
|
"""
|
|
Defines the view set for the django ACLStandardRule model & associates it to a view.
|
|
"""
|
|
|
|
queryset = models.ACLStandardRule.objects.prefetch_related(
|
|
"access_list",
|
|
"tags",
|
|
"source_prefix",
|
|
)
|
|
serializer_class = ACLStandardRuleSerializer
|
|
filterset_class = filtersets.ACLStandardRuleFilterSet
|
|
|
|
|
|
class ACLExtendedRuleViewSet(NetBoxModelViewSet):
|
|
"""
|
|
Defines the view set for the django ACLExtendedRule model & associates it to a view.
|
|
"""
|
|
|
|
queryset = models.ACLExtendedRule.objects.prefetch_related(
|
|
"access_list",
|
|
"tags",
|
|
"source_prefix",
|
|
"destination_prefix",
|
|
)
|
|
serializer_class = ACLExtendedRuleSerializer
|
|
filterset_class = filtersets.ACLExtendedRuleFilterSet
|