jumpserver/apps/assets/api/asset/host.py

16 lines
380 B
Python

from assets.models import Host
from assets.serializers import HostSerializer
from .common import AssetViewSet
__all__ = ['HostViewSet']
class HostViewSet(AssetViewSet):
model = Host
def get_serializer_classes(self):
serializer_classes = super().get_serializer_classes()
serializer_classes['default'] = HostSerializer
return serializer_classes