79 lines
1.9 KiB
Python
79 lines
1.9 KiB
Python
import pprint
|
|
from dataclasses import asdict, dataclass
|
|
from uuid import uuid4
|
|
|
|
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, KeyChain, KeyPair
|
|
|
|
from boto3clients import dynamodb_client
|
|
from config import ENROLLMENT_TABLE
|
|
|
|
logger = Logger(__name__)
|
|
dyn = 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']
|
|
order_id = new_image['id']
|
|
enrollments = dyn.collection.query(
|
|
key=KeyPair(order_id, 'ENROLLMENT#'),
|
|
).get('items', [])
|
|
|
|
if not enrollments:
|
|
items = dyn.collection.get_item(
|
|
KeyPair(order_id, 'ITEMS'),
|
|
raise_on_error=False,
|
|
default=[],
|
|
)
|
|
pprint.pp(items)
|
|
# docx = {
|
|
# 'id': f'SEAT#ORG#{org_id}',
|
|
# 'sk': f'ORDER#{order_id}#ENROLLMENT#{uuid4()}',
|
|
# 'course': {},
|
|
# 'created_at': now_,
|
|
# }
|
|
|
|
pprint.pp(enrollments)
|
|
|
|
return True
|
|
|
|
|
|
# Se houver matriculas
|
|
# -> com: scheduled_for
|
|
# -> tenta agendar, se não joga para vagas
|
|
# -> tenta matriculas, se falhar, joga para vagas
|
|
|
|
# se não houver vagas, gera as vagas.
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class Course:
|
|
id: str
|
|
name: str
|
|
access_period: int
|
|
|
|
|
|
def _get_courses(ids: set) -> tuple[Course, ...]:
|
|
pairs = tuple(KeyPair(idx, '0') for idx in ids)
|
|
r = dyn.collection.get_items(
|
|
KeyChain(pairs),
|
|
flatten_top=False,
|
|
)
|
|
courses = tuple(
|
|
Course(
|
|
id=idx,
|
|
name=obj['name'],
|
|
access_period=obj['access_period'],
|
|
)
|
|
for idx, obj in r.items()
|
|
)
|
|
|
|
return courses
|