mirror of https://github.com/jumpserver/jumpserver
16 lines
391 B
Python
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
|