add payment retries

2026-01-15 19:33:32 -03:00
parent 6d078d178b
commit ca52384b53
7 changed files with 116 additions and 1 deletions
--- a/api.saladeaula.digital/app/routes/orders/payment_retries.py
+++ b/api.saladeaula.digital/app/routes/orders/payment_retries.py
@@ -0,0 +1,66 @@
+from http import HTTPStatus
+from typing import Annotated
+
+from aws_lambda_powertools.event_handler.api_gateway import Router
+from aws_lambda_powertools.event_handler.exceptions import NotFoundError
+from aws_lambda_powertools.event_handler.openapi.params import Body
+from layercake.dateutils import now, ttl
+from layercake.dynamodb import (
+    DynamoDBPersistenceLayer,
+    KeyPair,
+)
+from layercake.extra_types import CreditCard
+from pydantic import UUID4
+
+from api_gateway import JSONResponse
+from boto3clients import dynamodb_client
+from config import ORDER_TABLE
+from exceptions import OrderConflictError
+
+router = Router()
+dyn = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
+
+
+class InvoiceNotFoundError(NotFoundError): ...
+
+
+@router.post('/<order_id>/payment-retries')
+def payment_retries(
+    order_id: str,
+    credit_card: Annotated[CreditCard, Body(embed=True)],
+    invoice_id: Annotated[UUID4 | str, Body(embed=True)],
+    installments: Annotated[int, Body(embed=True)],
+):
+    now_ = now()
+
+    with dyn.transact_writer() as transact:
+        transact.condition(
+            key=KeyPair(order_id, '0'),
+            cond_expr='attribute_exists(sk) AND installments = :installments',
+            expr_attr_values={
+                ':installments': installments,
+            },
+            exc_cls=OrderConflictError,
+        )
+        transact.condition(
+            key=KeyPair(order_id, 'INVOICE'),
+            cond_expr='attribute_exists(sk) AND invoice_id = :invoice_id',
+            expr_attr_values={
+                ':invoice_id': invoice_id,
+            },
+            exc_cls=InvoiceNotFoundError,
+        )
+        transact.put(
+            item={
+                'id': order_id,
+                'sk': 'TRANSACTION',
+                'invoice_id': invoice_id,
+                'credit_card': credit_card.model_dump(),
+                'installments': installments,
+                'ttl': ttl(start_dt=now_, minutes=5),
+                'created_at': now_,
+            },
+            cond_expr='attribute_not_exists(sk)',
+        )
+
+    return JSONResponse(status_code=HTTPStatus.CREATED)