44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from datetime import date
|
|
from typing import Annotated
|
|
|
|
from aws_lambda_powertools.event_handler.api_gateway import Router
|
|
from aws_lambda_powertools.event_handler.openapi.params import Path, Query
|
|
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
|
|
|
|
from boto3clients import dynamodb_client
|
|
from config import ORDER_TABLE, USER_TABLE
|
|
|
|
router = Router()
|
|
order_layer = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
|
|
user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
|
|
|
|
|
|
@router.get('/<id>', compress=True, tags=['Billing'])
|
|
def get_billing(
|
|
id: Annotated[str, Path()],
|
|
start_date: Annotated[date, Query()],
|
|
end_date: Annotated[date, Query()],
|
|
):
|
|
result = order_layer.collection.query(
|
|
KeyPair(
|
|
pk=f'BILLING#ORG#{id}',
|
|
sk=f'START#{start_date}#END#{end_date}',
|
|
),
|
|
limit=100,
|
|
)
|
|
return result
|
|
|
|
|
|
@router.get('/<id>/terms', compress=True, tags=['Billing'])
|
|
def get_terms(id: Annotated[str, Path()]):
|
|
return user_layer.collection.get_item(
|
|
KeyPair(
|
|
pk=id,
|
|
sk='metadata#billing_policy',
|
|
# Post-migration: uncomment the following line
|
|
# sk='METADATA#BILLING_TERMS',
|
|
),
|
|
raise_on_error=False,
|
|
default={},
|
|
)
|