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.
91 lines
2.8 KiB
91 lines
2.8 KiB
# -*- coding: utf-8 -*- |
|
# |
|
from django.db.models import Q |
|
from rest_framework import filters |
|
from rest_framework.compat import coreapi, coreschema |
|
|
|
from assets.utils import get_node_from_request, is_query_node_all_assets |
|
|
|
|
|
class AssetByNodeFilterBackend(filters.BaseFilterBackend): |
|
fields = ['node', 'all'] |
|
|
|
def get_schema_fields(self, view): |
|
return [ |
|
coreapi.Field( |
|
name=field, location='query', required=False, |
|
type='string', example='', description='', schema=None, |
|
) |
|
for field in self.fields |
|
] |
|
|
|
def filter_node_related_all(self, queryset, node): |
|
return queryset.filter( |
|
Q(nodes__key__istartswith=f'{node.key}:') | |
|
Q(nodes__key=node.key) |
|
).distinct() |
|
|
|
def filter_node_related_direct(self, queryset, node): |
|
return queryset.filter(nodes__key=node.key).distinct() |
|
|
|
def filter_queryset(self, request, queryset, view): |
|
node = get_node_from_request(request) |
|
if node is None: |
|
return queryset |
|
|
|
query_all = is_query_node_all_assets(request) |
|
if query_all: |
|
return self.filter_node_related_all(queryset, node) |
|
else: |
|
return self.filter_node_related_direct(queryset, node) |
|
|
|
|
|
class NodeFilterBackend(filters.BaseFilterBackend): |
|
""" |
|
需要与 `assets.api.mixin.NodeFilterMixin` 配合使用 |
|
""" |
|
fields = ['node', 'all'] |
|
|
|
def get_schema_fields(self, view): |
|
return [ |
|
coreapi.Field( |
|
name=field, location='query', required=False, |
|
type='string', example='', description='', schema=None, |
|
) |
|
for field in self.fields |
|
] |
|
|
|
def filter_queryset(self, request, queryset, view): |
|
node = get_node_from_request(request) |
|
if node is None: |
|
return queryset |
|
|
|
query_all = is_query_node_all_assets(request) |
|
if query_all: |
|
return queryset.filter( |
|
Q(nodes__key__startswith=f'{node.key}:') | |
|
Q(nodes__key=node.key) |
|
).distinct() |
|
else: |
|
return queryset.filter(nodes__key=node.key).distinct() |
|
|
|
|
|
class IpInFilterBackend(filters.BaseFilterBackend): |
|
def filter_queryset(self, request, queryset, view): |
|
ips = request.query_params.get('ips') |
|
if not ips: |
|
return queryset |
|
ip_list = [i.strip() for i in ips.split(',')] |
|
queryset = queryset.filter(address__in=ip_list) |
|
return queryset |
|
|
|
def get_schema_fields(self, view): |
|
return [ |
|
coreapi.Field( |
|
name='ips', location='query', required=False, type='string', |
|
schema=coreschema.String( |
|
title='ips', |
|
description='address in filter' |
|
) |
|
) |
|
]
|
|
|