add support to save executed scheduled

This commit is contained in:
2025-12-12 11:09:56 -03:00
parent 3954b148fe
commit 3895e953af
2 changed files with 61 additions and 22 deletions

View File

@@ -1,7 +1,7 @@
from datetime import date, datetime, time, timedelta
from decimal import Decimal
from http import HTTPStatus
from typing import Annotated
from typing import Annotated, TypedDict
from uuid import uuid4
import pytz
@@ -18,7 +18,6 @@ from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
from layercake.extra_types import CpfStr, NameStr
from layercake.strutils import md5_hash
from pydantic import UUID4, BaseModel, EmailStr, Field, FutureDate
from typing_extensions import TypedDict
from boto3clients import dynamodb_client
from config import DEDUP_WINDOW_OFFSET_DAYS, ENROLLMENT_TABLE, TZ, USER_TABLE
@@ -276,10 +275,13 @@ def enroll_later(enrollment: Enrollment, context: Context):
lock_hash = md5_hash(f'{user.id}{course.id}')
with dyn.transact_writer() as transact:
pk = f'SCHEDULED#ORG#{org.id}'
sk = f'{scheduled_for.isoformat()}#{lock_hash}'
transact.put(
item={
'id': f'SCHEDULED#ORG#{org.id}',
'sk': f'{scheduled_for.isoformat()}#{lock_hash}',
'id': pk,
'sk': sk,
'user': user.model_dump(),
'course': course.model_dump(),
'org_name': org.name,
@@ -301,6 +303,10 @@ def enroll_later(enrollment: Enrollment, context: Context):
item={
'id': 'LOCK#SCHEDULED',
'sk': lock_hash,
'scheduled': {
'id': pk,
'sk': sk,
},
'created_at': now_,
},
cond_expr='attribute_not_exists(sk)',