mirror of https://github.com/jumpserver/jumpserver
perf: 优化 一些继承
parent
492dd34cb8
commit
4818e759bc
|
@ -1,20 +1,20 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
#
|
#
|
||||||
|
|
||||||
from django.utils.translation import ugettext as _
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from rest_framework_bulk import BulkModelViewSet
|
from django.utils.translation import ugettext as _
|
||||||
from rest_framework.generics import RetrieveAPIView
|
|
||||||
from rest_framework.exceptions import PermissionDenied
|
from rest_framework.exceptions import PermissionDenied
|
||||||
|
from rest_framework.generics import RetrieveAPIView
|
||||||
|
|
||||||
from common.utils import get_logger
|
|
||||||
from common.permissions import IsValidUser
|
|
||||||
from users.models import User, UserGroup
|
|
||||||
from assets.models import (
|
from assets.models import (
|
||||||
Asset, Domain, Label, Node,
|
Asset, Domain, Label, Node,
|
||||||
)
|
)
|
||||||
from perms.models import AssetPermission
|
from common.api import JMSBulkModelViewSet
|
||||||
|
from common.permissions import IsValidUser
|
||||||
|
from common.utils import get_logger
|
||||||
from orgs.utils import current_org, tmp_to_root_org
|
from orgs.utils import current_org, tmp_to_root_org
|
||||||
|
from perms.models import AssetPermission
|
||||||
|
from users.models import User, UserGroup
|
||||||
from .models import Organization
|
from .models import Organization
|
||||||
from .serializers import (
|
from .serializers import (
|
||||||
OrgSerializer, CurrentOrgSerializer
|
OrgSerializer, CurrentOrgSerializer
|
||||||
|
@ -29,7 +29,7 @@ org_related_models = [
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
class OrgViewSet(BulkModelViewSet):
|
class OrgViewSet(JMSBulkModelViewSet):
|
||||||
filterset_fields = ('name',)
|
filterset_fields = ('name',)
|
||||||
search_fields = ('name', 'comment')
|
search_fields = ('name', 'comment')
|
||||||
queryset = Organization.objects.all()
|
queryset = Organization.objects.all()
|
||||||
|
|
|
@ -26,6 +26,7 @@ logger = get_logger(__name__)
|
||||||
class BaseUserPermedAssetsApi(SelfOrPKUserMixin, ListAPIView):
|
class BaseUserPermedAssetsApi(SelfOrPKUserMixin, ListAPIView):
|
||||||
ordering = ('name',)
|
ordering = ('name',)
|
||||||
search_fields = ('name', 'address', 'comment')
|
search_fields = ('name', 'address', 'comment')
|
||||||
|
ordering_fields = ("name", "address")
|
||||||
filterset_class = AssetFilterSet
|
filterset_class = AssetFilterSet
|
||||||
serializer_class = serializers.AssetPermedSerializer
|
serializer_class = serializers.AssetPermedSerializer
|
||||||
only_fields = serializers.AssetPermedSerializer.Meta.only_fields
|
only_fields = serializers.AssetPermedSerializer.Meta.only_fields
|
||||||
|
|
|
@ -1,22 +1,23 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
#
|
#
|
||||||
import logging
|
import logging
|
||||||
from rest_framework.views import APIView, Response
|
|
||||||
from rest_framework_bulk import BulkModelViewSet
|
|
||||||
from rest_framework import status
|
from rest_framework import status
|
||||||
from rest_framework.permissions import IsAuthenticated
|
from rest_framework.permissions import IsAuthenticated
|
||||||
|
from rest_framework.views import APIView, Response
|
||||||
|
|
||||||
|
from common.api import JMSBulkModelViewSet
|
||||||
from common.utils import get_object_or_none
|
from common.utils import get_object_or_none
|
||||||
from orgs.utils import tmp_to_root_org
|
from orgs.utils import tmp_to_root_org
|
||||||
from terminal.models import Session, Task
|
|
||||||
from terminal import serializers
|
from terminal import serializers
|
||||||
|
from terminal.models import Session, Task
|
||||||
from terminal.utils import is_session_approver
|
from terminal.utils import is_session_approver
|
||||||
|
|
||||||
__all__ = ['TaskViewSet', 'KillSessionAPI', 'KillSessionForTicketAPI']
|
__all__ = ['TaskViewSet', 'KillSessionAPI', 'KillSessionForTicketAPI']
|
||||||
logger = logging.getLogger(__file__)
|
logger = logging.getLogger(__file__)
|
||||||
|
|
||||||
|
|
||||||
class TaskViewSet(BulkModelViewSet):
|
class TaskViewSet(JMSBulkModelViewSet):
|
||||||
queryset = Task.objects.all()
|
queryset = Task.objects.all()
|
||||||
serializer_class = serializers.TaskSerializer
|
serializer_class = serializers.TaskSerializer
|
||||||
filterset_fields = ('is_finished',)
|
filterset_fields = ('is_finished',)
|
||||||
|
@ -51,7 +52,7 @@ class KillSessionAPI(APIView):
|
||||||
|
|
||||||
|
|
||||||
class KillSessionForTicketAPI(APIView):
|
class KillSessionForTicketAPI(APIView):
|
||||||
permission_classes = (IsAuthenticated, )
|
permission_classes = (IsAuthenticated,)
|
||||||
|
|
||||||
def post(self, request, *args, **kwargs):
|
def post(self, request, *args, **kwargs):
|
||||||
session_ids = request.data
|
session_ids = request.data
|
||||||
|
|
Loading…
Reference in New Issue