mirror of https://github.com/jumpserver/jumpserver
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
# -*- 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 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(ExtraFilterFieldsMixin):
|
|
pass
|