diff --git a/enrollments-events/app/events/emails/email_.py b/enrollments-events/app/events/emails/email_.py new file mode 100644 index 0000000..e5f2562 --- /dev/null +++ b/enrollments-events/app/events/emails/email_.py @@ -0,0 +1,71 @@ +from typing import TYPE_CHECKING, TypedDict + +from aws_lambda_powertools import Logger +from layercake.dateutils import now +from layercake.dynamodb import DynamoDBPersistenceLayer +from layercake.email_ import Message +from layercake.strutils import first_word, truncate_str + +logger = Logger(__name__) + +if TYPE_CHECKING: + from mypy_boto3_sesv2 import SESV2Client +else: + SESV2Client = object + +Event = TypedDict('Event', {'id': str, 'sk': str}) + + +def send_email( + to: tuple[str, str], + subject: str, + message: str, + context: dict = {}, + *, + sender: tuple[str, str], + event: Event, + sesv2_client: SESV2Client, + dynamodb_persistence_layer: DynamoDBPersistenceLayer, +) -> bool: + now_ = now() + name, _ = to + emailmsg = Message( + from_=sender, + to=to, + subject=subject.format(course=truncate_str(context['course'])), + ) + emailmsg.add_alternative( + message.format(first_name=first_word(name), course=context['course']) + ) + + try: + sesv2_client.send_email( + Content={ + 'Raw': { + 'Data': emailmsg.as_bytes(), + }, + } + ) + logger.info('Email sent') + except Exception as exc: + logger.exception(exc) + + dynamodb_persistence_layer.put_item( + item={ + 'id': event['id'], + 'sk': f'{event["sk"]}#FAILED', + 'created_at': now_, + } + ) + + return False + else: + dynamodb_persistence_layer.put_item( + item={ + 'id': event['id'], + 'sk': f'{event["sk"]}#EXECUTED', + 'created_at': now_, + } + ) + + return True diff --git a/enrollments-events/app/events/emails/reminder_access_period_before_30_days.py b/enrollments-events/app/events/emails/reminder_access_period_before_30_days.py new file mode 100644 index 0000000..2722142 --- /dev/null +++ b/enrollments-events/app/events/emails/reminder_access_period_before_30_days.py @@ -0,0 +1,60 @@ +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.dynamodb import DynamoDBPersistenceLayer, KeyPair + +from boto3clients import dynamodb_client, sesv2_client +from config import ( + EMAIL_SENDER, + ENROLLMENT_TABLE, +) + +from .email_ import send_email + +logger = Logger(__name__) +enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client) + + +SUBJECT = 'Seu acesso ao curso de {course} termina em 30 dias' +MESSAGE = """ +Oi {first_name}, tudo bem?

+ +Faltam 30 dias para o término do seu acesso ao curso {course}.
+Conclua dentro desse prazo para garantir sua certificação.

+ +👉 Acesse agora seu curso +""" + + +@event_source(data_class=EventBridgeEvent) +@logger.inject_lambda_context +def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool: + """30 days before the course access period ends.""" + old_image = event.detail['old_image'] + + # Post-migration: Remove the following lines + if 'email' not in old_image: + # If email is missing, use enrollment email + cur_image = enrollment_layer.get_item(KeyPair(old_image['id'], '0')) + old_image['name'] = cur_image['user']['name'] + old_image['email'] = cur_image['user']['email'] + old_image['course'] = cur_image['course']['name'] + + return send_email( + to=(old_image['name'], old_image['email']), + subject=SUBJECT, + message=MESSAGE, + context={ + 'course': old_image['course'], + }, + sender=EMAIL_SENDER, + sesv2_client=sesv2_client, + event={ + 'id': old_image['id'], + 'sk': 'SCHEDULE#REMINDER_NO_ACCESS_AFTER_3_DAYS', + }, + dynamodb_persistence_layer=enrollment_layer, + ) diff --git a/enrollments-events/app/events/reenroll_if_failed.py b/enrollments-events/app/events/reenroll_if_failed.py new file mode 100644 index 0000000..c3bb895 --- /dev/null +++ b/enrollments-events/app/events/reenroll_if_failed.py @@ -0,0 +1,35 @@ +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.dynamodb import DynamoDBPersistenceLayer, SortKey, TransactKey + +from boto3clients import dynamodb_client +from config import ( + ENROLLMENT_TABLE, +) + +logger = Logger(__name__) +enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client) + + +@event_source(data_class=EventBridgeEvent) +@logger.inject_lambda_context +def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool: + new_image = event.detail['new_image'] + subscription = enrollment_layer.collection.get_items( + TransactKey(new_image['id']) + + SortKey('METADATA#SUBSCRIPTION_COVERED') + + SortKey('author') + + SortKey('tenant') + # Post-migration: uncommet the following lines + # + SortKey('CREATED_BY') + # + SortKey('ORG') + ) + + with enrollment_layer.transact_writer() as transact_writer: + ... + + return True diff --git a/enrollments-events/app/events/schedule_reminders.py b/enrollments-events/app/events/schedule_reminders.py new file mode 100644 index 0000000..a3bad8c --- /dev/null +++ b/enrollments-events/app/events/schedule_reminders.py @@ -0,0 +1,70 @@ +from datetime import timedelta + +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, ttl +from layercake.dynamodb import DynamoDBPersistenceLayer + +from boto3clients import dynamodb_client +from config import ( + ENROLLMENT_TABLE, +) + +logger = Logger(__name__) +enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client) + + +@event_source(data_class=EventBridgeEvent) +@logger.inject_lambda_context +def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool: + new_image = event.detail['new_image'] + enrollment_id = new_image['id'] + user = new_image['user'] + course = new_image['course'] + now_ = now() + + with enrollment_layer.transact_writer() as transact: + transact.put( + item={ + 'id': enrollment_id, + 'sk': 'SCHEDULE#REMINDER_NO_ACCESS_AFTER_3_DAYS', + 'name': user.name, + 'email': user.email, + 'course': course.name, + 'created_at': now_, + 'ttl': ttl(days=3, start_dt=now_), + }, + ) + # By default, the enrollment will expire when the access period ends + # (scheduled below). + # If the enrollment is completed earlier (e.g., certificate issued), + # the expiration schedule is canceled and an archive schedule + # (`SCHEDULE#SET_AS_ARCHIVED`) is created instead. + transact.put( + item={ + 'id': enrollment_id, + 'sk': 'SCHEDULE#SET_AS_EXPIRED', + 'name': user.name, + 'email': user.email, + 'course': course.name, + 'created_at': now_, + 'ttl': ttl(start_dt=now_ + timedelta(days=course.access_period)), + }, + ) + transact.put( + item={ + 'id': enrollment_id, + 'sk': 'REMINDER_ACCESS_PERIOD_BEFORE_15_DAYS', + 'name': user.name, + 'email': user.email, + 'course': course.name, + 'created_at': now_, + 'ttl': ttl(start_dt=now_ + timedelta(days=course.access_period - 15)), + }, + ) + + return True diff --git a/enrollments-events/tests/events/emails/test_reminder_access_period_before_30_days.py b/enrollments-events/tests/events/emails/test_reminder_access_period_before_30_days.py new file mode 100644 index 0000000..8a83387 --- /dev/null +++ b/enrollments-events/tests/events/emails/test_reminder_access_period_before_30_days.py @@ -0,0 +1,18 @@ +import app.events.emails.reminder_access_period_before_30_days as app +from aws_lambda_powertools.utilities.typing import LambdaContext + + +def test_reminder_access_period_before_30_days( + dynamodb_seeds, + lambda_context: LambdaContext, +): + event = { + 'detail': { + 'old_image': { + 'id': '47ZxxcVBjvhDS5TE98tpfQ', + 'sk': 'SCHEDULE#REMINDER_ACCESS_PERIOD_BEFORE_30_DAYS', + } + } + } + + assert app.lambda_handler(event, lambda_context) # type: ignore diff --git a/enrollments-events/tests/events/emails/test_reminder_cert_expiration_before_30_days.py b/enrollments-events/tests/events/emails/test_reminder_cert_expiration_before_30_days.py new file mode 100644 index 0000000..0f2c986 --- /dev/null +++ b/enrollments-events/tests/events/emails/test_reminder_cert_expiration_before_30_days.py @@ -0,0 +1,18 @@ +import app.events.emails.reminder_cert_expiration_before_30_days as app +from aws_lambda_powertools.utilities.typing import LambdaContext + + +def test_reminder_cert_expiration_before_30_days( + dynamodb_seeds, + lambda_context: LambdaContext, +): + event = { + 'detail': { + 'old_image': { + 'id': '47ZxxcVBjvhDS5TE98tpfQ', + 'sk': 'SCHEDULE#REMINDER_CERT_EXPIRATION_BEFORE_30_DAYS', + } + } + } + + assert app.lambda_handler(event, lambda_context) # type: ignore diff --git a/enrollments-events/tests/events/emails/test_reminder_no_activity_after_7_days.py b/enrollments-events/tests/events/emails/test_reminder_no_activity_after_7_days.py new file mode 100644 index 0000000..98722ea --- /dev/null +++ b/enrollments-events/tests/events/emails/test_reminder_no_activity_after_7_days.py @@ -0,0 +1,18 @@ +import app.events.emails.reminder_no_activity_after_7_days as app +from aws_lambda_powertools.utilities.typing import LambdaContext + + +def test_reminder_no_activity_after_7_days( + dynamodb_seeds, + lambda_context: LambdaContext, +): + event = { + 'detail': { + 'old_image': { + 'id': '47ZxxcVBjvhDS5TE98tpfQ', + 'sk': 'SCHEDULE#REMINDER_NO_ACTIVITY_AFTER_7_DAYS', + } + } + } + + assert app.lambda_handler(event, lambda_context) # type: ignore