mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-08-25 17:39:25 +00:00
d3be16ffe8
* perf(perms): 资产授权列表关联数据改为 `prefetch_related` * perf(perms): 优化一波 * dispatch_mapping_node_tasks.delay * perf: 在做一些优化 * perf: 再优化一波 * perf(perms): 授权更改节点慢的问题 * fix: 修改一处bug * perf(perms): ungrouped 资产数量计算方式 * fix: 修复dispatch data中的bug * fix(assets): add_nodes_assets_to_system_users celery task * fix: 修复ungrouped的bug * feat(nodes): 添加 favorite 节点 * feat(node): 添加 favorite api * fix: 修复clean keys的bug Co-authored-by: xinwen <coderWen@126.com> Co-authored-by: ibuler <ibuler@qq.com>
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from rest_framework.pagination import LimitOffsetPagination
|
|
from rest_framework.request import Request
|
|
|
|
from common.utils import get_logger
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class GrantedAssetLimitOffsetPagination(LimitOffsetPagination):
|
|
def get_count(self, queryset):
|
|
exclude_query_params = {
|
|
self.limit_query_param,
|
|
self.offset_query_param,
|
|
'key', 'all', 'show_current_asset',
|
|
'cache_policy', 'display', 'draw'
|
|
}
|
|
for k, v in self._request.query_params.items():
|
|
if k not in exclude_query_params and v is not None:
|
|
return super().get_count(queryset)
|
|
node = getattr(self._view, 'pagination_node', None)
|
|
if node:
|
|
logger.debug(f'{self._request.get_full_path()} hit node.assets_amount[{node.assets_amount}]')
|
|
return node.assets_amount
|
|
else:
|
|
return super().get_count(queryset)
|
|
|
|
def paginate_queryset(self, queryset, request: Request, view=None):
|
|
self._request = request
|
|
self._view = view
|
|
return super().paginate_queryset(queryset, request, view=None)
|