github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-25 23:00:57 +00:00
Code Issues Projects Releases Wiki Activity

[Update] 添加ids filter

Browse Source
This commit is contained in:
ibuler
2020-04-13 10:40:13 +08:00
parent 42c5c02709
commit 0d2b4d7ca3
4 changed files with 25 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
apps/common/drf/filters.py
View File

@@ -9,7 +9,7 @@ import logging
from common import const
__all__ = ["DatetimeRangeFilter", "IDSpmFilter", "CustomFilter"]
__all__ = ["DatetimeRangeFilter", "IDSpmFilter", 'IDInFilter', "CustomFilter"]
class DatetimeRangeFilter(filters.BaseFilterBackend):
@@ -68,6 +68,25 @@ class IDSpmFilter(filters.BaseFilterBackend):
return queryset
class IDInFilter(filters.BaseFilterBackend):
def get_schema_fields(self, view):
return [
coreapi.Field(
name='ids', location='query', required=False,
type='string', example='/api/v1/users/users?ids=1,2,3',
description='Filter by id set'
)
]
def filter_queryset(self, request, queryset, view):
ids = request.query_params.get('ids')
if not ids:
return queryset
id_list = [i.strip() for i in ids.split(',')]
queryset = queryset.filter(id__in=id_list)
return queryset
class CustomFilter(filters.BaseFilterBackend):
def get_schema_fields(self, view):