def filter_queryset(self, queryset) -> Any: queryset = queryset.filter(document__processed=True, document__delete_pending=False) # perm check - use only allowed docs user_id = self.extra_kwargs.get('user_id') if user_id: allowed_document_ids = Document.get_allowed_document_ids(user_id) queryset = queryset.filter(document_id__in=allowed_document_ids) if not self.project_id: return queryset project_ids = [self.project_id] if isinstance(self.project_id, (int, str)) else self.project_id or [] return queryset.filter(document__project_id__in=project_ids)
def filter_queryset(self, queryset) -> Any: # TODO: this is copied from TextUnitFeatures, consider to inherit from 2 parents? queryset = queryset.filter(document__processed=True, document__delete_pending=False) # perm check - use only allowed docs user_id = self.extra_kwargs.get('user_id') if user_id: allowed_document_ids = Document.get_allowed_document_ids(user_id) queryset = queryset.filter(document_id__in=allowed_document_ids) if not self.project_id: return queryset project_ids = [self.project_id] if isinstance(self.project_id, (int, str)) else self.project_id or [] return queryset.filter(document__project_id__in=project_ids)