mirror of https://github.com/jumpserver/jumpserver
26 lines
724 B
Python
26 lines
724 B
Python
from rest_framework.generics import ListAPIView
|
|
|
|
from assets import serializers
|
|
from assets.const import Protocol
|
|
from common.permissions import IsValidUser
|
|
|
|
__all__ = ['ProtocolListApi']
|
|
|
|
|
|
class ProtocolListApi(ListAPIView):
|
|
serializer_class = serializers.ProtocolSerializer
|
|
permission_classes = (IsValidUser,)
|
|
|
|
def get_queryset(self):
|
|
return list(Protocol.protocols())
|
|
|
|
def filter_queryset(self, queryset):
|
|
search = self.request.query_params.get("search", "").lower().strip()
|
|
if not search:
|
|
return queryset
|
|
queryset = [
|
|
p for p in queryset
|
|
if search in p['label'].lower() or search in p['value'].lower()
|
|
]
|
|
return queryset
|