mirror of https://github.com/jumpserver/jumpserver
[future] 使用mixin去掉重复多余代码
Administrator
parent
1e835d2fa9
commit
79971d677d
|
|
@ -1,249 +1,86 @@
|
||||||
# ~*~ coding: utf-8 ~*~
|
# ~*~ coding: utf-8 ~*~
|
||||||
from __future__ import unicode_literals
|
from __future__ import unicode_literals
|
||||||
from rest_framework import viewsets
|
from rest_framework import viewsets, mixins
|
||||||
from rest_framework import status
|
|
||||||
from rest_framework.response import Response
|
|
||||||
|
|
||||||
from serializers import *
|
from serializers import *
|
||||||
from permissions import *
|
from permissions import *
|
||||||
|
|
||||||
import exc
|
|
||||||
|
|
||||||
|
class HostAliaViewSet(mixins.CreateModelMixin,
|
||||||
class HostAliaViewSet(viewsets.GenericViewSet):
|
mixins.ListModelMixin,
|
||||||
|
mixins.RetrieveModelMixin,
|
||||||
|
mixins.UpdateModelMixin,
|
||||||
|
mixins.DestroyModelMixin,
|
||||||
|
viewsets.GenericViewSet):
|
||||||
queryset = HostAlia.objects.all()
|
queryset = HostAlia.objects.all()
|
||||||
serializer_class = HostAliaSerializer
|
serializer_class = HostAliaSerializer
|
||||||
permission_classes = (AdminUserRequired,)
|
permission_classes = (AdminUserRequired,)
|
||||||
|
|
||||||
def list(self, *args, **kwargs):
|
|
||||||
h_alias = self.get_queryset()
|
|
||||||
h_serializer = self.get_serializer(h_alias, many=True)
|
|
||||||
return Response(h_serializer.data)
|
|
||||||
|
|
||||||
def create(self, *args, **kwargs):
|
class CmdAliaViewSet(mixins.CreateModelMixin,
|
||||||
serializer = self.get_serializer(data=self.request.data)
|
mixins.ListModelMixin,
|
||||||
serializer.is_valid(raise_exception=True)
|
mixins.RetrieveModelMixin,
|
||||||
self.perform_create()
|
mixins.UpdateModelMixin,
|
||||||
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
mixins.DestroyModelMixin,
|
||||||
|
viewsets.GenericViewSet):
|
||||||
def retrieve(self, *args, **kwargs):
|
|
||||||
h_alias = self.get_object()
|
|
||||||
serializer = self.get_serializer(h_alias)
|
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
||||||
def update(self, *args, **kwargs):
|
|
||||||
h_alias = self.get_object()
|
|
||||||
serializer = self.get_serializer(h_alias, data=self.request.data)
|
|
||||||
serializer.is_valid(raise_exception=False)
|
|
||||||
self.perform_create()
|
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
||||||
def destroy(self, *args, **kwargs):
|
|
||||||
h_alias = self.get_object()
|
|
||||||
h_alias.delete()
|
|
||||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
||||||
|
|
||||||
def perform_create(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class CmdAliaViewSet(viewsets.GenericViewSet):
|
|
||||||
queryset = CmdAlia.objects.all()
|
queryset = CmdAlia.objects.all()
|
||||||
serializer_class = CmdAliaSerializer
|
serializer_class = CmdAliaSerializer
|
||||||
permission_classes = (AdminUserRequired,)
|
permission_classes = (AdminUserRequired,)
|
||||||
|
|
||||||
def list(self, *args, **kwargs):
|
|
||||||
c_alias = self.get_queryset()
|
|
||||||
c_serializer = self.get_serializer(c_alias, many=True)
|
|
||||||
return Response(c_serializer.data)
|
|
||||||
|
|
||||||
def create(self, *args, **kwargs):
|
class UserAliaViewSet(mixins.CreateModelMixin,
|
||||||
serializer = self.get_serializer(data=self.request.data)
|
mixins.ListModelMixin,
|
||||||
serializer.is_valid(raise_exception=True)
|
mixins.RetrieveModelMixin,
|
||||||
self.perform_create()
|
mixins.UpdateModelMixin,
|
||||||
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
mixins.DestroyModelMixin,
|
||||||
|
viewsets.GenericViewSet):
|
||||||
def retrieve(self, *args, **kwargs):
|
|
||||||
c_alias = self.get_object()
|
|
||||||
serializer = self.get_serializer(c_alias)
|
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
||||||
def update(self, *args, **kwargs):
|
|
||||||
c_alias = self.get_object()
|
|
||||||
serializer = self.get_serializer(c_alias, data=self.request.data)
|
|
||||||
serializer.is_valid(raise_exception=False)
|
|
||||||
self.perform_create()
|
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
||||||
def destroy(self, *args, **kwargs):
|
|
||||||
c_alias = self.get_object()
|
|
||||||
c_alias.delete()
|
|
||||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
||||||
|
|
||||||
def perform_create(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class UserAliaViewSet(viewsets.GenericViewSet):
|
|
||||||
queryset = UserAlia.objects.all()
|
queryset = UserAlia.objects.all()
|
||||||
serializer_class = UserAliaSerializer
|
serializer_class = UserAliaSerializer
|
||||||
permission_classes = (AdminUserRequired,)
|
permission_classes = (AdminUserRequired,)
|
||||||
|
|
||||||
def list(self, *args, **kwargs):
|
|
||||||
u_alias = self.get_queryset()
|
|
||||||
u_serializer = self.get_serializer(u_alias, many=True)
|
|
||||||
return Response(u_serializer.data)
|
|
||||||
|
|
||||||
def create(self, *args, **kwargs):
|
class RunasAliaViewSet(mixins.CreateModelMixin,
|
||||||
serializer = self.get_serializer(data=self.request.data)
|
mixins.ListModelMixin,
|
||||||
serializer.is_valid(raise_exception=True)
|
mixins.RetrieveModelMixin,
|
||||||
self.perform_create()
|
mixins.UpdateModelMixin,
|
||||||
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
mixins.DestroyModelMixin,
|
||||||
|
viewsets.GenericViewSet):
|
||||||
def retrieve(self, *args, **kwargs):
|
|
||||||
u_alias = self.get_object()
|
|
||||||
serializer = self.get_serializer(u_alias)
|
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
||||||
def update(self, *args, **kwargs):
|
|
||||||
u_alias = self.get_object()
|
|
||||||
serializer = self.get_serializer(u_alias, data=self.request.data)
|
|
||||||
serializer.is_valid(raise_exception=False)
|
|
||||||
self.perform_create()
|
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
||||||
def destroy(self, *args, **kwargs):
|
|
||||||
u_alias = self.get_object()
|
|
||||||
u_alias.delete()
|
|
||||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
||||||
|
|
||||||
def perform_create(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class RunasAliaViewSet(viewsets.GenericViewSet):
|
|
||||||
queryset = RunasAlia.objects.all()
|
queryset = RunasAlia.objects.all()
|
||||||
serializer_class = RunasAliaSerializer
|
serializer_class = RunasAliaSerializer
|
||||||
permission_classes = (AdminUserRequired,)
|
permission_classes = (AdminUserRequired,)
|
||||||
|
|
||||||
def list(self, *args, **kwargs):
|
|
||||||
r_alias = self.get_queryset()
|
|
||||||
r_serializer = self.get_serializer(r_alias, many=True)
|
|
||||||
return Response(r_serializer.data)
|
|
||||||
|
|
||||||
def create(self, *args, **kwargs):
|
class ExtraconfViewSet(mixins.CreateModelMixin,
|
||||||
serializer = self.get_serializer(data=self.request.data)
|
mixins.ListModelMixin,
|
||||||
serializer.is_valid(raise_exception=True)
|
mixins.RetrieveModelMixin,
|
||||||
self.perform_create()
|
mixins.UpdateModelMixin,
|
||||||
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
mixins.DestroyModelMixin,
|
||||||
|
viewsets.GenericViewSet):
|
||||||
def retrieve(self, *args, **kwargs):
|
|
||||||
r_alias = self.get_object()
|
|
||||||
serializer = self.get_serializer(r_alias)
|
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
||||||
def update(self, *args, **kwargs):
|
|
||||||
r_alias = self.get_object()
|
|
||||||
serializer = self.get_serializer(r_alias, data=self.request.data)
|
|
||||||
serializer.is_valid(raise_exception=False)
|
|
||||||
self.perform_create()
|
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
||||||
def destroy(self, *args, **kwargs):
|
|
||||||
r_alias = self.get_object()
|
|
||||||
r_alias.delete()
|
|
||||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
||||||
|
|
||||||
def perform_create(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class ExtraconfViewSet(viewsets.GenericViewSet):
|
|
||||||
queryset = Extra_conf.objects.all()
|
queryset = Extra_conf.objects.all()
|
||||||
serializer_class = ExtraconfSerializer
|
serializer_class = ExtraconfSerializer
|
||||||
permission_classes = (AdminUserRequired,)
|
permission_classes = (AdminUserRequired,)
|
||||||
|
|
||||||
def list(self, *args, **kwargs):
|
|
||||||
e_alias = self.get_queryset()
|
|
||||||
e_serializer = self.get_serializer(e_alias, many=True)
|
|
||||||
return Response(e_serializer.data)
|
|
||||||
|
|
||||||
def create(self, *args, **kwargs):
|
class PrivilegeViewSet(mixins.CreateModelMixin,
|
||||||
serializer = self.get_serializer(data=self.request.data)
|
mixins.ListModelMixin,
|
||||||
serializer.is_valid(raise_exception=True)
|
mixins.RetrieveModelMixin,
|
||||||
self.perform_create()
|
mixins.UpdateModelMixin,
|
||||||
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
mixins.DestroyModelMixin,
|
||||||
|
viewsets.GenericViewSet):
|
||||||
def retrieve(self, *args, **kwargs):
|
|
||||||
e_alias = self.get_object()
|
|
||||||
serializer = self.get_serializer(e_alias)
|
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
||||||
def update(self, *args, **kwargs):
|
|
||||||
e_alias = self.get_object()
|
|
||||||
serializer = self.get_serializer(e_alias, data=self.request.data)
|
|
||||||
serializer.is_valid(raise_exception=False)
|
|
||||||
self.perform_create()
|
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
||||||
def destroy(self, *args, **kwargs):
|
|
||||||
e_alias = self.get_object()
|
|
||||||
e_alias.delete()
|
|
||||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
||||||
|
|
||||||
def perform_create(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class PrivilegeViewSet(viewsets.GenericViewSet):
|
|
||||||
queryset = Privilege.objects.all()
|
queryset = Privilege.objects.all()
|
||||||
serializer_class = PrivilegeSerializer
|
serializer_class = PrivilegeSerializer
|
||||||
permission_classes = (AdminUserRequired,)
|
permission_classes = (AdminUserRequired,)
|
||||||
|
|
||||||
def list(self, *args, **kwargs):
|
|
||||||
raise exc.ServiceNotImplemented
|
|
||||||
|
|
||||||
def create(self, *args, **kwargs):
|
class SudoViewSet(mixins.CreateModelMixin,
|
||||||
raise exc.ServiceNotImplemented
|
mixins.ListModelMixin,
|
||||||
|
mixins.RetrieveModelMixin,
|
||||||
def retrieve(self, *args, **kwargs):
|
mixins.UpdateModelMixin,
|
||||||
raise exc.ServiceNotImplemented
|
mixins.DestroyModelMixin,
|
||||||
|
viewsets.GenericViewSet):
|
||||||
def update(self, *args, **kwargs):
|
|
||||||
raise exc.ServiceNotImplemented
|
|
||||||
|
|
||||||
def destroy(self, *args, **kwargs):
|
|
||||||
privilege = self.get_object()
|
|
||||||
privilege.delete()
|
|
||||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
||||||
|
|
||||||
def perform_create(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class SudoViewSet(viewsets.GenericViewSet):
|
|
||||||
queryset = Sudo.objects.all()
|
queryset = Sudo.objects.all()
|
||||||
serializer_class = SudoSerializer
|
serializer_class = SudoSerializer
|
||||||
permission_classes = (AdminUserRequired,)
|
permission_classes = (AdminUserRequired,)
|
||||||
|
|
||||||
def list(self, *args, **kwargs):
|
|
||||||
raise exc.ServiceNotImplemented
|
|
||||||
|
|
||||||
def create(self, *args, **kwargs):
|
|
||||||
raise exc.ServiceNotImplemented
|
|
||||||
|
|
||||||
def retrieve(self, *args, **kwargs):
|
|
||||||
raise exc.ServiceNotImplemented
|
|
||||||
|
|
||||||
def update(self, *args, **kwargs):
|
|
||||||
raise exc.ServiceNotImplemented
|
|
||||||
|
|
||||||
def destroy(self, *args, **kwargs):
|
|
||||||
sudo = self.get_object()
|
|
||||||
sudo.delete()
|
|
||||||
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
||||||
|
|
||||||
def perform_create(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue