feature: checking if project exists in template type creation

weni-ai · Sep 21, 2023 · a4fb100 · a4fb100
1 parent aea1a48
commit a4fb100
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/chats/apps/projects/consumers/template_type_consumer.py b/chats/apps/projects/consumers/template_type_consumer.py
@@ -20,11 +20,6 @@ def consume(message: amqp.Message):
         print(f"[TemplateTypeConsumer] - Consuming a message. Body: {message.body}")
         body = JSONParser.parse(message.body)
 
-        try:
-            Project.objects.get(uuid=body.get("project_uuid"))
-        except Exception as err:
-            raise InvalidProjectData(err)
-
         template_type_creation = TemplateTypeCreation(config=body)
         template_type_creation.create()
 

diff --git a/chats/apps/projects/usecases/__init__.py b/chats/apps/projects/usecases/__init__.py
@@ -1,3 +1,3 @@
 from .dead_letter_handler import DeadLetterHandler
-from .exceptions import InvalidProjectData
+from .exceptions import InvalidProjectData, InvalidTemplateTypeData
 from .template_type_creation import TemplateTypeCreation
diff --git a/chats/apps/projects/usecases/template_type_creation.py b/chats/apps/projects/usecases/template_type_creation.py
@@ -1,12 +1,19 @@
 from chats.apps.projects.models import TemplateType
 from chats.apps.sectors.models import Sector
+from chats.apps.projects.models import Project
+from chats.apps.projects.usecases import InvalidTemplateTypeData
 
 
 class TemplateTypeCreation:
     def __init__(self, config: dict) -> None:
         self.config = config
 
     def create(self) -> TemplateType:
+        try:
+            Project.objects.get(uuid=self.config.get("project_uuid"))
+        except Exception as err:
+            raise InvalidTemplateTypeData(err)
+
         setup = {
             "sectors": [
                 sector.template_type_setup