fix address
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user