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

16 lines
397 B
Python

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