Merge pull request #11 from maxsonferovante/feature/tech-floripa

Feature/tech floripa

Author: maxsonferovante

aws/__init__.py

@@ -0,0 +1,15 @@
+from boto3 import client
+from enum import Enum
+from config import config
+
+class ServiceNameAWS(Enum):
+    S3 = 's3'
+    SQS = 'sqs'
+
+def get_instance_aws(service_name: ServiceNameAWS):
+    return client(
+        service_name.value,
+        region_name=config.REGION,
+        aws_access_key_id=config.KEY_ACCESS,
+        aws_secret_access_key=config.KEY_SECRET
+    )

aws/boto_aws.py

@@ -0,0 +1,24 @@
+from aws.boto_aws import get_instance_aws, ServiceNameAWS
+from config import config
+import logging
+
+logger = logging.getLogger(__name__)
+
+class S3Service:
+    def __init__(self):
+        self.aws = get_instance_aws(
+            ServiceNameAWS.S3
+        )
+        self.bucket_name = config.BUCKET_NAME
+
+    def upload_file(self, file_path: str, key: str):
+        try:
+            response = self.aws.upload_file(
+                file_path,
+                self.bucket_name,
+                key
+            )
+            logger.info(f"Arquivo {file_path} enviado para o bucket {self.bucket_name} com sucesso")
+        except Exception as e:
+            logger.error(f"Erro ao enviar o arquivo {file_path} para o bucket {self.bucket_name}: {e}")
+            raise e

aws/s3_service.py

@@ -0,0 +1,32 @@
+from aws.boto_aws import get_instance_aws, ServiceNameAWS
+from config import config
+from typing import Dict
+import json
+import logging
+import boto3
+from botocore.exceptions import ClientError
+
+logger = logging.getLogger(__name__)
+
+class SQSService:
+    def __init__(self):
+        self.aws = get_instance_aws(ServiceNameAWS.SQS)
+        self.queue_url = config.QUEUE_URL        
+
+    def send_message(self, message: Dict):
+        try:
+            # Add MessageGroupId for FIFO queue
+            # Using order_id as MessageGroupId to ensure messages for the same order are processed in order
+            message_group_id = str(message.get('order_id', 'default'))
+            
+            response = self.aws.send_message(
+                QueueUrl=self.queue_url,
+                MessageBody=json.dumps(message),
+                MessageGroupId=message_group_id,
+                MessageDeduplicationId=f"{message_group_id}_{message.get('email', '')}"  # Ensure unique messages per order/email
+            )
+            logger.info(f"Mensagem enviada com sucesso: {response['MessageId']}")
+            return response
+        except ClientError as e:
+            logger.error(f"Erro ao enviar mensagem para a fila {self.queue_url}: {str(e)}")
+            raise

aws/sqs_service.py

@@ -6,7 +6,6 @@
 import os
 from certified_builder.utils.fetch_file_certificate import fetch_file_certificate
 import tempfile
-
 FONT_NAME = os.path.join(os.path.dirname(__file__), "fonts/PinyonScript/PinyonScript-Regular.ttf")
 VALIDATION_CODE = os.path.join(os.path.dirname(__file__), "fonts/ChakraPetch/ChakraPetch-SemiBold.ttf")
 DETAILS_FONT = os.path.join(os.path.dirname(__file__), "fonts/ChakraPetch/ChakraPetch-Regular.ttf")
@@ -19,6 +18,7 @@ def __init__(self):
         # Ensure temp directory exists
         self.temp_dir = "/tmp/certificates"
         os.makedirs(self.temp_dir, exist_ok=True)
+
 
     def build_certificates(self, participants: List[Participant]):
         """Build certificates for all participants."""
@@ -39,8 +39,10 @@ def build_certificates(self, participants: List[Participant]):
                     results.append({
                         "participant": participant.model_dump(),
                         "certificate_path": certificate_path,
+                        "certificate_key": f"certificates/{participant.event.product_id}/{participant.event.order_id}/{participant.create_name_certificate()}",
                         "success": True
                     })
+                                        
 
                     logger.info(f"Certificado gerado para {participant.name_completed()} com codigo de validação {participant.formated_validation_code()}")
                 except Exception as e:
@@ -50,7 +52,7 @@ def build_certificates(self, participants: List[Participant]):
                         "error": str(e),
                         "success": False
                     })
-            
+                
             return results
         except Exception as e:
             logger.error(f"Erro geral na geração de certificados: {str(e)}")
@@ -186,12 +188,11 @@ def save_certificate(self, certificate: Image, participant: Participant) -> str:
         try:
             name_certificate = participant.create_name_certificate()
             file_path = os.path.join(self.temp_dir, name_certificate)
-            
             # Optimize image before saving
             certificate = certificate.convert('RGB')
-            certificate.save(file_path, format="PNG", optimize=True)
-            
+            certificate.save(file_path, format="PNG", optimize=True)            
             return file_path
+            
         except Exception as e:
             logger.error(f"Erro ao salvar certificado: {str(e)}")
             raise

certified_builder/certified_builder.py

@@ -0,0 +1,21 @@
+import logging
+from pydantic_settings import BaseSettings
+
+logger = logging.getLogger(__name__)
+
+class Config(BaseSettings):
+    KEY_ACCESS: str
+    KEY_SECRET: str
+    REGION: str
+    BUCKET_NAME: str    
+    QUEUE_URL: str
+
+
+    class Config:
+        env_file = ".env"
+        env_file_encoding = "utf-8"        
+
+
+config = Config()
+
+

config.py

@@ -7,6 +7,8 @@
 from datetime import datetime
 import base64
 import os
+from aws.s3_service import S3Service
+from aws.sqs_service import SQSService
 
 # Configure logging for CloudWatch
 logger = logging.getLogger()
@@ -92,6 +94,7 @@ def format_result(result):
 def lambda_handler(event, context):
     # Log the start of the Lambda execution
     try:
+
         logger.info("Starting Lambda execution")    
         body = extract_data_body(event)
         participants_data = body
@@ -105,7 +108,9 @@ def lambda_handler(event, context):
                     'message': 'Nenhum participante encontrado para processamento'
                 })
             }
-        
+        s3_service = S3Service()
+        sqs_service = SQSService()
+
         logger.info(f"Processing {len(participants_data)} participants")
 
         # Create list of participants
@@ -129,8 +134,23 @@ def lambda_handler(event, context):
             builder = CertifiedBuilder()
             certificates_results = builder.build_certificates(participants)
             # Format results before adding to response
-            formatted_results = [format_result(result) for result in certificates_results]
-            results.extend(formatted_results)
+            formatted_results = []
+            
+            for result in certificates_results:
+                formatted_results.extend(format_result(result))            
+                
+                if result.get('success'):
+                    s3_service.upload_file(result.get('certificate_path'), result.get('certificate_key'))
+                
+                sqs_service.send_message({
+                    "order_id": result.get('participant', {}).get('event', {}).get('order_id', ""),
+                    "product_id": result.get('participant', {}).get('event', {}).get('product_id', ""),
+                    "product_name": result.get('participant', {}).get('event', {}).get('product_name', ""),
+                    "email": result.get('participant', {}).get('email', ""),
+                    "certificate_key": result.get('certificate_key', ""),
+                    "success": result.get('success', False)
+                })
+            
 
             logger.info("Certificados gerados com sucesso")
             return {
@@ -160,4 +180,3 @@ def lambda_handler(event, context):
                 'message': 'Erro ao gerar certificados'
             })
         }
-

lambda_function.py

@@ -1,5 +1,7 @@
 annotated-types==0.7.0
 anyio==4.6.2.post1
+boto3==1.37.22
+botocore==1.37.22
 certifi==2024.8.30
 dnspython==2.7.0
 email_validator==2.2.0
@@ -8,11 +10,18 @@ httpcore==1.0.7
 httpx==0.27.2
 idna==3.10
 iniconfig==2.0.0
+jmespath==1.0.1
 packaging==24.2
 pillow==11.0.0
 pluggy==1.5.0
 pydantic==2.10.1
+pydantic-settings==2.8.1
 pydantic_core==2.27.1
 pytest==8.3.3
+python-dateutil==2.9.0.post0
+python-dotenv==1.1.0
+s3transfer==0.11.4
+six==1.17.0
 sniffio==1.3.1
 typing_extensions==4.12.2
+urllib3==2.3.0