Maybe a little cleaner

This commit is contained in:
shamoon 2025-03-07 11:06:54 -08:00
parent 75caba8edd
commit fe774f9901

View File

@ -32,6 +32,7 @@ from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework.viewsets import ModelViewSet from rest_framework.viewsets import ModelViewSet
from documents.index import DelayedQuery
from documents.permissions import PaperlessObjectPermissions from documents.permissions import PaperlessObjectPermissions
from paperless.filters import GroupFilterSet from paperless.filters import GroupFilterSet
from paperless.filters import UserFilterSet from paperless.filters import UserFilterSet
@ -66,19 +67,17 @@ class StandardPagination(PageNumberPagination):
) )
def get_all_result_ids(self): def get_all_result_ids(self):
ids = [] query = self.page.paginator.object_list
if hasattr(self.page.paginator.object_list, "saved_results"): if isinstance(query, DelayedQuery):
results_page = self.page.paginator.object_list.saved_results[0] try:
if results_page is not None: ids = [
for doc_num in results_page.results.docs(): query.searcher.ixreader.stored_fields(
try: doc_num,
fields = results_page.results.searcher.ixreader.stored_fields( )["id"]
doc_num, for doc_num in query.saved_results.get(0).results.docs()
) ]
if "id" in fields: except Exception:
ids.append(fields["id"]) pass
except Exception:
pass
else: else:
ids = self.page.paginator.object_list.values_list("pk", flat=True) ids = self.page.paginator.object_list.values_list("pk", flat=True)
return ids return ids