mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.6 KiB
58 lines
1.6 KiB
# -*- coding: utf-8 -*-
|
|
#
|
|
from django.http import JsonResponse
|
|
from rest_framework.settings import api_settings
|
|
|
|
from ..filters import IDSpmFilter, CustomFilter
|
|
|
|
__all__ = [
|
|
"JSONResponseMixin", "CommonApiMixin",
|
|
"IDSpmFilterMixin", "CommonApiMixin",
|
|
]
|
|
|
|
|
|
class JSONResponseMixin(object):
|
|
"""JSON mixin"""
|
|
@staticmethod
|
|
def render_json_response(context):
|
|
return JsonResponse(context)
|
|
|
|
|
|
class IDSpmFilterMixin:
|
|
def get_filter_backends(self):
|
|
backends = super().get_filter_backends()
|
|
backends.append(IDSpmFilter)
|
|
return backends
|
|
|
|
|
|
class SerializerMixin:
|
|
def get_serializer_class(self):
|
|
if self.request.method.lower() == 'get' and\
|
|
self.request.query_params.get('draw') \
|
|
and hasattr(self, 'serializer_display_class'):
|
|
return self.serializer_display_class
|
|
return super().get_serializer_class()
|
|
|
|
|
|
class ExtraFilterFieldsMixin:
|
|
default_added_filters = [CustomFilter, IDSpmFilter]
|
|
filter_backends = api_settings.DEFAULT_FILTER_BACKENDS
|
|
extra_filter_fields = []
|
|
extra_filter_backends = []
|
|
|
|
def get_filter_backends(self):
|
|
if self.filter_backends != self.__class__.filter_backends:
|
|
return self.filter_backends
|
|
return list(self.filter_backends) + \
|
|
self.default_added_filters + \
|
|
list(self.extra_filter_backends)
|
|
|
|
def filter_queryset(self, queryset):
|
|
for backend in self.get_filter_backends():
|
|
queryset = backend().filter_queryset(self.request, queryset, self)
|
|
return queryset
|
|
|
|
|
|
class CommonApiMixin(SerializerMixin, ExtraFilterFieldsMixin):
|
|
pass
|