fix address

This commit is contained in:
2025-07-15 20:26:42 -03:00
parent 5c80502715
commit 893dabe24b
3 changed files with 23 additions and 12 deletions

View File

@@ -5,7 +5,6 @@ from aws_lambda_powertools.event_handler.exceptions import (
BadRequestError as PowertoolsBadRequestError,
)
from layercake.dynamodb import (
DynamoDBCollection,
DynamoDBPersistenceLayer,
KeyPair,
MissingError,
@@ -25,7 +24,6 @@ class BadRequestError(MissingError, PowertoolsBadRequestError): ...
router = Router()
user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
user_collect = DynamoDBCollection(user_layer, exc_cls=BadRequestError)
@router.get(
@@ -37,7 +35,7 @@ user_collect = DynamoDBCollection(user_layer, exc_cls=BadRequestError)
def get_emails(id: str):
start_key = router.current_event.get_query_string_value('start_key', None)
return user_collect.query(
return user_layer.collection.query(
KeyPair(id, PrefixKey('emails')),
start_key=start_key,
)
@@ -52,7 +50,7 @@ class Email(BaseModel):
compress=True,
tags=['User'],
summary='Add user email',
middlewares=[AuditLogMiddleware('EMAIL_ADD', user_collect, ('email',))],
middlewares=[AuditLogMiddleware('EMAIL_ADD', user_layer.collection, ('email',))],
)
def post_email(id: str, payload: Email):
add_email(id, payload.email, persistence_layer=user_layer)
@@ -77,7 +75,7 @@ class EmailAsPrimary(BaseModel):
middlewares=[
AuditLogMiddleware(
'EMAIL_CHANGE',
user_collect,
user_layer.collection,
(
'new_email',
'old_email',
@@ -105,7 +103,7 @@ def patch_email(id: str, payload: EmailAsPrimary):
compress=True,
tags=['User'],
summary='Delete user email',
middlewares=[AuditLogMiddleware('EMAIL_DEL', user_collect, ('email',))],
middlewares=[AuditLogMiddleware('EMAIL_DEL', user_layer.collection, ('email',))],
)
def delete_email(id: str, payload: Email):
del_email(