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)
