add fix

http-api/user.py

@@ -0,0 +1,85 @@
+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, '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.')