update events
This commit is contained in:
@@ -25,7 +25,8 @@ def update_progress(
|
||||
# Update progress only if the enrollment status is `IN_PROGRESS`
|
||||
transact.update(
|
||||
key=KeyPair(id, '0'),
|
||||
update_expr='SET progress = :progress, updated_at = :updated_at',
|
||||
update_expr='SET progress = :progress, \
|
||||
updated_at = :now',
|
||||
cond_expr='#status = :in_progress',
|
||||
expr_attr_names={
|
||||
'#status': 'status',
|
||||
@@ -33,7 +34,7 @@ def update_progress(
|
||||
expr_attr_values={
|
||||
':in_progress': 'IN_PROGRESS',
|
||||
':progress': progress,
|
||||
':updated_at': now_,
|
||||
':now': now_,
|
||||
},
|
||||
exc_cls=EnrollmentConflictError,
|
||||
)
|
||||
@@ -49,11 +50,12 @@ def update_progress(
|
||||
except EnrollmentConflictError:
|
||||
with dynamodb_persistence_layer.transact_writer() as transact:
|
||||
# If the enrollment status is `PENDING`, set it to `IN_PROGRESS`
|
||||
# and update progress and updated_at date
|
||||
# and update `progress` and `updated_at`
|
||||
transact.update(
|
||||
key=KeyPair(id, '0'),
|
||||
update_expr='SET progress = :progress, #status = :in_progress, \
|
||||
updated_at = :updated_at',
|
||||
update_expr='SET progress = :progress, \
|
||||
#status = :in_progress, \
|
||||
updated_at = :now',
|
||||
cond_expr='#status = :pending',
|
||||
expr_attr_names={
|
||||
'#status': 'status',
|
||||
@@ -62,7 +64,7 @@ def update_progress(
|
||||
':in_progress': 'IN_PROGRESS',
|
||||
':pending': 'PENDING',
|
||||
':progress': progress,
|
||||
':updated_at': now_,
|
||||
':now': now_,
|
||||
},
|
||||
exc_cls=EnrollmentConflictError,
|
||||
)
|
||||
@@ -189,15 +191,15 @@ def _set_status_as_completed(
|
||||
update_expr='SET #status = :completed, \
|
||||
progress = :progress, \
|
||||
score = :score, \
|
||||
updated_at = :updated_at',
|
||||
updated_at = :now',
|
||||
cond_expr='#status = :in_progress',
|
||||
expr_attr_names={'#status': 'status'},
|
||||
expr_attr_values={
|
||||
':completed': 'COMPLETED',
|
||||
':in_progress': 'IN_PROGRESS',
|
||||
':score': score,
|
||||
':updated_at': now_,
|
||||
':progress': progress,
|
||||
':now': now_,
|
||||
},
|
||||
exc_cls=EnrollmentConflictError,
|
||||
)
|
||||
@@ -214,7 +216,7 @@ def _set_status_as_completed(
|
||||
transact.put(
|
||||
item={
|
||||
'id': id,
|
||||
'sk': 'SCHEDULE#REMINDER_CERT_EXPIRATION_BEFORE_30_DAYS',
|
||||
'sk': 'SCHEDULE#SET_CERT_EXPIRED',
|
||||
'ttl': cert_exp_ttl,
|
||||
'created_at': now_,
|
||||
}
|
||||
@@ -222,7 +224,7 @@ def _set_status_as_completed(
|
||||
transact.put(
|
||||
item={
|
||||
'id': id,
|
||||
'sk': 'SCHEDULE#SET_CERT_EXPIRED',
|
||||
'sk': 'SCHEDULE#REMINDER_CERT_EXPIRATION_BEFORE_30_DAYS',
|
||||
'ttl': cert_exp_reminder_ttl,
|
||||
'created_at': now_,
|
||||
}
|
||||
@@ -288,8 +290,8 @@ def _set_status_as_failed(
|
||||
update_expr='SET #status = :failed, \
|
||||
progress = :progress, \
|
||||
score = :score, \
|
||||
access_expired = :access_expired, \
|
||||
updated_at = :updated_at',
|
||||
access_expired = :true, \
|
||||
updated_at = :now',
|
||||
cond_expr='#status = :in_progress',
|
||||
expr_attr_names={'#status': 'status'},
|
||||
expr_attr_values={
|
||||
@@ -297,8 +299,8 @@ def _set_status_as_failed(
|
||||
':in_progress': 'IN_PROGRESS',
|
||||
':score': score,
|
||||
':progress': progress,
|
||||
':access_expired': True,
|
||||
':updated_at': now_,
|
||||
':true': True,
|
||||
':now': now_,
|
||||
},
|
||||
exc_cls=EnrollmentConflictError,
|
||||
)
|
||||
|
||||
@@ -18,10 +18,12 @@ dyn = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
|
||||
@logger.inject_lambda_context
|
||||
def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
|
||||
new_image = event.detail['new_image']
|
||||
data = dyn.get_item(KeyPair(new_image['id'], 'konviva'))
|
||||
r = dyn.get_item(KeyPair(new_image['id'], 'konviva'))
|
||||
# Post-migration: uncomment the following line
|
||||
# r = dyn.get_item(KeyPair(new_image['id'], 'KONVIVA'))
|
||||
|
||||
try:
|
||||
result = konviva.cancel_enrollment(data['enrollment_id'])
|
||||
result = konviva.cancel_enrollment(r['enrollment_id'])
|
||||
except Exception as exc:
|
||||
logger.exception(exc)
|
||||
return False
|
||||
|
||||
@@ -47,10 +47,10 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
|
||||
transact.update(
|
||||
key=KeyPair(new_image['id'], '0'),
|
||||
update_expr='SET metadata__konviva_user_id = :user_id, \
|
||||
updated_at = :updated_at',
|
||||
updated_at = :now',
|
||||
expr_attr_values={
|
||||
':user_id': user_id,
|
||||
':updated_at': now_,
|
||||
':now': now_,
|
||||
},
|
||||
cond_expr='attribute_exists(sk)',
|
||||
)
|
||||
|
||||
@@ -35,6 +35,8 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
|
||||
with dyn.transact_writer() as transact:
|
||||
transact.update(
|
||||
key=KeyPair(new_image['id'], 'konviva'),
|
||||
# Post-migration: uncomment the following line
|
||||
# key=KeyPair(new_image['id'], 'KONVIVA'),
|
||||
update_expr='SET enrollment_id = :enrollment_id',
|
||||
cond_expr='attribute_exists(sk)',
|
||||
expr_attr_values={
|
||||
@@ -45,6 +47,8 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
|
||||
transact.put(
|
||||
item={
|
||||
'id': 'konviva',
|
||||
# Post-migration: uncomment the following line
|
||||
# 'id': 'KONVIVA',
|
||||
'sk': str(enrollment_id),
|
||||
'enrollment_id': new_image['id'],
|
||||
'created_at': now_,
|
||||
|
||||
Reference in New Issue
Block a user