add payment retries

api.saladeaula.digital/app/app.py

@@ -49,6 +49,7 @@ app.include_router(users.orgs, prefix='/users')
 app.include_router(users.password, prefix='/users')
 app.include_router(orders.router, prefix='/orders')
 app.include_router(orders.checkout, prefix='/orders')
+app.include_router(orders.payment_retries, prefix='/orders')
 app.include_router(orgs.add, prefix='/orgs')
 app.include_router(orgs.address, prefix='/orgs')
 app.include_router(orgs.admins, prefix='/orgs')

api.saladeaula.digital/app/exceptions.py

@@ -14,6 +14,9 @@ class ConflictError(ServiceError):
 class OrderNotFoundError(NotFoundError): ...
 
 
+class OrderConflictError(NotFoundError): ...
+
+
 class UserNotFoundError(NotFoundError): ...
 
 

api.saladeaula.digital/app/routes/courses/__init__.py

@@ -112,6 +112,7 @@ def sample(course_id: str, s3_uri: Annotated[str, Body(embed=True)]):
     # Send template URI and data to Paperforge API to generate a PDF
     r = requests.post(
         PAPERFORGE_API,
+        timeout=(1, 3),
         json={
             'template_uri': s3_uri,
             'args': {

api.saladeaula.digital/app/routes/orders/__init__.py

@@ -11,8 +11,9 @@ from config import ORDER_TABLE
 from exceptions import OrderNotFoundError
 
 from .checkout import router as checkout
+from .payment_retries import router as payment_retries
 
-__all__ = ['checkout']
+__all__ = ['checkout', 'payment_retries']
 
 router = Router()
 dyn = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)

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)