52 lines
1.3 KiB
Python
52 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.exceptions import NotFoundError
|
|
from aws_lambda_powertools.event_handler.openapi.params import Query
|
|
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
|
|
|
|
from boto3clients import dynamodb_client
|
|
from config import ORDER_TABLE, USER_TABLE
|
|
|
|
router = Router()
|
|
dyn = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
|
|
|
|
|
|
class BillingNotFoundError(NotFoundError): ...
|
|
|
|
|
|
@router.get('/<org_id>/subscription')
|
|
def subscription(org_id: str):
|
|
return dyn.collection.get_item(
|
|
KeyPair(
|
|
pk=org_id,
|
|
sk='METADATA#SUBSCRIPTION',
|
|
table_name=USER_TABLE,
|
|
),
|
|
raise_on_error=False,
|
|
default={},
|
|
)
|
|
|
|
|
|
@router.get('/<org_id>/billing')
|
|
def billing(
|
|
org_id: str,
|
|
start_date: Annotated[date, Query()],
|
|
end_date: Annotated[date, Query()],
|
|
):
|
|
pk = f'BILLING#ORG#{org_id}'
|
|
sk = f'START#{start_date}#END#{end_date}'
|
|
r = dyn.collection.get_item(
|
|
KeyPair(pk, sk),
|
|
exc_cls=BillingNotFoundError,
|
|
)
|
|
|
|
return r | dyn.collection.query(
|
|
KeyPair(
|
|
pk=pk,
|
|
sk=f'{sk}#ENROLLMENT',
|
|
),
|
|
limit=150,
|
|
)
|