Merge pull request #208 from mapswipe/feat/delete-assets-bulk

Author: thenav56

apps/project/graphql/mutations.py

@@ -1,8 +1,10 @@
 import strawberry
 import strawberry_django
+from django.utils.translation import gettext
 from strawberry_django.permissions import IsAuthenticated
 
-from apps.project.models import Organization, Project
+from apps.common.models import AssetTypeEnum
+from apps.project.models import Organization, Project, ProjectAsset, ProjectAssetInputTypeEnum
 from apps.project.serializers import (
     OrganizationSerializer,
     ProcessedProjectUpdateSerializer,
@@ -13,7 +15,7 @@
 )
 from main.graphql.context import Info
 from utils.graphql.mutations import ModelMutation
-from utils.graphql.types import MutationResponseType
+from utils.graphql.types import CustomErrorType, MutationResponseType
 
 from .inputs.inputs import (
     OrganizationCreateInput,
@@ -24,7 +26,7 @@
     ProjectStatusUpdateInput,
     ProjectUpdateInput,
 )
-from .types.types import OrganizationType, ProjectAssetType, ProjectType
+from .types.types import OrganizationType, ProjectAssetsDeleteType, ProjectAssetType, ProjectType
 
 
 @strawberry.type
@@ -90,3 +92,44 @@ async def update_project_status(
     ) -> MutationResponseType[ProjectType]:
         project = await Project.objects.aget(pk=pk)
         return await ModelMutation(ProjectStatusUpdateSerializer).handle_update_mutation(data, info, project)
+
+    @strawberry_django.mutation(extensions=[IsAuthenticated()])
+    async def delete_project_assets(
+        self,
+        info: Info,
+        project_id: strawberry.ID,
+        asset_input_type: list[ProjectAssetInputTypeEnum],
+    ) -> MutationResponseType[ProjectAssetsDeleteType]:
+        project = await Project.objects.aget(pk=project_id)
+
+        if project.status_enum not in [Project.Status.DRAFT, Project.Status.PROCESSING_FAILED]:
+            return MutationResponseType(
+                ok=False,
+                errors=CustomErrorType(
+                    {
+                        "array_errors": None,
+                        "client_id": project.client_id,
+                        "field": "nonFieldErrors",
+                        "messages": gettext("Cannot delete assets of project with status %s") % project.status_enum.label,
+                    },
+                ),
+                result=None,
+            )
+
+        deleted_count = await (
+            ProjectAsset.usable_objects()
+            .filter(
+                project_id=project_id,
+                type=AssetTypeEnum.INPUT,
+                input_type__in=asset_input_type,
+            )
+            .aupdate(marked_as_deleted=True)
+        )
+
+        return MutationResponseType(
+            ok=True,
+            errors=None,
+            result=ProjectAssetsDeleteType(
+                count=deleted_count,
+            ),
+        )

apps/project/graphql/queries.py

@@ -7,7 +7,7 @@
 
 from apps.project.custom_options import get_custom_options
 from apps.project.graphql.inputs.inputs import ProjectNameInput
-from apps.project.models import Organization, Project, ProjectTypeEnum
+from apps.project.models import Organization, Project, ProjectAsset, ProjectTypeEnum
 from utils.geo.raster_tile_server.config import RasterConfig, RasterTileServerNameEnum, RasterTileServerNameEnumWithoutCustom
 from utils.geo.vector_tile_server.config import VectorConfig, VectorTileServerNameEnum, VectorTileServerNameEnumWithoutCustom
 
@@ -88,11 +88,19 @@ def default_custom_options(self, project_type: ProjectTypeEnum) -> list[CustomOp
     project_asset: ProjectAssetType = strawberry_django.field(extensions=[IsAuthenticated()])
 
     # --- Paginated
-    project_assets: OffsetPaginated[ProjectAssetType] = strawberry_django.offset_paginated(
+    @strawberry_django.offset_paginated(
+        OffsetPaginated[ProjectAssetType],
         order=ProjectAssetOrder,
         filters=ProjectAssetFilter,
         extensions=[IsAuthenticated()],
     )
+    def project_assets(
+        self,
+        include_all: bool = False,
+    ) -> QuerySet[ProjectAsset]:
+        if include_all:
+            return ProjectAsset.objects.all()
+        return ProjectAsset.usable_objects()
 
     organization: OrganizationType = strawberry_django.field(extensions=[IsAuthenticated()])
     public_organization: OrganizationType = strawberry_django.field()

apps/project/graphql/types/types.py

@@ -142,6 +142,11 @@ async def export_moderate_to_high_agreement_yes_maybe_geometries(
         return await info.context.dl.project.export.moderate_to_high_agreement_yes_maybe_geometries.load(project.pk)
 
 
+@strawberry.type
+class ProjectAssetsDeleteType:
+    count: int
+
+
 @strawberry_django.type(Project)
 class ProjectType(UserResourceTypeMixin, ProjectExportAssetTypeMixin, FirebasePushResourceTypeMixin):
     id: strawberry.ID

apps/tutorial/graphql/queries.py

@@ -4,7 +4,7 @@
 from strawberry_django.pagination import OffsetPaginated
 from strawberry_django.permissions import IsAuthenticated
 
-from apps.tutorial.models import Tutorial
+from apps.tutorial.models import Tutorial, TutorialAsset
 
 from .filters import TutorialAssetFilter, TutorialFilter
 from .orders import TutorialAssetOrder, TutorialOrder
@@ -35,8 +35,16 @@ def tutorials(
     tutorial_asset: TutorialAssetType = strawberry_django.field(extensions=[IsAuthenticated()])
 
     # --- Paginated
-    tutorial_assets: OffsetPaginated[TutorialAssetType] = strawberry_django.offset_paginated(
+    @strawberry_django.offset_paginated(
+        OffsetPaginated[TutorialAssetType],
         order=TutorialAssetOrder,
         filters=TutorialAssetFilter,
         extensions=[IsAuthenticated()],
     )
+    def tutorial_assets(
+        self,
+        include_all: bool = False,
+    ) -> QuerySet[TutorialAsset]:
+        if include_all:
+            return TutorialAsset.objects.all()
+        return TutorialAsset.usable_objects()

schema.graphql

@@ -771,6 +771,8 @@ input DeleteInput {
   id: ID!
 }
 
+union DeleteProjectAssetsPayload = OperationInfo | ProjectAssetsDeleteTypeMutationResponseType
+
 union DeleteProjectPayload = OperationInfo | ProjectType
 
 union DeleteTutorialPayload = OperationInfo | TutorialType
@@ -958,6 +960,7 @@ type Mutation {
   createTutorial(data: TutorialCreateInput!): CreateTutorialPayload! @isAuthenticated
   createTutorialAsset(data: TutorialAssetCreateInput!): CreateTutorialAssetPayload! @isAuthenticated
   deleteProject: DeleteProjectPayload! @isAuthenticated
+  deleteProjectAssets(projectId: ID!, assetInputType: [ProjectAssetInputTypeEnum!]!): DeleteProjectAssetsPayload! @isAuthenticated
   deleteTutorial: DeleteTutorialPayload! @isAuthenticated
   login(username: String!, password: String!): UserMeType!
   logout: Boolean!
@@ -1388,6 +1391,16 @@ type ProjectAssetTypeOffsetPaginated {
   totalCount: Int!
 }
 
+type ProjectAssetsDeleteType {
+  count: Int!
+}
+
+type ProjectAssetsDeleteTypeMutationResponseType {
+  errors: CustomErrorType
+  ok: Boolean!
+  result: ProjectAssetsDeleteType
+}
+
 """Model representing the project."""
 input ProjectCreateInput {
   """
@@ -2033,7 +2046,7 @@ type Query {
   organizations(includeAll: Boolean! = false, filters: OrganizationFilter, order: OrganizationOrder, pagination: OffsetPaginationInput): OrganizationTypeOffsetPaginated! @isAuthenticated
   project(id: ID!): ProjectType! @isAuthenticated
   projectAsset(id: ID!): ProjectAssetType! @isAuthenticated
-  projectAssets(pagination: OffsetPaginationInput, filters: ProjectAssetFilter, order: ProjectAssetOrder): ProjectAssetTypeOffsetPaginated! @isAuthenticated
+  projectAssets(includeAll: Boolean! = false, filters: ProjectAssetFilter, order: ProjectAssetOrder, pagination: OffsetPaginationInput): ProjectAssetTypeOffsetPaginated! @isAuthenticated
   projectName(params: ProjectNameInput): String!
   projects(includeAll: Boolean! = false, filters: ProjectFilter, order: ProjectOrder, pagination: OffsetPaginationInput): ProjectTypeOffsetPaginated! @isAuthenticated
   publicOrganization(id: ID!): OrganizationType!
@@ -2043,7 +2056,7 @@ type Query {
   tileServers: RasterTileServersType! @isAuthenticated
   tutorial(id: ID!): TutorialType! @isAuthenticated
   tutorialAsset(id: ID!): TutorialAssetType! @isAuthenticated
-  tutorialAssets(pagination: OffsetPaginationInput, filters: TutorialAssetFilter, order: TutorialAssetOrder): TutorialAssetTypeOffsetPaginated! @isAuthenticated
+  tutorialAssets(includeAll: Boolean! = false, filters: TutorialAssetFilter, order: TutorialAssetOrder, pagination: OffsetPaginationInput): TutorialAssetTypeOffsetPaginated! @isAuthenticated
   tutorials(includeAll: Boolean! = false, filters: TutorialFilter, order: TutorialOrder, pagination: OffsetPaginationInput): TutorialTypeOffsetPaginated! @isAuthenticated
   users(filters: UserFilter, order: UserOrder, pagination: OffsetPaginationInput): UserTypeOffsetPaginated! @isAuthenticated
 }