This commit is contained in:
2025-07-10 17:36:51 -03:00
parent 72b1338135
commit bd6fbf7166
19 changed files with 250 additions and 16 deletions

View File

@@ -11,3 +11,4 @@ def get_dynamodb_client():
dynamodb_client = get_dynamodb_client()
sesv2_client = boto3.client('sesv2')

View File

@@ -5,8 +5,9 @@ ORDER_TABLE: str = os.getenv('ORDER_TABLE') # type: ignore
ENROLLMENT_TABLE: str = os.getenv('ENROLLMENT_TABLE') # type: ignore
COURSE_TABLE: str = os.getenv('COURSE_TABLE') # type: ignore
EMAIL_SENDER = ('EDUSEG', 'noreply@eduseg.com.br')
# Post-migration: remove the lines below
# Post-migration: Remove the following lines
if os.getenv('AWS_LAMBDA_FUNCTION_NAME'):
SQLITE_DATABASE = 'courses_export_2025-06-18_110214.db'
else:

View File

@@ -0,0 +1,64 @@
from email.mime.application import MIMEApplication
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from email.utils import formataddr
from pathlib import Path
class Message:
def __init__(
self,
from_: tuple[str | None, str],
to: tuple[str | None, str],
subject: str,
reply_to: tuple[str | None, str] | None = None,
content: str | None = None,
) -> None:
self._references = set()
self._body = MIMEMultipart('alternative')
self._message = MIMEMultipart('mixed')
self._message['From'] = formataddr(from_)
self._message['To'] = formataddr(to)
self._message['Subject'] = subject
self._message.attach(self._body)
if reply_to:
self._message['Reply-To'] = formataddr(reply_to)
if content:
self.add_alternative(content, subtype='plain')
def add_header(self, name: str, value: str) -> None:
self._message.add_header(name, value)
def add_in_reply_to(self, message_id: str) -> None:
self._message['In-Reply-To'] = message_id
self._references.add(message_id) # Add to set avoids duplicates
def add_alternative(
self,
text: str,
/,
subtype: str = 'html',
charset: str = 'utf-8',
) -> None:
self._body.attach(MIMEText(text, subtype, charset))
def attach(self, path: Path, filename: str | None = None) -> None:
if not path.is_file():
return None
with path.open('rb') as fp:
part = MIMEApplication(fp.read())
part.add_header(
'Content-Disposition',
'attachment',
filename=filename or path.name,
)
self._message.attach(part)
def as_bytes(self) -> bytes:
if self._references:
self._message['References'] = ' '.join(self._references)
return self._message.as_bytes()

View File

@@ -0,0 +1,94 @@
from aws_lambda_powertools import Logger
from aws_lambda_powertools.utilities.data_classes import (
EventBridgeEvent,
event_source,
)
from aws_lambda_powertools.utilities.typing import LambdaContext
from layercake.dateutils import now
from layercake.dynamodb import ComposeKey, DynamoDBPersistenceLayer, KeyPair
from layercake.strutils import first_word, truncate_str
from boto3clients import dynamodb_client, sesv2_client
from config import (
EMAIL_SENDER,
ENROLLMENT_TABLE,
)
from email_ import Message
logger = Logger(__name__)
enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
SUBJECT = 'Seu curso de {course} está esperando por você na EDUSEG®'
MESSAGE = """
Oi {first_name}, tudo bem?<br/><br/>
Há 3 dias você foi matriculado no curso de <b>{course}</b>, mas ainda não iniciou.<br/>
Não perca a oportunidade de aprender e aproveitar ao máximo seu curso!<br/><br/>
Clique no link abaixo para acessar seu curso:
<a href="https://saladeaula.digital">https://saladeaula.digital</a>
"""
@event_source(data_class=EventBridgeEvent)
@logger.inject_lambda_context
def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
old_image = event.detail['old_image']
now_ = now()
# Post-migration: Remove the following lines
if 'email' not in old_image:
data = enrollment_layer.get_item(KeyPair(old_image['id'], '0'))
old_image['name'] = data['user']['name']
old_image['email'] = data['user']['email']
old_image['course'] = data['course']['name']
emailmsg = Message(
from_=EMAIL_SENDER,
to=(
old_image['name'],
old_image['email'],
),
subject=SUBJECT.format(course=truncate_str(old_image['course'])),
)
emailmsg.add_alternative(
MESSAGE.format(
first_name=first_word(old_image['name']),
course=old_image['course'],
)
)
try:
sesv2_client.send_email(
Content={
'Raw': {
'Data': emailmsg.as_bytes(),
},
}
)
except Exception as exc:
logger.exception(exc)
enrollment_layer.put_item(
item={
'id': old_image['id'],
'sk': ComposeKey('failed', 'schedules#reminder_no_access_3_days'),
# Post-migration: Uncomment the following line
# 'sk': ComposeKey('failed', old_image['sk']),
'created_at': now_,
}
)
return False
else:
enrollment_layer.put_item(
item={
'id': old_image['id'],
'sk': ComposeKey('completed', 'schedules#reminder_no_access_3_days'),
# Post-migration: Uncomment the following line
# 'sk': ComposeKey('completed', old_image['sk']),
'created_at': now_,
}
)
return True

View File

@@ -8,13 +8,11 @@ from layercake.dynamodb import DynamoDBPersistenceLayer
from boto3clients import dynamodb_client
from config import (
COURSE_TABLE,
ENROLLMENT_TABLE,
)
logger = Logger(__name__)
enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
course_layer = DynamoDBPersistenceLayer(COURSE_TABLE, dynamodb_client)
@event_source(data_class=EventBridgeEvent)

View File

@@ -62,6 +62,36 @@ Resources:
new_image:
sk: ["0"]
EventReminderNoAccess3DaysFunction:
Type: AWS::Serverless::Function
Properties:
Handler: events.emails.reminder_no_access_3_days.lambda_handler
LoggingConfig:
LogGroup: !Ref EventLog
Policies:
- DynamoDBCrudPolicy:
TableName: !Ref EnrollmentTable
- Version: 2012-10-17
Statement:
- Effect: Allow
Action:
- ses:SendRawEmail
Resource:
- !Sub arn:aws:ses:${AWS::Region}:${AWS::AccountId}:identity/eduseg.com.br
- !Sub arn:aws:ses:${AWS::Region}:${AWS::AccountId}:configuration-set/tracking
Events:
DynamoDBEvent:
Type: EventBridgeRule
Properties:
Pattern:
resources: [!Ref EnrollmentTable]
detail-type: [EXPIRE]
detail:
keys:
sk:
- schedules#does_not_access
- schedules#reminder_no_access_3_days
EventIssueCertFunction:
Type: AWS::Serverless::Function
Properties:

View File

@@ -18,8 +18,6 @@ def pytest_configure():
os.environ['COURSE_TABLE'] = PYTEST_TABLE_NAME
os.environ['ORDER_TABLE'] = PYTEST_TABLE_NAME
os.environ['ENROLLMENT_TABLE'] = PYTEST_TABLE_NAME
# Post-migration: remove it
os.environ['OLD_ENROLLMENT_TABLE'] = PYTEST_TABLE_NAME
@dataclass

View File

@@ -0,0 +1,19 @@
import app.events.emails.reminder_no_access_3_days as app
from aws_lambda_powertools.utilities.typing import LambdaContext
def test_reminder_no_access_3_days(
dynamodb_client,
dynamodb_seeds,
lambda_context: LambdaContext,
):
event = {
'detail': {
'new_image': {
'id': '47ZxxcVBjvhDS5TE98tpfQ',
'sk': 'schedules#reminder_no_access_3_days',
}
}
}
assert app.lambda_handler(event, lambda_context)