add usecase to collaborative

Author: Deepthi-Chand

api/models/Collaborative.py

@@ -35,6 +35,7 @@ class Collaborative(models.Model):
         choices=CollaborativeStatus.choices,
     )
     datasets = models.ManyToManyField("api.Dataset", blank=True)
+    use_cases = models.ManyToManyField("api.UseCase", blank=True)
     tags = models.ManyToManyField("api.Tag", blank=True)
     sectors = models.ManyToManyField(
         "api.Sector", blank=True, related_name="collaboratives"

api/schema/collborative_schema.py

@@ -24,6 +24,7 @@
     Organization,
     Sector,
     Tag,
+    UseCase,
 )
 from api.schema.extensions import TrackActivity, TrackModelActivity
 from api.types.type_collaborative import (
@@ -37,6 +38,7 @@
 )
 from api.types.type_dataset import TypeDataset
 from api.types.type_organization import TypeOrganization
+from api.types.type_usecase import TypeUseCase
 from api.utils.enums import CollaborativeStatus, OrganizationRelationshipType
 from api.utils.graphql_telemetry import trace_resolver
 from authorization.models import User
@@ -453,6 +455,32 @@ def add_dataset_to_collaborative(
         collaborative.save()
         return TypeCollaborative.from_django(collaborative)
 
+    @strawberry_django.mutation(handle_django_errors=True)
+    def add_usecase_to_collaborative(
+        self, info: Info, collaborative_id: str, usecase_id: str
+    ) -> TypeCollaborative:
+        """Add a usecase to a collaborative."""
+        try:
+            usecase = UseCase.objects.get(id=usecase_id)
+        except UseCase.DoesNotExist:
+            raise ValueError(f"UseCase with ID {usecase_id} does not exist.")
+
+        try:
+            collaborative = Collaborative.objects.get(id=collaborative_id)
+        except Collaborative.DoesNotExist:
+            raise ValueError(
+                f"Collaborative with ID {collaborative_id} does not exist."
+            )
+
+        if collaborative.status != CollaborativeStatus.DRAFT:
+            raise ValueError(
+                f"Collaborative with ID {collaborative_id} is not in draft status."
+            )
+
+        collaborative.use_cases.add(usecase)
+        collaborative.save()
+        return TypeCollaborative.from_django(collaborative)
+
     @strawberry_django.mutation(handle_django_errors=True)
     def remove_dataset_from_collaborative(
         self, info: Info, collaborative_id: str, dataset_id: uuid.UUID
@@ -477,6 +505,30 @@ def remove_dataset_from_collaborative(
         collaborative.save()
         return TypeCollaborative.from_django(collaborative)
 
+    @strawberry_django.mutation(handle_django_errors=True)
+    def remove_usecase_from_collaborative(
+        self, info: Info, collaborative_id: str, usecase_id: str
+    ) -> TypeCollaborative:
+        """Remove a usecase from a collaborative."""
+        try:
+            usecase = UseCase.objects.get(id=usecase_id)
+        except UseCase.DoesNotExist:
+            raise ValueError(f"UseCase with ID {usecase_id} does not exist.")
+        try:
+            collaborative = Collaborative.objects.get(id=collaborative_id)
+        except Collaborative.DoesNotExist:
+            raise ValueError(
+                f"Collaborative with ID {collaborative_id} does not exist."
+            )
+
+        if collaborative.status != CollaborativeStatus.DRAFT:
+            raise ValueError(
+                f"Collaborative with ID {collaborative_id} is not in draft status."
+            )
+        collaborative.use_cases.remove(usecase)
+        collaborative.save()
+        return TypeCollaborative.from_django(collaborative)
+
     @strawberry_django.mutation(handle_django_errors=True)
     @trace_resolver(
         name="update_collaborative_datasets",
@@ -501,6 +553,30 @@ def update_collaborative_datasets(
         collaborative.save()
         return TypeCollaborative.from_django(collaborative)
 
+    @strawberry_django.mutation(handle_django_errors=True)
+    @trace_resolver(
+        name="update_collaborative_use_cases",
+        attributes={"component": "collaborative", "operation": "mutation"},
+    )
+    def update_collaborative_use_cases(
+        self, info: Info, collaborative_id: str, use_case_ids: List[uuid.UUID]
+    ) -> TypeCollaborative:
+        """Update the use cases of a collaborative."""
+        try:
+            use_cases = UseCase.objects.filter(id__in=use_case_ids)
+            collaborative = Collaborative.objects.get(id=collaborative_id)
+        except Collaborative.DoesNotExist:
+            raise ValueError(f"Collaborative with ID {collaborative_id} doesn't exist")
+
+        if collaborative.status != CollaborativeStatus.DRAFT:
+            raise ValueError(
+                f"Collaborative with ID {collaborative_id} is not in draft status."
+            )
+
+        collaborative.use_cases.set(use_cases)
+        collaborative.save()
+        return TypeCollaborative.from_django(collaborative)
+
     @strawberry_django.mutation(
         handle_django_errors=True,
         extensions=[

api/types/type_collaborative.py

@@ -20,6 +20,7 @@
 from api.types.type_dataset import TypeDataset, TypeTag
 from api.types.type_organization import TypeOrganization
 from api.types.type_sector import TypeSector
+from api.types.type_usecase import TypeUseCase
 from api.utils.enums import CollaborativeStatus, OrganizationRelationshipType
 from authorization.types import TypeUser
 
@@ -81,6 +82,17 @@ def datasets(self) -> Optional[List["TypeDataset"]]:
         except Exception:
             return []
 
+    @strawberry.field(description="Get use cases associated with this collaborative.")
+    def use_cases(self) -> Optional[List["TypeUseCase"]]:
+        """Get use cases associated with this collaborative."""
+        try:
+            queryset = self.use_cases.all()  # type: ignore
+            if not queryset.exists():
+                return []
+            return TypeUseCase.from_django_list(queryset)
+        except Exception:
+            return []
+
     @strawberry.field(
         description="Get the count of datasets associated with this collaborative."
     )
@@ -91,6 +103,16 @@ def dataset_count(self: "TypeCollaborative", info: Info) -> int:
         except Exception:
             return 0
 
+    @strawberry.field(
+        description="Get the count of use cases associated with this collaborative."
+    )
+    def use_case_count(self: "TypeCollaborative", info: Info) -> int:
+        """Get the count of use cases associated with this collaborative."""
+        try:
+            return self.use_cases.count()  # type: ignore
+        except Exception:
+            return 0
+
     @strawberry.field(description="Get publishers associated with this collaborative.")
     def publishers(self) -> Optional[List["TypeOrganization"]]:
         """Get publishers associated with this collaborative."""