Incorrect Error Message for Duplicate Module Name

backend/apps/mentorship/api/internal/mutations/module.py

@@ -6,6 +6,7 @@
 import strawberry
 from django.core.exceptions import ObjectDoesNotExist, PermissionDenied, ValidationError
 from django.db import transaction
+from django.db.utils import IntegrityError
 from django.utils import timezone
 
 from apps.github.models import User as GithubUser
@@ -100,18 +101,26 @@ def create_module(self, info: strawberry.Info, input_data: CreateModuleInput) ->
             program.ended_at,
         )
 
-        module = Module.objects.create(
-            name=input_data.name,
-            description=input_data.description,
-            experience_level=input_data.experience_level.value,
-            started_at=started_at,
-            ended_at=ended_at,
-            domains=input_data.domains,
-            labels=input_data.labels,
-            tags=input_data.tags,
-            program=program,
-            project=project,
-        )
+        try:
+            with transaction.atomic():
+                module = Module.objects.create(
+                    name=input_data.name,
+                    description=input_data.description,
+                    experience_level=input_data.experience_level.value,
+                    started_at=started_at,
+                    ended_at=ended_at,
+                    domains=input_data.domains,
+                    labels=input_data.labels,
+                    tags=input_data.tags,
+                    program=program,
+                    project=project,
+                )
+        except IntegrityError as e:
+            error_message = str(e)
+            if "unique_module_key_in_program" in error_message:
+                msg = "This module name already exists in this program"
+                raise ValidationError(message=msg) from e
+            raise
 
         if module.experience_level not in program.experience_levels:
             program.experience_levels.append(module.experience_level)
@@ -384,7 +393,15 @@ def update_module(self, info: strawberry.Info, input_data: UpdateModuleInput) ->
             mentors_to_set = resolve_mentors_from_logins(input_data.mentor_logins)
             module.mentors.set(mentors_to_set)
 
-        module.save()
+        try:
+            with transaction.atomic():
+                module.save()
+        except IntegrityError as e:
+            error_message = str(e)
+            if "unique_module_key_in_program" in error_message:
+                msg = "This module name already exists in this program"
+                raise ValidationError(message=msg) from e
+            raise
 
         if module.experience_level not in module.program.experience_levels:
             module.program.experience_levels.append(module.experience_level)