Add configuration management and AWS service integration

maxsonferovante · maxsonferovante · commit 3351787f1049 · 2025-03-28T09:30:26.000-03:00
- Introduced a new config.py file for managing environment variables using Pydantic.
- Implemented S3Service and SQSService classes for AWS S3 and SQS interactions, respectively.
- Updated lambda_function.py to utilize these services for file uploads and message sending.
- Enhanced the CertifiedBuilder class to generate certificate keys and handle file uploads.
- Updated requirements.txt to include necessary AWS SDK dependencies.
diff --git a/aws/__init__.py b/aws/__init__.py
diff --git a/aws/boto_aws.py b/aws/boto_aws.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
+    )
diff --git a/aws/s3_service.py b/aws/s3_service.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
diff --git a/aws/sqs_service.py b/aws/sqs_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
diff --git a/certified_builder/certified_builder.py b/certified_builder/certified_builder.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
diff --git a/config.py b/config.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()
+
+
diff --git a/lambda_function.py b/lambda_function.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 {
@@ -161,3 +181,48 @@ def lambda_handler(event, context):
             })
         }
 
+if __name__ == "__main__":
+    lambda_handler({
+	"Records": [
+		{
+			"body": [
+				{
+					"order_id": 452,
+					"first_name": "Jardel",
+					"last_name": "Godinho",
+					"email": "jardelgodinho@gmail.com",
+					"phone": "(48) 98866-7447",
+					"cpf": "",
+					"city": "São José",
+					"product_id": 316,
+					"product_name": "Evento de Teste",
+					"certificate_details": "In recognition of their participation in the 84st edition of the Python Floripa Community Meeting, held on March 29, 2025, in Florianópolis, Brazil.",
+					"certificate_logo": "https://tech.floripa.br/wp-content/uploads/2025/03/84o-Python-Floripa-e1741729144453.png",
+					"certificate_background": "https://tech.floripa.br/wp-content/uploads/2025/03/Background.png",
+					"order_date": "2025-03-26 20:55:25",
+					"checkin_latitude": "-27.5460492",
+					"checkin_longitude": "-48.6227075",
+					"time_checkin": "2025-03-26 20:55:44"
+				},
+				{
+					"order_id": 317,
+					"first_name": "Jardel",
+					"last_name": "Godinho",
+					"email": "jardel.godinho@gmail.com",
+					"phone": "(48) 98866-7447",
+					"cpf": "",
+					"city": "",
+					"product_id": 316,
+					"product_name": "Evento de Teste",
+					"certificate_details": "In recognition of their participation in the 84st edition of the Python Floripa Community Meeting, held on March 29, 2025, in Florianópolis, Brazil.",
+					"certificate_logo": "https://tech.floripa.br/wp-content/uploads/2025/03/84o-Python-Floripa-e1741729144453.png",
+					"certificate_background": "https://tech.floripa.br/wp-content/uploads/2025/03/Background.png",
+					"order_date": "2025-03-19 06:04:51",
+					"checkin_latitude": "-27.5460492",
+					"checkin_longitude": "-48.6227075",
+					"time_checkin": "2025-03-19 16:04:16"
+				}
+			]
+		}
+	]
+}, {})
diff --git a/requirements.txt b/requirements.txt
@@ -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
