mirror of https://github.com/jumpserver/jumpserver
17 lines
421 B
Python
17 lines
421 B
Python
from assets.models import Custom, Asset
|
|
from assets.serializers import CustomSerializer
|
|
|
|
from .asset import AssetViewSet
|
|
|
|
__all__ = ['CustomViewSet']
|
|
|
|
|
|
class CustomViewSet(AssetViewSet):
|
|
model = Custom
|
|
perm_model = Asset
|
|
|
|
def get_serializer_classes(self):
|
|
serializer_classes = super().get_serializer_classes()
|
|
serializer_classes['default'] = CustomSerializer
|
|
return serializer_classes
|