add support to milti transaction

This commit is contained in:
2026-01-13 03:18:05 -03:00
parent ad23e9aa51
commit da10a36a1d
11 changed files with 287 additions and 56 deletions

View File

@@ -1,3 +1,4 @@
from enum import Enum
from http import HTTPStatus
from typing import Any
from urllib.parse import parse_qsl
@@ -7,6 +8,7 @@ from aws_lambda_powertools.event_handler.api_gateway import (
APIGatewayHttpResolver,
Response,
)
from aws_lambda_powertools.event_handler.exceptions import NotFoundError
from aws_lambda_powertools.logging import correlation_paths
from aws_lambda_powertools.utilities.typing import LambdaContext
from layercake.dateutils import now
@@ -21,35 +23,73 @@ app = APIGatewayHttpResolver(enable_validation=True)
dyn = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
class OrderNotFoundError(NotFoundError): ...
class InvoiceNotFoundError(NotFoundError): ...
class StatusAttr(Enum):
PAID = 'paid_at'
CANCELED = 'canceled_at'
REFUNDED = 'refunded_at'
EXPIRED = 'expired_at'
EXTERNALLY_PAID = 'paid_at'
def _status_attr(status: str) -> StatusAttr | None:
try:
return StatusAttr[status]
except KeyError:
return None
@app.post('/<order_id>/postback')
@tracer.capture_method
def postback(order_id: str):
decoded_body = dict(parse_qsl(app.current_event.decoded_body))
logger.info('IUGU Postback', decoded_body=decoded_body)
event = decoded_body['event']
status = decoded_body['data[status]'].upper()
if event != 'invoice.status_changed':
now_ = now()
event = decoded_body['event']
status = decoded_body.get('data[status]', '').upper()
status_attr = _status_attr(status)
if event != 'invoice.status_changed' or not status_attr:
return Response(status_code=HTTPStatus.NO_CONTENT)
try:
dyn.update_item(
with dyn.transact_writer() as transact:
transact.update(
key=KeyPair(order_id, '0'),
update_expr='SET #status = :status, \
#status_attr = :now, \
updated_at = :now',
cond_expr='attribute_exists(sk)',
expr_attr_names={
'#status': 'status',
'#status_attr': status_attr.value,
},
expr_attr_values={
':status': status,
':now': now(),
':now': now_,
},
exc_cls=OrderNotFoundError,
)
except Exception:
return Response(status_code=HTTPStatus.NOT_FOUND)
else:
return Response(status_code=HTTPStatus.NO_CONTENT)
if status == 'EXTERNALLY_PAID':
transact.update(
key=KeyPair(order_id, 'INVOICE'),
cond_expr='attribute_exists(sk)',
update_expr='SET externally_paid = :true, \
updated_at = :now',
expr_attr_values={
':true': True,
':now': now_,
},
exc_cls=InvoiceNotFoundError,
)
return Response(status_code=HTTPStatus.NO_CONTENT)
@logger.inject_lambda_context(correlation_id_path=correlation_paths.API_GATEWAY_HTTP)

View File

@@ -4,9 +4,12 @@ from aws_lambda_powertools.utilities.data_classes import (
event_source,
)
from aws_lambda_powertools.utilities.typing import LambdaContext
from layercake.dateutils import now
from layercake.dynamodb import (
DynamoDBPersistenceLayer,
KeyPair,
)
from layercake.extra_types import CreditCard
from boto3clients import dynamodb_client
from config import IUGU_ACCOUNT_ID, IUGU_API_TOKEN, IUGU_TEST_MODE, ORDER_TABLE
@@ -25,4 +28,72 @@ iugu = Iugu(
@event_source(data_class=EventBridgeEvent)
@logger.inject_lambda_context
def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool: ...
def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
new_image = event.detail['new_image']
order_id = new_image['id']
invoice_id = new_image['invoice_id']
installments = new_image['installments']
credit_card = CreditCard(**new_image['credit_card'])
now_ = now()
token = iugu.payment_token(credit_card)
charge = iugu.charge(
invoice_id=invoice_id,
token=token['id'],
installments=installments,
)
with dyn.transact_writer() as transact:
transact.delete(key=KeyPair(order_id, 'TRANSACTION'))
transact.update(
key=KeyPair(order_id, 'TRANSACTION#STATS'),
update_expr='SET #count = if_not_exists(#count, :zero) + :one, \
updated_at = :now',
expr_attr_names={
'#count': 'payment_attempts',
},
expr_attr_values={
':zero': 0,
':one': 1,
':now': now(),
},
)
if charge['success'] is True:
transact.update(
key=KeyPair(order_id, '0'),
update_expr='SET #status = :status, \
paid_at = :now, \
updated_at = :now',
expr_attr_names={
'#status': 'status',
},
expr_attr_values={
':status': 'PAID',
':now': now_,
},
cond_expr='attribute_exists(sk)',
)
transact.put(
item={
'id': order_id,
'sk': f'TRANSACTION#ATTEMPTS#{now_.isoformat()}',
'brand': credit_card.brand,
'last4': credit_card.last4,
'status': 'SUCCEEDED',
'transaction': charge,
},
)
else:
transact.put(
item={
'id': order_id,
'sk': f'TRANSACTION#ATTEMPTS#{now_.isoformat()}',
'brand': credit_card.brand,
'last4': credit_card.last4,
'status': 'FAILED',
'transaction': charge,
},
)
return charge['success']

View File

@@ -4,9 +4,10 @@ from aws_lambda_powertools.utilities.data_classes import (
event_source,
)
from aws_lambda_powertools.utilities.typing import LambdaContext
from layercake.dateutils import now
from layercake.dateutils import now, ttl
from layercake.dynamodb import (
DynamoDBPersistenceLayer,
KeyPair,
SortKey,
TransactKey,
)
@@ -41,7 +42,8 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
r = dyn.collection.get_items(
TransactKey(order_id)
+ SortKey('ADDRESS', rename_key='address')
+ SortKey('ITEMS', path_spec='items', rename_key='items'),
+ SortKey('ITEMS', path_spec='items', rename_key='items')
+ SortKey('CREDIT_CARD#PAYMENT_INTENT', rename_key='credit_card'),
flatten_top=False,
)
@@ -59,21 +61,40 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
)
try:
dyn.put_item(
item={
'id': order_id,
'sk': 'INVOICE',
'payment_method': payment_method,
'secure_id': invoice['secure_id'],
'secure_url': invoice['secure_url'],
'created_at': now_,
# Uncomment this when adding for multiple payment providers
# 'payment_provider': 'iugu',
}
| ({'bank_slip': invoice['bank_slip']} if is_bank_slip else {})
| ({'pix': invoice['pix']} if is_pix else {}),
cond_expr='attribute_not_exists(sk)',
)
with dyn.transact_writer() as transact:
transact.put(
item={
'id': order_id,
'sk': 'INVOICE',
'payment_method': payment_method,
'secure_id': invoice['secure_id'],
'secure_url': invoice['secure_url'],
'created_at': now_,
# Uncomment this when adding for multiple payment providers
# 'payment_provider': 'iugu',
}
| ({'bank_slip': invoice['bank_slip']} if is_bank_slip else {})
| ({'pix': invoice['pix']} if is_pix else {}),
cond_expr='attribute_not_exists(sk)',
)
if 'credit_card' in r:
transact.delete(
key=KeyPair(order_id, 'CREDIT_CARD#PAYMENT_INTENT'),
)
transact.put(
item={
'id': order_id,
'sk': 'TRANSACTION',
'invoice_id': invoice['secure_id'],
'credit_card': r['credit_card'],
'installments': int(new_image.get('installments', 1)),
'ttl': ttl(start_dt=now_, minutes=5),
'created_at': now_,
},
cond_expr='attribute_not_exists(sk)',
)
except Exception:
pass

View File

@@ -71,6 +71,7 @@ class Order(BaseModel):
address: Address
items: tuple[Item, ...]
payment_method: PaymentMethod
discount: Decimal = Decimal('0')
cpf: str | None = None
cnpj: str | None = None
@@ -127,6 +128,7 @@ class Iugu:
}
for item in order.items
]
payload = {
'order_id': order.id,
'external_reference': order.id,
@@ -135,6 +137,7 @@ class Iugu:
'email': order.email,
'payable_with': order.payment_method.lower(),
'notification_url': postback_url.geturl(),
'discount_cents': int(order.discount * -100),
'payer': {
'name': order.name,
'email': order.email,

View File

@@ -116,8 +116,20 @@ Resources:
detail-type: [INSERT]
detail:
new_image:
sk: ['INVOICE']
payment_method: ['CREDIT_CARD']
sk: ['TRANSACTION']
invoice_id:
- exists: true
credit_card:
holder_name:
- exists: true
number:
- exists: true
exp_month:
- exists: true
exp_year:
- exists: true
cvv:
- exists: true
EventBillingAppendEnrollmentFunction:
Type: AWS::Serverless::Function

View File

@@ -0,0 +1,48 @@
import requests
from aws_lambda_powertools.utilities.typing.lambda_context import LambdaContext
from layercake.dynamodb import DynamoDBPersistenceLayer
import events.payments.charge_credit_card as app
from ...test_iugu import MockResponse
event = {
'detail': {
'new_image': {
'credit_card': {
'number': '4242424242424242',
'cvv': '123',
'created_at': '2026-01-13T02:33:57.088176-03:00',
'exp_month': '03',
'exp_year': '2027',
'ttl': 1768282737,
'holder_name': 'Sergio R Siqueira',
},
'created_at': '2026-01-13T02:34:00.065619-03:00',
'invoice_id': '493dd32a-b646-40b2-a307-0fcb3d42d954-2097',
'installments': 12,
'ttl': 1768282740,
'id': '121c1140-779d-4664-8d99-4a006a22f547',
'sk': 'TRANSACTION',
}
}
}
def test_charge_credit_card(
monkeypatch,
dynamodb_seeds,
dynamodb_persistence_layer: DynamoDBPersistenceLayer,
lambda_context: LambdaContext,
):
monkeypatch.setattr(
requests,
'post',
lambda url, *args, **kwargs: MockResponse(
'tests/samples/iugu_payment_token.json'
if 'payment_token' in url
else 'tests/samples/iugu_charge_paid.json'
),
)
assert app.lambda_handler(event, lambda_context) # type: ignore

View File

@@ -2,7 +2,7 @@ from decimal import Decimal
import requests
from aws_lambda_powertools.utilities.typing.lambda_context import LambdaContext
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, SortKey, TransactKey
import events.payments.create_invoice as app
@@ -89,3 +89,37 @@ def test_create_bank_slip_pix(
invoice['pix']['qrcode_text']
== 'http://faturas.iugu.com/iugu_pix/970cb579-c396-4e59-a323-ce61ae04f7bc-196c/test/pay'
)
def test_create_credit_card(
monkeypatch,
dynamodb_seeds,
dynamodb_persistence_layer: DynamoDBPersistenceLayer,
lambda_context: LambdaContext,
):
monkeypatch.setattr(
requests,
'post',
lambda *args, **kwargs: MockResponse(
'tests/samples/iugu_invoice_credit_card.json'
),
)
assert app.lambda_handler(_event('CREDIT_CARD'), lambda_context) # type: ignore
invoice = dynamodb_persistence_layer.collection.get_items(
TransactKey(order_id)
+ SortKey('0')
+ SortKey('INVOICE')
+ SortKey('TRANSACTION')
)
print(invoice)
# assert (
# invoice['pix']['qrcode_text']
# == 'http://faturas.iugu.com/iugu_pix/970cb579-c396-4e59-a323-ce61ae04f7bc-196c/test/pay'
# )
# assert (
# invoice['pix']['qrcode_text']
# == 'http://faturas.iugu.com/iugu_pix/970cb579-c396-4e59-a323-ce61ae04f7bc-196c/test/pay'
# )

View File

@@ -16,11 +16,12 @@
// Seeds for Iugu
// file: tests/test_app.py
// file: tests/events/payments/test_charge_credit_card.py
// file: tests/events/payments/test_create_invoice.py
{"id": "121c1140-779d-4664-8d99-4a006a22f547", "sk": "0", "total": "267.3", "name": "Beta Educação", "payment_method": "BANK_SLIP", "create_date": "2026-01-07T19:07:49.272967-03:00", "due_date": "2026-01-12T00:35:44.897447-03:00", "coupon": "10OFF", "discount": "-29.7", "updated_at": "2026-01-07T19:07:51.512605-03:00", "tenant_id": "cJtK9SsnJhKPyxESe7g3DG", "email": "org+15608435000190@users.noreply.saladeaula.digital", "org_id": "cJtK9SsnJhKPyxESe7g3DG", "cnpj": "15608435000190", "status": "PENDING", "subtotal": 297}
{"id": "121c1140-779d-4664-8d99-4a006a22f547", "sk": "ADDRESS", "city": "São José", "neighborhood": "Campinas", "address2": "", "postcode": "88101001", "state": "SC", "address1": "Avenida Presidente Kennedy", "created_at": "2026-01-07T19:07:49.272967-03:00"}
{"id": "121c1140-779d-4664-8d99-4a006a22f547", "sk": "ITEMS", "items": [{"name": "CIPA Grau de Risco 2", "id": "99bb3b60-4ded-4a8e-937c-ba2d78ec6454", "quantity": 3, "unit_price": 99}], "created_at": "2026-01-07T19:07:49.272967-03:00"}
{"id": "121c1140-779d-4664-8d99-4a006a22f547", "sk": "CREDIT_CARD#PAYMENT_INTENT", "credit_card": {"number": "4111111111111111","cvv": "123","created_at": "2026-01-13T02:33:57.088176-03:00","exp_month": "03","exp_year": "2027","ttl": 1768282737,"holder_name": "Sergio R Siqueira"}}
// User data
{"id": "5OxmMjL-ujoR5IMGegQz", "sk": "0", "name": "Sérgio R Siqueira"}

View File

@@ -35,6 +35,7 @@ event = {
'unit_price': 100,
},
],
'discount': -10,
}
@@ -53,27 +54,6 @@ class MockResponse:
def raise_for_status(): ...
def test_create_invoice_pix(monkeypatch):
monkeypatch.setattr(
requests,
'post',
lambda *args, **kwargs: MockResponse('tests/samples/iugu_invoice_pix.json'),
)
order = Order(
id=str(uuid4()),
payment_method='PIX', # type: ignore
**event,
)
invoice = iugu.create_invoice(order, postback_url='http://localhost')
assert invoice['id'] == '970CB579C3964E59A323CE61AE04F7BC'
assert (
invoice['pix']['qrcode_text']
== 'http://faturas.iugu.com/iugu_pix/970cb579-c396-4e59-a323-ce61ae04f7bc-196c/test/pay'
)
def test_create_invoice_bank_slip(monkeypatch):
monkeypatch.setattr(
requests,
@@ -97,6 +77,27 @@ def test_create_invoice_bank_slip(monkeypatch):
)
def test_create_invoice_pix(monkeypatch):
monkeypatch.setattr(
requests,
'post',
lambda *args, **kwargs: MockResponse('tests/samples/iugu_invoice_pix.json'),
)
order = Order(
id=str(uuid4()),
payment_method='PIX', # type: ignore
**event,
)
invoice = iugu.create_invoice(order, postback_url='http://localhost')
assert invoice['id'] == '970CB579C3964E59A323CE61AE04F7BC'
assert (
invoice['pix']['qrcode_text']
== 'http://faturas.iugu.com/iugu_pix/970cb579-c396-4e59-a323-ce61ae04f7bc-196c/test/pay'
)
def test_payment_token(monkeypatch):
monkeypatch.setattr(
requests,