Files
jumpserver/apps/assets/api/asset/network.py
2022-08-23 10:23:48 +08:00

16 lines
391 B
Python

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