Class TaskQuerySet

label_studio/data_manager/managers.py:501–544

QuerySet for Task model with Data Manager filters and ordering support.

class TaskQuerySet(FSMStateQuerySetMixin, models.QuerySet):
    """QuerySet for Task model with Data Manager filters and ordering support."""

    def prepared(self, prepare_params=None):
        """Apply filters, ordering and selected items to queryset

        :param prepare_params: prepare params with project, filters, orderings, etc
        :return: ordered and filtered queryset

        Note: For multi-project queries, filters and ordering will use the first project's
        configuration (label config, custom fields, etc.). This is backwards compatible
        with single-project queries.
        """
        from projects.models import Project

        queryset = self

        if prepare_params is None:
            return queryset

        # Get the project for filter/ordering configuration
        # For multi-project queries, use the first project's configuration
        if prepare_params.is_multi_project:
            project = Project.objects.get(pk=prepare_params.projects[0])
        else:
            # Backwards compatible: prepare_params.project is an int
            project = Project.objects.get(pk=prepare_params.project)

        request = prepare_params.request
        queryset = apply_filters(queryset, prepare_params.filters, project, request)
        queryset = apply_ordering(queryset, prepare_params.ordering, project, request, view_data=prepare_params.data)

        if not prepare_params.selectedItems:
            return queryset

        # included selected items
        if prepare_params.selectedItems.all is False and prepare_params.selectedItems.included:
            queryset = queryset.filter(id__in=prepare_params.selectedItems.included)

        # excluded selected items
        elif prepare_params.selectedItems.all is True and prepare_params.selectedItems.excluded:
            queryset = queryset.exclude(id__in=prepare_params.selectedItems.excluded)

        return queryset


class GroupConcat(Aggregate):