add draft validation and expand update_usecase mutation

Author: Deepthi-Chand

api/schema/usecase_schema.py

@@ -64,6 +64,7 @@ class UseCaseInputPartial:
     """Input type for use case updates."""
 
     id: str
+    title: auto
     slug: auto
     summary: auto
 
@@ -241,7 +242,7 @@ class Mutation:
     """Mutations for use cases."""
 
     create_use_case: TypeUseCase = mutations.create(UseCaseInput)
-    update_use_case: TypeUseCase = mutations.update(UseCaseInputPartial, key_attr="id")
+    # update_use_case: TypeUseCase = mutations.update(UseCaseInputPartial, key_attr="id")
 
     @strawberry_django.mutation(
         handle_django_errors=True,
@@ -326,12 +327,41 @@ def add_update_usecase_metadata(
         except UseCase.DoesNotExist:
             raise ValueError(f"UseCase with ID {usecase_id} does not exist.")
 
+        if usecase.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {usecase_id} is not in draft status.")
+
         if update_metadata_input.tags is not None:
             _update_usecase_tags(usecase, update_metadata_input.tags)
         _add_update_usecase_metadata(usecase, metadata_input)
         _update_usecase_sectors(usecase, update_metadata_input.sectors)
         return TypeUseCase.from_django(usecase)
 
+    @strawberry_django.mutation(handle_django_errors=True)
+    @trace_resolver(
+        name="update_use_case",
+        attributes={"component": "usecase", "operation": "mutation"},
+    )
+    def update_use_case(
+        self, info: Info, use_case_input_partial: UseCaseInputPartial
+    ) -> TypeUseCase:
+        usecase_id = use_case_input_partial.id
+        try:
+            usecase = UseCase.objects.get(id=usecase_id)
+        except UseCase.DoesNotExist:
+            raise ValueError(f"UseCase with ID {usecase_id} does not exist.")
+
+        if usecase.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {usecase_id} is not in draft status.")
+
+        if use_case_input_partial.title == "":
+            raise ValueError("Title cannot be empty.")
+        if use_case_input_partial.title is not None:
+            usecase.title = use_case_input_partial.title
+        if use_case_input_partial.summary is not None:
+            usecase.summary = use_case_input_partial.summary
+        usecase.save()
+        return TypeUseCase.from_django(usecase)
+
     @strawberry_django.mutation(
         handle_django_errors=False,
         extensions=[
@@ -371,6 +401,9 @@ def add_dataset_to_use_case(
         except UseCase.DoesNotExist:
             raise ValueError(f"UseCase with ID {use_case_id} does not exist.")
 
+        if use_case.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {use_case_id} is not in draft status.")
+
         use_case.datasets.add(dataset)
         use_case.save()
         return TypeUseCase.from_django(use_case)
@@ -384,12 +417,13 @@ def remove_dataset_from_use_case(
             dataset = Dataset.objects.get(id=dataset_id)
         except Dataset.DoesNotExist:
             raise ValueError(f"Dataset with ID {dataset_id} does not exist.")
-
         try:
             use_case = UseCase.objects.get(id=use_case_id)
         except UseCase.DoesNotExist:
             raise ValueError(f"UseCase with ID {use_case_id} does not exist.")
 
+        if use_case.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {use_case_id} is not in draft status.")
         use_case.datasets.remove(dataset)
         use_case.save()
         return TypeUseCase.from_django(use_case)
@@ -409,6 +443,9 @@ def update_usecase_datasets(
         except UseCase.DoesNotExist:
             raise ValueError(f"Use Case with ID {use_case_id} doesn't exist")
 
+        if use_case.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {use_case_id} is not in draft status.")
+
         use_case.datasets.set(datasets)
         use_case.save()
         return TypeUseCase.from_django(use_case)
@@ -487,6 +524,9 @@ def add_contributor_to_use_case(
         except UseCase.DoesNotExist:
             raise ValueError(f"UseCase with ID {use_case_id} does not exist.")
 
+        if use_case.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {use_case_id} is not in draft status.")
+
         use_case.contributors.add(user)
         use_case.save()
         return TypeUseCase.from_django(use_case)
@@ -511,6 +551,9 @@ def remove_contributor_from_use_case(
         except UseCase.DoesNotExist:
             raise ValueError(f"UseCase with ID {use_case_id} does not exist.")
 
+        if use_case.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {use_case_id} is not in draft status.")
+
         use_case.contributors.remove(user)
         use_case.save()
         return TypeUseCase.from_django(use_case)
@@ -545,6 +588,9 @@ def update_usecase_contributors(
         except UseCase.DoesNotExist:
             raise ValueError(f"Use Case with ID {use_case_id} doesn't exist")
 
+        if use_case.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {use_case_id} is not in draft status.")
+
         use_case.contributors.set(users)
         use_case.save()
         return TypeUseCase.from_django(use_case)
@@ -569,6 +615,9 @@ def add_supporting_organization_to_use_case(
         except UseCase.DoesNotExist:
             raise ValueError(f"UseCase with ID {use_case_id} does not exist.")
 
+        if use_case.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {use_case_id} is not in draft status.")
+
         # Create or get the relationship
         relationship, created = UseCaseOrganizationRelationship.objects.get_or_create(
             usecase=use_case,
@@ -621,6 +670,9 @@ def add_partner_organization_to_use_case(
         except UseCase.DoesNotExist:
             raise ValueError(f"UseCase with ID {use_case_id} does not exist.")
 
+        if use_case.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {use_case_id} is not in draft status.")
+
         # Create or get the relationship
         relationship, created = UseCaseOrganizationRelationship.objects.get_or_create(
             usecase=use_case,
@@ -691,6 +743,9 @@ def update_usecase_organization_relationships(
         except UseCase.DoesNotExist:
             raise ValueError(f"UseCase with ID {use_case_id} does not exist.")
 
+        if use_case.status != UseCaseStatus.DRAFT:
+            raise ValueError(f"UseCase with ID {use_case_id} is not in draft status.")
+
         # Clear existing relationships
         UseCaseOrganizationRelationship.objects.filter(usecase=use_case).delete()