add user id to order

order-events/app/events/assign_tenant_cnpj.py

@@ -1,69 +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
-from layercake.dateutils import now
-from layercake.dynamodb import (
-    DynamoDBPersistenceLayer,
-    KeyPair,
-    SortKey,
-)
-
-from boto3clients import dynamodb_client
-from config import ORDER_TABLE, USER_TABLE
-
-logger = Logger(__name__)
-user_layer = DynamoDBPersistenceLayer(USER_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:
-    new_image = event.detail['new_image']
-    now_ = now()
-    ids = user_layer.collection.get_items(
-        KeyPair(
-            pk='cnpj',
-            sk=SortKey(new_image['cnpj'], path_spec='user_id'),
-            rename_key='org_id',
-        )
-        + KeyPair(
-            pk='email',
-            sk=SortKey(new_image['email'], path_spec='user_id'),
-            rename_key='user_id',
-        ),
-        flatten_top=False,
-    )
-
-    # Sometimes the function executes before the user insertion completes,
-    # so an exception is raised to trigger a retry.
-    if len(ids) < 2:
-        raise ValueError('IDs not found.')
-
-    logger.info('IDs found', ids=ids)
-
-    with order_layer.transact_writer() as transact:
-        transact.update(
-            key=KeyPair(new_image['id'], '0'),
-            update_expr='SET tenant_id = :tenant_id, updated_at = :updated_at',
-            expr_attr_values={
-                ':tenant_id': ids['org_id'],
-                ':updated_at': now_,
-            },
-        )
-
-        transact.update(
-            key=KeyPair(new_image['id'], 'author'),
-            update_expr='SET user_id = :user_id, updated_at = :updated_at',
-            expr_attr_values={
-                ':user_id': ids['user_id'],
-                ':updated_at': now_,
-            },
-        )
-
-        logger.info('IDs updated')
-
-    return True

order-events/app/events/remove_slots_if_canceled.py

@@ -17,23 +17,18 @@ enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
 order_layer = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
 
 
-class TenantDoesNotExistError(Exception):
-    def __init__(self, *args):
-        super().__init__('Tenant does not exist')
-
-
 @event_source(data_class=EventBridgeEvent)
 @logger.inject_lambda_context
 def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
     order_id = new_image['id']
-    tenant_id = new_image['tenant']
+    org_id = new_image['tenant_id']
 
     result = enrollment_layer.collection.query(
         KeyPair(
             # Post-migration: Uncomment the following line
-            # f'slots#org#{tenant_id}',
-            f'vacancies#{tenant_id}',
+            # f'SLOT#ORG#{org_id}',
+            f'vacancies#{org_id}',
             order_id,
         )
     )
@@ -45,8 +40,8 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
             batch.delete_item(
                 Key={
                     # Post-migration: Uncomment the following line
-                    # 'id': {'S': f'slots#org#{tenant_id}'},
-                    'id': {'S': f'vacancies#{tenant_id}'},
+                    # 'id': {'S': f'SLOT#ORG#{org_id}'},
+                    'id': {'S': f'vacancies#{org_id}'},
                     'sk': {'S': pair['sk']},
                 }
             )