add billing period

enrollments-events/app/enrollment.py

@@ -40,29 +40,41 @@ class LifecycleEvents(str, Enum):
     """Lifecycle events related to scheduling actions."""
 
     # Reminder if the user does not access within 3 days
-    # REMINDER_NO_ACCESS_3_DAYS = 'SCHEDULES#REMINDER_NO_ACCESS_3_DAYS'
+    # REMINDER_NO_ACCESS_AFTER_3_DAYS = 'SCHEDULE#REMINDER_NO_ACCESS_AFTER_3_DAYS'
     DOES_NOT_ACCESS = 'schedules#does_not_access'
 
     # When there is no activity 7 days after the first access
-    # NO_ACTIVITY_7_DAYS = 'SCHEDULES#NO_ACTIVITY_7_DAYS'
+    # REMINDER_NO_ACTIVITY_AFTER_7_DAYS = 'SCHEDULE#REMINDER_NO_ACTIVITY_AFTER_7_DAYS'
     NO_ACTIVITY = 'schedules#no_activity'
 
     # Reminder 30 days before the access period expires
-    # ACCESS_PERIOD_REMINDER_30_DAYS = 'SCHEDULES#ACCESS_PERIOD_REMINDER_30_DAYS'
+    # REMINDER_ACCESS_PERIOD_BEFORE_30_DAYS = 'SCHEDULE#REMINDER_ACCESS_PERIOD_BEFORE_30_DAYS'
     ACCESS_PERIOD_ENDS = 'schedules#access_period_ends'
 
     # Reminder for certificate expiration set to 30 days from now
-    CERT_EXP_REMINDER_30_DAYS = 'SCHEDULES#CERT_EXP_REMINDER_30_DAYS'
+    REMINDER_CERT_EXPIRATION_BEFORE_30_DAYS = (
+        'SCHEDULE#REMINDER_CERT_EXPIRATION_BEFORE_30_DAYS'
+    )
 
     # Archive the course after the certificate expires
-    # SET_AS_ARCHIVE = 'schedules#set_as_archive'
+    # SET_AS_ARCHIVE = 'SCHEDULE#SET_AS_ARCHIVE'
     ARCHIVE_IT = 'schedules#archive_it'
 
     # When the access period ends for a course without a certificate
-    # SET_AS_EXPIRE = 'schedules#set_as_expire'
+    # SET_AS_EXPIRE = 'SCHEDULE#SET_AS_EXPIRE'
     EXPIRATION = 'schedules#expiration'
 
 
+class DeduplicationConflictError(Exception):
+    def __init__(self, *args):
+        super().__init__('Enrollment already exists')
+
+
+class SlotDoesNotExistError(Exception):
+    def __init__(self, *args):
+        super().__init__('Slot does not exist')
+
+
 def enroll(
     enrollment: Enrollment,
     *,
@@ -112,8 +124,8 @@ def enroll(
             },
         )
         # Enrollment expires by default when the access period ends.
-        # When the course is finished, it is automatically removed,
+        # When the course is completed, it is automatically removed,
-        # and the `schedules#course_archived` event is created.
+        # and the `SCHEDULE#SET_AS_ARCHIVE` event is created.
         transact.put(
             item={
                 'id': enrollment.id,
@@ -146,9 +158,7 @@ def enroll(
             transact.put(
                 item={
                     'id': enrollment.id,
-                    # Post-migration: uncomment the following line
+                    'sk': f'LINKED_ENTITIES#{entity.type}',
-                    # 'sk': f'LINKED_ENTITIES#{entity.type}',
-                    'sk': f'linked_entities#{entity.type}',
                     'created_at': now_,
                     f'{keyprefix}_id': entity.id,
                 }
@@ -166,10 +176,6 @@ def enroll(
                 }
             )
 
-            class SlotDoesNotExistError(Exception):
-                def __init__(self, *args):
-                    super().__init__('Slot does not exist')
-
             transact.delete(
                 key=KeyPair(slot.id, slot.sk),
                 cond_expr='attribute_exists(sk)',
@@ -194,10 +200,6 @@ def enroll(
                 },
             )
 
-        class DeduplicationConflictError(Exception):
-            def __init__(self, *args):
-                super().__init__('Enrollment already exists')
-
         # Prevents the user from enrolling in the same course again until
         # the deduplication window expires or is removed.
         if deduplication_window:

enrollments-events/app/events/emails/reminder_access_period_before_15_days.py

@@ -1,12 +0,0 @@
-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
-
-logger = Logger(__name__)
-
-
-@event_source(data_classe=EventBridgeEvent)
-@logger.inject_lambda_context
-def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
-    new_image = event.detail['new_image']
-    return True

enrollments-events/app/events/emails/reminder_no_access_after_3_days.py

@@ -19,7 +19,7 @@ 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/>
+Você foi matriculado no curso de <b>{course}</b> há 3 dias, 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 para acessar seu curso:
@@ -39,10 +39,10 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     # Post-migration: Remove the following lines
     if 'email' not in old_image:
         # If email is missing, use enrollment email
-        data = enrollment_layer.get_item(KeyPair(old_image['id'], '0'))
+        cur_image = enrollment_layer.get_item(KeyPair(old_image['id'], '0'))
-        old_image['name'] = data['user']['name']
+        old_image['name'] = cur_image['user']['name']
-        old_image['email'] = data['user']['email']
+        old_image['email'] = cur_image['user']['email']
-        old_image['course'] = data['course']['name']
+        old_image['course'] = cur_image['course']['name']
 
     emailmsg = Message(
         from_=EMAIL_SENDER,
@@ -74,7 +74,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         enrollment_layer.put_item(
             item={
                 'id': old_image['id'],
-                'sk': 'SCHEDULES#REMINDER_NO_ACCESS_AFTER_3_DAYS#FAILED',
+                'sk': 'SCHEDULE#REMINDER_NO_ACCESS_AFTER_3_DAYS#FAILED',
                 # Post-migration: Uncomment the following line
                 # 'sk': f'{old_image["sk"]}#FAILED',
                 'created_at': now_,
@@ -86,7 +86,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         enrollment_layer.put_item(
             item={
                 'id': old_image['id'],
-                'sk': 'SCHEDULES#REMINDER_NO_ACCESS_AFTER_3_DAYS#EXECUTED',
+                'sk': 'SCHEDULE#REMINDER_NO_ACCESS_AFTER_3_DAYS#EXECUTED',
                 # Post-migration: Uncomment the following line
                 # 'sk': f'{old_image["sk"]}#EXECUTED',
                 'created_at': now_,

enrollments-events/app/events/emails/reminder_no_activity_after_7_days.py

@@ -1,12 +1,89 @@
 from aws_lambda_powertools import Logger
-from aws_lambda_powertools.utilities.data_classes import EventBridgeEvent, event_source
+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 DynamoDBPersistenceLayer, KeyPair
+from layercake.email_ import Message
+from layercake.strutils import first_word, truncate_str
+
+from boto3clients import dynamodb_client, sesv2_client
+from config import (
+    EMAIL_SENDER,
+    ENROLLMENT_TABLE,
+)
+
+SUBJECT = ''
+MESSAGE = """
+"""
 
 logger = Logger(__name__)
+enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
 
 
-@event_source(data_classe=EventBridgeEvent)
+@event_source(data_class=EventBridgeEvent)
 @logger.inject_lambda_context
 def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
-    new_image = event.detail['new_image']
+    old_image = event.detail['old_image']
-    return True
+    now_ = now()
+
+    # 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']
+
+    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(),
+                },
+            }
+        )
+        logger.info('Email sent')
+    except Exception as exc:
+        logger.exception(exc)
+
+        enrollment_layer.put_item(
+            item={
+                'id': old_image['id'],
+                'sk': 'SCHEDULE#REMINDER_NO_ACTIVITY_AFTER_7_DAYS#FAILED',
+                # Post-migration: Uncomment the following line
+                # 'sk': f'{old_image["sk"]}#FAILED',
+                'created_at': now_,
+            }
+        )
+
+        return False
+    else:
+        enrollment_layer.put_item(
+            item={
+                'id': old_image['id'],
+                'sk': 'SCHEDULE#REMINDER_NO_ACTIVITY_AFTER_7_DAYS#EXECUTED',
+                # Post-migration: Uncomment the following line
+                # 'sk': f'{old_image["sk"]}#EXECUTED',
+                'created_at': now_,
+            }
+        )
+
+        return True

enrollments-events/app/events/enroll.py

@@ -30,7 +30,7 @@ processor = BatchProcessor()
 
 @event_source(data_class=EventBridgeEvent)
 @logger.inject_lambda_context
-def lambda_handler(event: EventBridgeEvent, context: LambdaContext):
+def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
     order = order_layer.collection.get_items(
         TransactKey(new_image['id'])

enrollments-events/app/events/stopgap/set_subscription_covered.py

@@ -23,13 +23,13 @@ enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
 def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
     now_ = now()
-    data = user_layer.get_item(
+    terms = user_layer.get_item(
         # Post-migration: uncomment the following line
         # KeyPair(new_image['org_id'], 'METADATA#BILLING_TERMS'),
         KeyPair(new_image['tenant_id'], 'metadata#billing_policy'),
     )
 
-    if not data:
+    if not terms:
         return False
 
     try:
@@ -49,8 +49,8 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                     'id': new_image['id'],
                     'sk': 'METADATA#SUBSCRIPTION_COVERED',
                     'org_id': new_image['tenant_id'],
-                    'billing_day': data['billing_day'],
+                    'billing_day': terms['billing_day'],
-                    'created_at': now(),
+                    'created_at': now_,
                 },
                 cond_expr='attribute_not_exists(sk)',
             )

enrollments-events/template.yaml

@@ -23,7 +23,7 @@ Globals:
     Architectures:
       - x86_64
     Layers:
-      - !Sub arn:aws:lambda:sa-east-1:336641857101:layer:layercake:86
+      - !Sub arn:aws:lambda:sa-east-1:336641857101:layer:layercake:94
     Environment:
       Variables:
         TZ: America/Sao_Paulo
@@ -134,7 +134,7 @@ Resources:
               detail:
                 new_image:
                   # Post-migration: uncomment the following lines
-                  # sk: [slots]
+                  # sk: [SLOT]
                   # mode: [STANDALONE]
                   sk: [generated_items]
                   scope: [SINGLE_USER]
@@ -163,12 +163,25 @@ Resources:
               detail:
                 new_image:
                   # Post-migration: uncomment the following lines
-                  # sk: [slots]
+                  # sk: [SLOT]
                   # mode: [BATCH]
                   sk: [generated_items]
                   scope: [MULTI_USER]
                   status: [PENDING]
 
+  SesPolicy:
+    Type: AWS::IAM::ManagedPolicy
+    Properties:
+      PolicyDocument:
+        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
+
   EventReminderNoAccessAfter3DaysFunction:
     Type: AWS::Serverless::Function
     Properties:
@@ -176,16 +189,9 @@ Resources:
       LoggingConfig:
         LogGroup: !Ref EventLog
       Policies:
+        - !Ref SesPolicy
         - 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
@@ -196,11 +202,55 @@ Resources:
               detail:
                 keys:
                   sk:
-                    - SCHEDULES#REMINDER_NO_ACCESS_AFTER_3_DAYS
+                    - SCHEDULE#REMINDER_NO_ACCESS_AFTER_3_DAYS
                     # Post-migration: remove the following lines
+                    - SCHEDULES#REMINDER_NO_ACCESS_AFTER_3_DAYS
                     - schedules#does_not_access
                     - schedules#reminder_no_access_3_days
 
+  EventReminderNoActivityAfter7DaysFunction:
+    Type: AWS::Serverless::Function
+    Properties:
+      Handler: events.emails.reminder_no_activity_after_7_days.lambda_handler
+      LoggingConfig:
+        LogGroup: !Ref EventLog
+      Policies:
+        - !Ref SesPolicy
+        - DynamoDBCrudPolicy:
+            TableName: !Ref EnrollmentTable
+      Events:
+        DynamoDBEvent:
+          Type: EventBridgeRule
+          Properties:
+            Pattern:
+              resources: [!Ref EnrollmentTable]
+              detail-type: [EXPIRE]
+              detail:
+                keys:
+                  sk:
+                    - SCHEDULE#REMINDER_NO_ACTIVITY_AFTER_7_DAYS
+
+  EventScheduleRemindersFunction:
+    Type: AWS::Serverless::Function
+    Properties:
+      Handler: events.schedule_reminders.lambda_handler
+      LoggingConfig:
+        LogGroup: !Ref EventLog
+      Policies:
+        - DynamoDBCrudPolicy:
+            TableName: !Ref EnrollmentTable
+      Events:
+        DynamoDBEvent:
+          Type: EventBridgeRule
+          Properties:
+            Pattern:
+              resources: [!Ref EnrollmentTable]
+              detail-type: [INSERT]
+              detail:
+                new_image:
+                  sk: ["0"]
+                  status: [PENDING]
+
   EventIssueCertFunction:
     Type: AWS::Serverless::Function
     Properties:
@@ -219,4 +269,4 @@ Resources:
                 new_image:
                   status: [COMPLETED]
                 old_image:
-                  status: [PENDING]
+                  status: [IN_PROGRESS]

enrollments-events/uv.lock

@@ -29,6 +29,18 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/77/06/bb80f5f86020c4551da315d78b3ab75e8228f89f0162f2c3a819e407941a/attrs-25.3.0-py3-none-any.whl", hash = "sha256:427318ce031701fea540783410126f03899a97ffc6f61596ad581ac2e40e3bc3", size = 63815, upload-time = "2025-03-13T11:10:21.14Z" },
 ]
 
+[[package]]
+name = "authlib"
+version = "1.6.1"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "cryptography" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/8e/a1/d8d1c6f8bc922c0b87ae0d933a8ed57be1bef6970894ed79c2852a153cd3/authlib-1.6.1.tar.gz", hash = "sha256:4dffdbb1460ba6ec8c17981a4c67af7d8af131231b5a36a88a1e8c80c111cdfd", size = 159988, upload-time = "2025-07-20T07:38:42.834Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/f9/58/cc6a08053f822f98f334d38a27687b69c6655fb05cd74a7a5e70a2aeed95/authlib-1.6.1-py2.py3-none-any.whl", hash = "sha256:e9d2031c34c6309373ab845afc24168fe9e93dc52d252631f52642f21f5ed06e", size = 239299, upload-time = "2025-07-20T07:38:39.259Z" },
+]
+
 [[package]]
 name = "aws-encryption-sdk"
 version = "4.0.1"
@@ -46,15 +58,15 @@ wheels = [
 
 [[package]]
 name = "aws-lambda-powertools"
-version = "3.13.0"
+version = "3.19.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "jmespath" },
     { name = "typing-extensions" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/fd/2b/068efd467c0866e2272c5de7525ddb02ff4e694f71245c8d2a83d4948f23/aws_lambda_powertools-3.13.0.tar.gz", hash = "sha256:99dc11ac6eb81564f599fdd85ba79069f7740ae3481c99bca2cee8abb7c95543", size = 672664, upload-time = "2025-05-20T07:35:30.254Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/58/db/eb2708f7c27ab02b8d85936ce9308538e1e22c8c8224be5f00da3e6f44f7/aws_lambda_powertools-3.19.0.tar.gz", hash = "sha256:8897ba4be0b3a51f2b8f68946d650f3ef574fa2c40395544de03bd0c61979999", size = 689768, upload-time = "2025-08-12T08:45:46.887Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/98/cd/2241ff877528c66ee11ea636684c4242ceeadb6459a33b08507a40151414/aws_lambda_powertools-3.13.0-py3-none-any.whl", hash = "sha256:9df045f4c3ff944176655813dbff8c1160e056babf5e6d71d4e18c0003818f2e", size = 802546, upload-time = "2025-05-20T07:35:27.767Z" },
+    { url = "https://files.pythonhosted.org/packages/c6/52/5a73194286af329309263e9c4e2a57b8feac63bb6027be8d2d6222cd4da7/aws_lambda_powertools-3.19.0-py3-none-any.whl", hash = "sha256:98f18d35f843cd46b80ccadcf39eefc0c489325bea116383bd93048a5241d9fc", size = 832645, upload-time = "2025-08-12T08:45:44.982Z" },
 ]
 
 [package.optional-dependencies]
@@ -439,19 +451,22 @@ wheels = [
 
 [[package]]
 name = "layercake"
-version = "0.8.2"
+version = "0.9.12"
 source = { directory = "../layercake" }
 dependencies = [
     { name = "arnparse" },
+    { name = "authlib" },
     { name = "aws-lambda-powertools", extra = ["all"] },
     { name = "dictdiffer" },
     { name = "ftfy" },
     { name = "glom" },
     { name = "meilisearch" },
     { name = "orjson" },
+    { name = "passlib" },
     { name = "pycpfcnpj" },
     { name = "pydantic", extra = ["email"] },
     { name = "pydantic-extra-types" },
+    { name = "pyjwt" },
     { name = "pytz" },
     { name = "requests" },
     { name = "smart-open", extra = ["s3"] },
@@ -462,15 +477,18 @@ dependencies = [
 [package.metadata]
 requires-dist = [
     { name = "arnparse", specifier = ">=0.0.2" },
-    { name = "aws-lambda-powertools", extras = ["all"], specifier = ">=3.8.0" },
+    { name = "authlib", specifier = ">=1.6.1" },
+    { name = "aws-lambda-powertools", extras = ["all"], specifier = ">=3.18.0" },
     { name = "dictdiffer", specifier = ">=0.9.0" },
     { name = "ftfy", specifier = ">=6.3.1" },
     { name = "glom", specifier = ">=24.11.0" },
     { name = "meilisearch", specifier = ">=0.34.0" },
     { name = "orjson", specifier = ">=3.10.15" },
+    { name = "passlib", specifier = ">=1.7.4" },
     { name = "pycpfcnpj", specifier = ">=1.8" },
     { name = "pydantic", extras = ["email"], specifier = ">=2.10.6" },
     { name = "pydantic-extra-types", specifier = ">=2.10.3" },
+    { name = "pyjwt", specifier = ">=2.10.1" },
     { name = "pytz", specifier = ">=2025.1" },
     { name = "requests", specifier = ">=2.32.3" },
     { name = "smart-open", extras = ["s3"], specifier = ">=7.1.0" },
@@ -534,6 +552,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/20/12/38679034af332785aac8774540895e234f4d07f7545804097de4b666afd8/packaging-25.0-py3-none-any.whl", hash = "sha256:29572ef2b1f17581046b3a2227d5c611fb25ec70ca1ba8554b24b0e69331a484", size = 66469, upload-time = "2025-04-19T11:48:57.875Z" },
 ]
 
+[[package]]
+name = "passlib"
+version = "1.7.4"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/b6/06/9da9ee59a67fae7761aab3ccc84fa4f3f33f125b370f1ccdb915bf967c11/passlib-1.7.4.tar.gz", hash = "sha256:defd50f72b65c5402ab2c573830a6978e5f202ad0d984793c8dde2c4152ebe04", size = 689844, upload-time = "2020-10-08T19:00:52.121Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/3b/a4/ab6b7589382ca3df236e03faa71deac88cae040af60c071a78d254a62172/passlib-1.7.4-py2.py3-none-any.whl", hash = "sha256:aa6bca462b8d8bda89c70b382f0c298a20b5560af6cbfa2dce410c0a2fb669f1", size = 525554, upload-time = "2020-10-08T19:00:49.856Z" },
+]
+
 [[package]]
 name = "pluggy"
 version = "1.6.0"
@@ -645,6 +672,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/b6/5f/d6d641b490fd3ec2c4c13b4244d68deea3a1b970a97be64f34fb5504ff72/pydantic_settings-2.9.1-py3-none-any.whl", hash = "sha256:59b4f431b1defb26fe620c71a7d3968a710d719f5f4cdbbdb7926edeb770f6ef", size = 44356, upload-time = "2025-04-18T16:44:46.617Z" },
 ]
 
+[[package]]
+name = "pyjwt"
+version = "2.10.1"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/e7/46/bd74733ff231675599650d3e47f361794b22ef3e3770998dda30d3b63726/pyjwt-2.10.1.tar.gz", hash = "sha256:3cc5772eb20009233caf06e9d8a0577824723b44e6648ee0a2aedb6cf9381953", size = 87785, upload-time = "2024-11-28T03:43:29.933Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/61/ad/689f02752eeec26aed679477e80e632ef1b682313be70793d798c1d5fc8f/PyJWT-2.10.1-py3-none-any.whl", hash = "sha256:dcdd193e30abefd5debf142f9adfcdd2b58004e644f25406ffaebd50bd98dacb", size = 22997, upload-time = "2024-11-28T03:43:27.893Z" },
+]
+
 [[package]]
 name = "pytest"
 version = "8.3.5"

order-events/app/events/billing/append_enrollment.py

@@ -112,31 +112,46 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
             ),
             flatten_top=False,
         )
-        order_layer.put_item(
+        with order_layer.transact_writer() as transact:
-            item={
+            transact.put(
-                'id': pk,
+                item={
-                'sk': f'{sk}#ENROLLMENT#{enrollment["id"]}',
+                    'id': pk,
-                'user': pick(('id', 'name'), enrollment['user']),
+                    'sk': f'{sk}#ENROLLMENT#{enrollment["id"]}',
-                'course': pick(('id', 'name'), enrollment['course']),
+                    'user': pick(('id', 'name'), enrollment['user']),
-                'unit_price': course['unit_price'],
+                    'course': pick(('id', 'name'), enrollment['course']),
-                # Post-migration: uncomment the following line
+                    'unit_price': course['unit_price'],
-                # 'enrolled_at': enrollment['created_at'],
+                    # Post-migration: uncomment the following line
-                'enrolled_at': enrollment['create_date'],
+                    # 'enrolled_at': enrollment['created_at'],
-                'created_at': now_,
+                    'enrolled_at': enrollment['create_date'],
-            }
+                    'created_at': now_,
-            # Add author if present
-            | (
-                {
-                    'author': {
-                        'id': author['user_id'],
-                        'name': author['name'],
-                    }
                 }
-                if author
+                # Add author if present
-                else {}
+                | (
-            ),
+                    {
-            cond_expr='attribute_not_exists(sk)',
+                        'author': {
-        )
+                            'id': author['user_id'],
+                            'name': author['name'],
+                        }
+                    }
+                    if author
+                    else {}
+                ),
+                cond_expr='attribute_not_exists(sk)',
+            )
+            transact.update(
+                key=KeyPair(
+                    pk=new_image['id'],
+                    sk=new_image['sk'],
+                ),
+                table_name=ENROLLMENT_TABLE,
+                update_expr='SET billing_period = :billing_period, \
+                                updated_at = :updated_at',
+                expr_attr_values={
+                    ':billing_period': sk,
+                    ':updated_at': now_,
+                },
+                cond_expr='attribute_exists(sk)',
+            )
     except Exception as exc:
         logger.exception(
             exc,

order-events/app/events/billing/send_email_on_closing.py

@@ -38,7 +38,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
     # Key pattern `BILLING#ORG#{org_id}`
     *_, org_id = new_image['id'].split('#')
-    # Key pattern `START#{start_date}#END#{end_date}#SCHEDULE#AUTO_CLOSE`
+    # Key pattern `START#{start_date}#END#{end_date}
     _, start_date, _, end_date, *_ = new_image['sk'].split('#')
 
     emailmsg = Message(

order-events/app/events/remove_slots_if_canceled.py

@@ -23,13 +23,15 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
     order_id = new_image['id']
     org_id = new_image['tenant_id']
+    # Post-migration: Uncomment the following line
+    # org_id = new_image['org_id']
 
     result = enrollment_layer.collection.query(
         KeyPair(
             # Post-migration: Uncomment the following line
             # f'SLOT#ORG#{org_id}',
-            f'vacancies#{org_id}',
+            pk=f'vacancies#{org_id}',
-            order_id,
+            sk=order_id,
         ),
         limit=100,
     )

order-events/template.yaml

@@ -46,7 +46,7 @@ Resources:
     Properties:
       RetentionInDays: 90
 
-  EventBillingAddEnrollmentFunction:
+  EventBillingAppendEnrollmentFunction:
     Type: AWS::Serverless::Function
     Properties:
       Handler: events.billing.append_enrollment.lambda_handler
@@ -55,7 +55,7 @@ Resources:
       Policies:
         - DynamoDBCrudPolicy:
             TableName: !Ref OrderTable
-        - DynamoDBReadPolicy:
+        - DynamoDBCrudPolicy:
             TableName: !Ref EnrollmentTable
         - DynamoDBReadPolicy:
             TableName: !Ref CourseTable
@@ -69,6 +69,8 @@ Resources:
               detail:
                 new_image:
                   sk: ["METADATA#SUBSCRIPTION_COVERED"]
+                  billing_period:
+                    - exists: false
 
   EventBillingCancelEnrollmentFunction:
     Type: AWS::Serverless::Function
@@ -91,10 +93,10 @@ Resources:
               detail:
                 new_image:
                   sk: ["0"]
-                  status: ["CANCELED"]
+                  status: [CANCELED]
                   subscription_covered: [true]
                 old_image:
-                  status: ["PENDING"]
+                  status: [PENDING]
 
   EventBillingCloseWindowFunction:
     Type: AWS::Serverless::Function
@@ -117,6 +119,8 @@ Resources:
               detail-type: [EXPIRE]
               detail:
                 keys:
+                  id:
+                    - prefix: BILLING
                   sk:
                     - suffix: SCHEDULE#AUTO_CLOSE
 
@@ -149,9 +153,11 @@ Resources:
               detail-type: [MODIFY]
               detail:
                 new_image:
-                  status: [CLOSED]
+                  id:
+                    - prefix: BILLING
                   s3_uri:
                     - exists: true
+                  status: [CLOSED]
                 old_image:
                   status: [PENDING]
 

order-events/tests/events/billing/test_append_enrollment.py

@@ -9,10 +9,11 @@ def test_append_enrollment(
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
     lambda_context: LambdaContext,
 ):
+    enrollment_id = '945e8672-1d72-45c6-b76c-ac06aa8b52ab'
     event = {
         'detail': {
             'new_image': {
-                'id': '945e8672-1d72-45c6-b76c-ac06aa8b52ab',
+                'id': enrollment_id,
                 'sk': 'METADATA#SUBSCRIPTION_COVERED',
                 'billing_day': 6,
                 'created_at': '2025-07-23T18:09:22.785678-03:00',
@@ -34,3 +35,9 @@ def test_append_enrollment(
         == 'START#2025-05-06#END#2025-06-05#ENROLLMENT#945e8672-1d72-45c6-b76c-ac06aa8b52ab'
     )
     assert items[2]['sk'] == 'START#2025-05-06#END#2025-06-05'
+
+    print(
+        dynamodb_persistence_layer.collection.get_item(
+            KeyPair(enrollment_id, 'METADATA#SUBSCRIPTION_COVERED')
+        )
+    )

order-events/tests/seeds.jsonl

@@ -24,6 +24,7 @@
 // Enrollments
 {"id": "945e8672-1d72-45c6-b76c-ac06aa8b52ab", "sk": "0", "course": {"id": "123", "name": "pytest"}, "user": {"id": "5OxmMjL-ujoR5IMGegQz", "name": "Sérgio R Siqueira"}, "create_date": "2025-06-05T12:13:54.371416+00:00"}
 {"id": "945e8672-1d72-45c6-b76c-ac06aa8b52ab", "sk": "author", "name": "Carolina Brand", "user_id": "SMEXYk5MQkKCzknJpxqr8n"}
+{"id": "945e8672-1d72-45c6-b76c-ac06aa8b52ab", "sk": "METADATA#SUBSCRIPTION_COVERED", "billing_day": 6, "org_id": "cJtK9SsnJhKPyxESe7g3DG", "created_at": "2025-07-23T18:09:22.785678-03:00"}
 
 {"id": "77055ad7-03e1-4b07-98dc-a2f1a90913ba", "sk": "0", "course": {"id": "123", "name": "pytest"}, "user": {"id": "5OxmMjL-ujoR5IMGegQz", "name": "Sérgio R Siqueira"}, "create_date": "2025-06-05T12:13:54.371416+00:00"}
 {"id": "77055ad7-03e1-4b07-98dc-a2f1a90913ba", "sk": "METADATA#SUBSCRIPTION_COVERED", "billing_day": 6, "org_id": "cJtK9SsnJhKPyxESe7g3DG"}