mirror of https://github.com/jumpserver/jumpserver
21 lines
539 B
Python
21 lines
539 B
Python
|
|
||
|
|
from assets.models import Database
|
||
|
|
from assets.serializers import DatabaseSerializer
|
||
|
|
from .common import AssetViewSet
|
||
|
|
|
||
|
|
__all__ = ['DatabaseViewSet']
|
||
|
|
|
||
|
|
|
||
|
|
class DatabaseViewSet(AssetViewSet):
|
||
|
|
model = Database
|
||
|
|
|
||
|
|
def get_queryset(self):
|
||
|
|
queryset = super().get_queryset()
|
||
|
|
print("Datbase is: ", queryset)
|
||
|
|
return queryset
|
||
|
|
|
||
|
|
def get_serializer_classes(self):
|
||
|
|
serializer_classes = super().get_serializer_classes()
|
||
|
|
serializer_classes['default'] = DatabaseSerializer
|
||
|
|
return serializer_classes
|