cancel enrollment on billing

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

@@ -41,24 +41,23 @@ sqlite3.register_converter('json', json.loads)
 def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
     now_ = now()
-    enrollment_id = new_image['id']
     org_id = new_image['org_id']
-    data = enrollment_layer.collection.get_items(
-        TransactKey(enrollment_id) + SortKey('0') + SortKey('author')
+    enrollment = enrollment_layer.collection.get_items(
+        TransactKey(new_image['id']) + SortKey('0') + SortKey('author')
     )
 
-    if not data:
+    if not enrollment:
         logger.debug('Enrollment not found')
         return False
 
-    logger.info('Enrollment found', data=data)
+    logger.info('Enrollment found', data=enrollment)
 
     # Keep it until the migration has been completed
-    old_course = _get_course(data['course']['id'])
+    old_course = _get_course(enrollment['course']['id'])
     if old_course:
-        data['course'] = old_course
+        enrollment['course'] = old_course
 
-    created_at: datetime = fromisoformat(data['create_date'])  # type: ignore
+    created_at: datetime = fromisoformat(enrollment['create_date'])  # type: ignore
     start_date, end_date = get_billing_period(
         billing_day=new_image['billing_day'],
         date_=created_at,
@@ -69,7 +68,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         end=end_date.isoformat(),
     )
 
-    logger.info('Enrollment found', data=data)
+    logger.info('Enrollment found', data=enrollment)
 
     try:
         with order_layer.transact_writer() as transact:
@@ -81,7 +80,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                     'created_at': now_,
                 },
                 cond_expr='attribute_not_exists(sk)',
-                exc_cls=ExistingBillingConflictError,
+                exc_cls=BillingConflictError,
             )
             transact.put(
                 item={
@@ -93,12 +92,13 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                     'created_at': now_,
                 }
             )
-    except ExistingBillingConflictError:
+    except BillingConflictError:
         pass
 
+    # Add enrollment entry to billing
     try:
-        author = data.get('author')
-        course_id = data['course']['id']
+        author = enrollment.get('author')
+        course_id = enrollment['course']['id']
         course = course_layer.collection.get_items(
             KeyPair(
                 pk=course_id,
@@ -115,15 +115,16 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         order_layer.put_item(
             item={
                 'id': pk,
-                'sk': f'{sk}#ENROLLMENT#{enrollment_id}',
-                'user': pick(('id', 'name'), data['user']),
-                'course': pick(('id', 'name'), data['course']),
+                'sk': f'{sk}#ENROLLMENT#{enrollment["id"]}',
+                'user': pick(('id', 'name'), enrollment['user']),
+                'course': pick(('id', 'name'), enrollment['course']),
                 'unit_price': course['unit_price'],
                 # Post-migration: uncomment the following line
-                # 'enrolled_at': data['created_at'],
-                'enrolled_at': data['create_date'],
+                # 'enrolled_at': enrollment['created_at'],
+                'enrolled_at': enrollment['create_date'],
                 'created_at': now_,
             }
+            # Add author if present
             | (
                 {
                     'author': {
@@ -146,7 +147,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         return True
 
 
-class ExistingBillingConflictError(Exception): ...
+class BillingConflictError(Exception): ...
 
 
 class BillingNotFoundError(Exception): ...

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

@@ -0,0 +1,84 @@
+from datetime import datetime
+
+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 fromisoformat, now
+from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, SortKey, TransactKey
+from layercake.funcs import pick
+
+from boto3clients import dynamodb_client
+from config import ENROLLMENT_TABLE, ORDER_TABLE
+from utils import get_billing_period
+
+logger = Logger(__name__)
+enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
+order_layer = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
+
+
+@event_source(data_class=EventBridgeEvent)
+@logger.inject_lambda_context
+def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
+    now_ = now()
+    new_image = event.detail['new_image']
+    enrollment_id = new_image['id']
+    subscription = enrollment_layer.collection.get_items(
+        TransactKey(enrollment_id)
+        + SortKey('METADATA#SUBSCRIPTION_COVERED')
+        # Post-migration: uncomment the following line
+        # + SortKey('CANCELED', path_spec='author', rename_key='author')
+        + SortKey('canceled', path_spec='author', rename_key='author')
+    )
+
+    created_at: datetime = fromisoformat(new_image['create_date'])  # type: ignore
+    start_date, end_date = get_billing_period(
+        billing_day=int(subscription['billing_day']),
+        date_=created_at,
+    )
+    pk = 'BILLING#ORG#{org_id}'.format(org_id=subscription['org_id'])
+    sk = 'START#{start}#END#{end}'.format(
+        start=start_date.isoformat(),
+        end=end_date.isoformat(),
+    )
+
+    if now_.date() > end_date:
+        logger.debug('Enrollment outside the billing period')
+        return False
+
+    try:
+        author = subscription.get('author')
+        # Retrieve canceled enrollment data
+        old_enrollment = order_layer.collection.get_item(
+            KeyPair(
+                pk=pk,
+                sk=f'{sk}#ENROLLMENT#{enrollment_id}',
+            ),
+            exc_cls=EnrollmentNotFoundError,
+        )
+
+        order_layer.put_item(
+            item={
+                'id': pk,
+                'sk': f'{sk}#ENROLLMENT#{enrollment_id}#CANCELED',
+                'unit_price': old_enrollment['unit_price'] * -1,
+                'created_at': now_,
+            }
+            | pick(('user', 'course', 'enrolled_at'), old_enrollment)
+            # Add author if present
+            | ({'author': author} if author else {}),
+            cond_expr='attribute_not_exists(sk)',
+        )
+    except Exception as exc:
+        logger.exception(
+            exc,
+            keypair={'pk': pk, 'sk': sk},
+        )
+        return False
+    else:
+        return True
+
+
+class EnrollmentNotFoundError(Exception): ...

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

@@ -26,7 +26,6 @@ Em anexo você encontra o relatório das matrículas realizadas no período de
 <strong>{start_date}</strong> a <strong>{end_date}</strong>.<br/><br/>
 
 Qualquer dúvida, estamos à disposição.
-
 """
 
 logger = Logger(__name__)