86 lines
2.0 KiB
Python
86 lines
2.0 KiB
Python
from aws_lambda_powertools.event_handler.exceptions import (
|
|
BadRequestError,
|
|
)
|
|
from botocore.exceptions import ClientError
|
|
from layercake.dateutils import now
|
|
from layercake.dynamodb import (
|
|
ComposeKey,
|
|
DynamoDBPersistenceLayer,
|
|
KeyPair,
|
|
TransactItems,
|
|
)
|
|
|
|
|
|
def add_email(
|
|
id: str,
|
|
email: str,
|
|
/,
|
|
*,
|
|
persistence_layer: DynamoDBPersistenceLayer,
|
|
):
|
|
now_ = now()
|
|
transact = TransactItems(persistence_layer.table_name)
|
|
transact.update(
|
|
key=KeyPair(id, '0'),
|
|
update_expr='ADD emails :email',
|
|
expr_attr_values={
|
|
':email': {email},
|
|
},
|
|
)
|
|
transact.put(
|
|
item={
|
|
'id': id,
|
|
'sk': f'emails#{email}',
|
|
'email_primary': False,
|
|
'email_verified': False,
|
|
'create_date': now_,
|
|
},
|
|
cond_expr='attribute_not_exists(sk)',
|
|
)
|
|
transact.put(
|
|
item={
|
|
'id': 'email',
|
|
'sk': email,
|
|
'user_id': id,
|
|
'create_date': now_,
|
|
},
|
|
cond_expr='attribute_not_exists(sk)',
|
|
)
|
|
|
|
try:
|
|
return persistence_layer.transact_write_items(transact)
|
|
except ClientError:
|
|
raise BadRequestError('Email already exists.')
|
|
|
|
|
|
def del_email(
|
|
id: str,
|
|
email: str,
|
|
/,
|
|
*,
|
|
persistence_layer: DynamoDBPersistenceLayer,
|
|
) -> bool:
|
|
"""Delete any email except the primary email."""
|
|
|
|
transact = TransactItems(persistence_layer.table_name)
|
|
transact.delete(
|
|
key=KeyPair('email', email),
|
|
)
|
|
transact.delete(
|
|
key=KeyPair(id, ComposeKey(email, prefix='emails')),
|
|
cond_expr='email_primary <> :primary',
|
|
expr_attr_values={':primary': True},
|
|
)
|
|
transact.update(
|
|
key=KeyPair(id, '0'),
|
|
update_expr='DELETE emails :email',
|
|
expr_attr_values={
|
|
':email': {email},
|
|
},
|
|
)
|
|
|
|
try:
|
|
return persistence_layer.transact_write_items(transact)
|
|
except ClientError:
|
|
raise BadRequestError('Cannot remove the primary email.')
|