# -*- 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