bramw_baserow/backend/src/baserow/contrib/database/views/object_scopes.py

from django.db.models import Q, QuerySet

from baserow.contrib.database.object_scopes import DatabaseObjectScopeType
from baserow.contrib.database.table.object_scopes import DatabaseTableObjectScopeType
from baserow.contrib.database.views.models import (
    View,
    ViewDecoration,
    ViewFilter,
    ViewFilterGroup,
    ViewGroupBy,
    ViewSort,
)
from baserow.core.object_scopes import (
    ApplicationObjectScopeType,
    WorkspaceObjectScopeType,
)
from baserow.core.registries import ObjectScopeType, object_scope_type_registry


class DatabaseViewObjectScopeType(ObjectScopeType):
    type = "database_view"
    model_class = View

    def get_parent_scope(self):
        return object_scope_type_registry.get("database_table")

    def get_base_queryset(self, include_trash: bool = False) -> QuerySet:
        return (
            super()
            .get_base_queryset(include_trash)
            .filter(table__database__workspace__isnull=False)
        )

    def get_enhanced_queryset(self, include_trash: bool = False) -> QuerySet:
        return self.get_base_queryset(include_trash).select_related(
            "table__database__workspace"
        )

    def get_filter_for_scope_type(self, scope_type, scopes):
        if scope_type.type == WorkspaceObjectScopeType.type:
            return Q(table__database__workspace__in=[s.id for s in scopes])

        if (
            scope_type.type == DatabaseObjectScopeType.type
            or scope_type.type == ApplicationObjectScopeType.type
        ):
            return Q(table__database__in=[s.id for s in scopes])

        if scope_type.type == DatabaseTableObjectScopeType.type:
            return Q(table__in=[s.id for s in scopes])

        raise TypeError("The given type is not handled.")


class DatabaseViewDecorationObjectScopeType(ObjectScopeType):
    type = "database_view_decoration"
    model_class = ViewDecoration

    def get_parent_scope(self):
        return object_scope_type_registry.get("database_view")

    def get_base_queryset(self, include_trash: bool = False) -> QuerySet:
        return (
            super()
            .get_base_queryset(include_trash)
            .filter(view__table__database__workspace__isnull=False)
        )

    def get_enhanced_queryset(self, include_trash: bool = False) -> QuerySet:
        return self.get_base_queryset(include_trash).select_related(
            "view__table__database__workspace"
        )

    def get_filter_for_scope_type(self, scope_type, scopes):
        if scope_type.type == WorkspaceObjectScopeType.type:
            return Q(view_table__database__workspace__in=[s.id for s in scopes])

        if (
            scope_type.type == DatabaseObjectScopeType.type
            or scope_type.type == ApplicationObjectScopeType.type
        ):
            return Q(view__table__database__in=[s.id for s in scopes])

        if scope_type.type == DatabaseTableObjectScopeType.type:
            return Q(view__table__in=[s.id for s in scopes])

        if scope_type.type == DatabaseViewObjectScopeType.type:
            return Q(view__in=[s.id for s in scopes])

        raise TypeError("The given type is not handled.")


class DatabaseViewSortObjectScopeType(ObjectScopeType):
    type = "database_view_sort"
    model_class = ViewSort

    def get_parent_scope(self):
        return object_scope_type_registry.get("database_view")

    def get_base_queryset(self, include_trash: bool = False) -> QuerySet:
        return (
            super()
            .get_base_queryset(include_trash)
            .filter(view__table__database__workspace__isnull=False)
        )

    def get_enhanced_queryset(self, include_trash: bool = False) -> QuerySet:
        return self.get_base_queryset(include_trash).select_related(
            "view__table__database__workspace"
        )

    def get_filter_for_scope_type(self, scope_type, scopes):
        if scope_type.type == WorkspaceObjectScopeType.type:
            return Q(view_table__database__workspace__in=[s.id for s in scopes])

        if (
            scope_type.type == DatabaseObjectScopeType.type
            or scope_type.type == ApplicationObjectScopeType.type
        ):
            return Q(view__table__database__in=[s.id for s in scopes])

        if scope_type.type == DatabaseTableObjectScopeType.type:
            return Q(view__table__in=[s.id for s in scopes])

        if scope_type.type == DatabaseViewObjectScopeType.type:
            return Q(view__in=[s.id for s in scopes])

        raise TypeError("The given type is not handled.")


class DatabaseViewFilterObjectScopeType(ObjectScopeType):
    type = "database_view_filter"
    model_class = ViewFilter

    def get_parent_scope(self):
        return object_scope_type_registry.get("database_view")

    def get_base_queryset(self, include_trash: bool = False) -> QuerySet:
        return (
            super()
            .get_base_queryset(include_trash)
            .filter(view__table__database__workspace__isnull=False)
        )

    def get_enhanced_queryset(self, include_trash: bool = False) -> QuerySet:
        return self.get_base_queryset(include_trash).select_related(
            "view__table__database__workspace"
        )

    def get_filter_for_scope_type(self, scope_type, scopes):
        if scope_type.type == WorkspaceObjectScopeType.type:
            return Q(view_table__database__workspace__in=[s.id for s in scopes])

        if (
            scope_type.type == DatabaseObjectScopeType.type
            or scope_type.type == ApplicationObjectScopeType.type
        ):
            return Q(view__table__database__in=[s.id for s in scopes])

        if scope_type.type == DatabaseTableObjectScopeType.type:
            return Q(view__table__in=[s.id for s in scopes])

        if scope_type.type == DatabaseViewObjectScopeType.type:
            return Q(view__in=[s.id for s in scopes])

        raise TypeError("The given type is not handled.")


class DatabaseViewFilterGroupObjectScopeType(ObjectScopeType):
    type = "database_view_filter_group"
    model_class = ViewFilterGroup

    def get_parent_scope(self):
        return object_scope_type_registry.get("database_view")

    def get_base_queryset(self, include_trash: bool = False) -> QuerySet:
        return (
            super()
            .get_base_queryset(include_trash)
            .filter(view__table__database__workspace__isnull=False)
        )

    def get_enhanced_queryset(self, include_trash: bool = False) -> QuerySet:
        return self.get_base_queryset(include_trash).select_related(
            "view__table__database__workspace"
        )

    def get_filter_for_scope_type(self, scope_type, scopes):
        if scope_type.type == WorkspaceObjectScopeType.type:
            return Q(view_table__database__workspace__in=[s.id for s in scopes])

        if (
            scope_type.type == DatabaseObjectScopeType.type
            or scope_type.type == ApplicationObjectScopeType.type
        ):
            return Q(view__table__database__in=[s.id for s in scopes])

        if scope_type.type == DatabaseTableObjectScopeType.type:
            return Q(view__table__in=[s.id for s in scopes])

        if scope_type.type == DatabaseViewObjectScopeType.type:
            return Q(view__in=[s.id for s in scopes])

        raise TypeError("The given type is not handled.")


class DatabaseViewGroupByObjectScopeType(ObjectScopeType):
    type = "database_view_group"
    model_class = ViewGroupBy

    def get_parent_scope(self):
        return object_scope_type_registry.get("database_view")

    def get_base_queryset(self, include_trash: bool = False) -> QuerySet:
        return (
            super()
            .get_base_queryset(include_trash)
            .filter(view__table__database__workspace__isnull=False)
        )

    def get_enhanced_queryset(self, include_trash: bool = False) -> QuerySet:
        return self.get_base_queryset(include_trash).select_related(
            "view__table__database__workspace"
        )

    def get_filter_for_scope_type(self, scope_type, scopes):
        if scope_type.type == WorkspaceObjectScopeType.type:
            return Q(view_table__database__workspace__in=[s.id for s in scopes])

        if (
            scope_type.type == DatabaseObjectScopeType.type
            or scope_type.type == ApplicationObjectScopeType.type
        ):
            return Q(view__table__database__in=[s.id for s in scopes])

        if scope_type.type == DatabaseTableObjectScopeType.type:
            return Q(view__table__in=[s.id for s in scopes])

        if scope_type.type == DatabaseViewObjectScopeType.type:
            return Q(view__in=[s.id for s in scopes])

        raise TypeError("The given type is not handled.")