mirror of
https://github.com/lucaspalomodevelop/netbox-acls.git
synced 2026-03-16 17:04:36 +00:00
22 lines
695 B
Python
22 lines
695 B
Python
from django.db.models import Count
|
|
|
|
from netbox.api.viewsets import NetBoxModelViewSet
|
|
|
|
from .. import filtersets, models
|
|
from .serializers import AccessListSerializer, AccessListRuleSerializer
|
|
|
|
|
|
class AccessListViewSet(NetBoxModelViewSet):
|
|
queryset = models.AccessList.objects.prefetch_related('tags').annotate(
|
|
rule_count=Count('rules')
|
|
)
|
|
serializer_class = AccessListSerializer
|
|
|
|
|
|
class AccessListRuleViewSet(NetBoxModelViewSet):
|
|
queryset = models.AccessListRule.objects.prefetch_related(
|
|
'access_list', 'source_prefix', 'destination_prefix', 'tags'
|
|
)
|
|
serializer_class = AccessListRuleSerializer
|
|
filterset_class = filtersets.AccessListRuleFilterSet
|