Add find_by_id method to repository implementations: Implemented find_by_id methods in OrderRepositoryImpl, ParticipantRepositoryImpl, and ProductRepositoryImpl to support ID retrieval using both string and UUID types. Enhanced error handling and ID conversion logic for improved robustness.

Author: maxsonferovante

src/infrastructure/repository/order_repository_impl.py

@@ -1,6 +1,7 @@
 import json
 import logging
-from typing import List, Optional
+import uuid
+from typing import List, Optional, Union
 from src.domain.entity.order import Order
 from src.domain.repository.order_repository import OrderRepository
 from src.infrastructure.aws.dynamodb_service import DynamoDBService
@@ -37,6 +38,35 @@ def get_by_id(self, entity_id: int) -> Optional[Order]:
             logger.error(f"Erro ao buscar pedido por ID {entity_id}: {str(e)}")
             raise
 
+    def find_by_id(self, entity_id: Union[str, uuid.UUID]) -> Optional[Order]:
+        """
+        Busca pedido por ID, aceitando tanto string quanto UUID.
+        Implementa o método abstrato definido na classe base BaseRepository.
+        Nota: Como Order usa int como ID, converte string/UUID para int.
+        """
+        try:
+            # Converte para string primeiro, depois para int
+            if isinstance(entity_id, uuid.UUID):
+                # Se for UUID, converte para string e depois tenta extrair um int
+                # Isso pode não ser ideal - talvez precise de uma lógica específica
+                id_str = str(entity_id)
+                # Por enquanto, vamos tentar usar os primeiros dígitos como int
+                # Isso é uma solução temporária - talvez precise ajustar conforme a lógica de negócio
+                id_int = int(id_str.replace('-', '')[:10])  # Pega os primeiros 10 dígitos
+            else:
+                # Se for string, tenta converter para int
+                id_int = int(str(entity_id))
+            
+            # Reutiliza a lógica existente do get_by_id
+            return self.get_by_id(id_int)
+            
+        except (ValueError, TypeError) as e:
+            logger.error(f"Erro ao converter ID {entity_id} para int: {str(e)}")
+            return None
+        except Exception as e:
+            logger.error(f"Erro ao buscar pedido por ID {entity_id}: {str(e)}")
+            raise
+    
     def get_all(self) -> List[Order]:
 
         try:

src/infrastructure/repository/participant_repository_impl.py

@@ -1,6 +1,7 @@
 import json
 import logging
-from typing import List, Optional
+import uuid
+from typing import List, Optional, Union
 from src.domain.entity.participant import Participant
 from src.domain.repository.participant_repository import ParticipantRepository
 from src.infrastructure.aws.dynamodb_service import DynamoDBService
@@ -35,6 +36,22 @@ def get_by_id(self, entity_id: str) -> Optional[Participant]:
             logger.error(f"Erro ao buscar participante por ID {entity_id}: {str(e)}")
             raise
 
+    def find_by_id(self, entity_id: Union[str, uuid.UUID]) -> Optional[Participant]:
+        """
+        Busca participante por ID, aceitando tanto string quanto UUID.
+        Implementa o método abstrato definido na classe base BaseRepository.
+        """
+        try:
+            # Converte UUID para string se necessário
+            id_str = str(entity_id) if isinstance(entity_id, uuid.UUID) else entity_id
+            
+            # Reutiliza a lógica existente do get_by_id
+            return self.get_by_id(id_str)
+            
+        except Exception as e:
+            logger.error(f"Erro ao buscar participante por ID {entity_id}: {str(e)}")
+            raise
+    
     def get_all(self) -> List[Participant]:        
         try:
             items = self.dynamodb_service.scan_table(self.table_name)

src/infrastructure/repository/product_repository_impl.py

@@ -1,6 +1,7 @@
 import json
 import logging
-from typing import List, Optional
+import uuid
+from typing import List, Optional, Union
 from src.domain.entity.product import Product
 from src.domain.repository.product_repository import ProductRepository
 from src.infrastructure.aws.dynamodb_service import DynamoDBService
@@ -41,6 +42,33 @@ def get_by_id(self, entity_id: int) -> Optional[Product]:
             logger.error(f"Erro ao buscar produto por ID {entity_id}: {str(e)}")
             raise
 
+    def find_by_id(self, entity_id: Union[str, uuid.UUID]) -> Optional[Product]:
+        """
+        Busca produto por ID, aceitando tanto string quanto UUID.
+        Implementa o método abstrato definido na classe base BaseRepository.
+        Nota: Como Product usa int como ID, converte string/UUID para int.
+        """
+        try:
+            # Converte para string primeiro, depois para int
+            if isinstance(entity_id, uuid.UUID):
+                # Se for UUID, converte para string e depois tenta extrair um int
+                id_str = str(entity_id)
+                # Tenta usar os primeiros dígitos como int
+                id_int = int(id_str.replace('-', '')[:10])  # Pega os primeiros 10 dígitos
+            else:
+                # Se for string, tenta converter para int
+                id_int = int(str(entity_id))
+            
+            # Reutiliza a lógica existente do get_by_id
+            return self.get_by_id(id_int)
+            
+        except (ValueError, TypeError) as e:
+            logger.error(f"Erro ao converter ID {entity_id} para int: {str(e)}")
+            return None
+        except Exception as e:
+            logger.error(f"Erro ao buscar produto por ID {entity_id}: {str(e)}")
+            raise
+    
     def get_all(self) -> List[Product]:
         try:
             items = self.dynamodb_service.scan_table(self.table_name)