perf: 修复应用账号选择部分导出问题

This commit is contained in:
ibuler
2021-08-17 14:16:41 +08:00
committed by Jiangjie.Bai
parent 5cd23b843a
commit 1605a57df6
2 changed files with 9 additions and 1 deletions

View File

@@ -112,7 +112,10 @@ class IDSpmFilter(filters.BaseFilterBackend):
resource_ids = cache.get(cache_key)
if resource_ids is None or not isinstance(resource_ids, list):
return queryset
queryset = queryset.filter(id__in=resource_ids)
if hasattr(view, 'filter_spm_queryset'):
queryset = view.filter_spm_queryset(resource_ids, queryset)
else:
queryset = queryset.filter(id__in=resource_ids)
return queryset