mirror of https://github.com/jumpserver/jumpserver
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from typing import Callable
|
|
|
|
from django.utils.translation import gettext as _
|
|
from rest_framework.decorators import action
|
|
from rest_framework.request import Request
|
|
from rest_framework.response import Response
|
|
|
|
from common.const.http import POST
|
|
|
|
__all__ = ['SuggestionMixin', 'RenderToJsonMixin']
|
|
|
|
|
|
class SuggestionMixin:
|
|
suggestion_limit = 10
|
|
|
|
filter_queryset: Callable
|
|
get_queryset: Callable
|
|
paginate_queryset: Callable
|
|
get_serializer: Callable
|
|
get_paginated_response: Callable
|
|
|
|
@action(methods=['get'], detail=False, url_path='suggestions')
|
|
def match(self, request, *args, **kwargs):
|
|
queryset = self.filter_queryset(self.get_queryset())
|
|
queryset = queryset[:self.suggestion_limit]
|
|
page = self.paginate_queryset(queryset)
|
|
|
|
if page is not None:
|
|
serializer = self.get_serializer(page, many=True)
|
|
return self.get_paginated_response(serializer.data)
|
|
|
|
serializer = self.get_serializer(queryset, many=True)
|
|
return Response(serializer.data)
|
|
|
|
|
|
class RenderToJsonMixin:
|
|
@action(methods=[POST], detail=False, url_path='render-to-json')
|
|
def render_to_json(self, request: Request, *args, **kwargs):
|
|
data = {
|
|
'title': (),
|
|
'data': request.data,
|
|
}
|
|
|
|
jms_context = getattr(request, 'jms_context', {})
|
|
column_title_field_pairs = jms_context.get('column_title_field_pairs', ())
|
|
data['title'] = column_title_field_pairs
|
|
|
|
if isinstance(request.data, (list, tuple)) and not any(request.data):
|
|
error = _("Request file format may be wrong")
|
|
return Response(data={"error": error}, status=400)
|
|
return Response(data=data)
|