81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
from decimal import Decimal
|
|
from http import HTTPStatus
|
|
|
|
from aws_lambda_powertools.event_handler.api_gateway import Router
|
|
from aws_lambda_powertools.event_handler.exceptions import (
|
|
BadRequestError,
|
|
)
|
|
from layercake.dateutils import now
|
|
from layercake.dynamodb import (
|
|
DynamoDBPersistenceLayer,
|
|
KeyPair,
|
|
PartitionKey,
|
|
)
|
|
from pydantic import BaseModel
|
|
from pydantic.types import UUID4
|
|
|
|
from api_gateway import JSONResponse
|
|
from boto3clients import dynamodb_client
|
|
from config import COURSE_TABLE
|
|
|
|
router = Router()
|
|
course_layer = DynamoDBPersistenceLayer(COURSE_TABLE, dynamodb_client)
|
|
|
|
|
|
@router.get('/<id>/custompricing', compress=True)
|
|
def get_custom_pricing(id: str):
|
|
return course_layer.collection.query(
|
|
PartitionKey(f'CUSTOM_PRICING#ORG#{id}'),
|
|
limit=100,
|
|
)
|
|
|
|
|
|
class CustomPricing(BaseModel):
|
|
course_id: UUID4
|
|
unit_price: Decimal
|
|
|
|
|
|
@router.post('/<id>/custompricing', compress=True)
|
|
def add_custom_pricing(id: str, custom_princing: CustomPricing):
|
|
now_ = now()
|
|
|
|
with course_layer.transact_writer() as transact:
|
|
transact.put(
|
|
item={
|
|
'id': f'CUSTOM_PRICING#ORG#{id}',
|
|
'sk': f'COURSE#{custom_princing.course_id}',
|
|
'unit_price': custom_princing.unit_price,
|
|
'created_at': now_,
|
|
},
|
|
cond_expr='attribute_not_exists(sk)',
|
|
exc_cls=CoursConflictError,
|
|
)
|
|
transact.condition(
|
|
key=KeyPair(str(custom_princing.course_id), '0'),
|
|
cond_expr='attribute_exists(sk)',
|
|
exc_cls=CourseNotFoundError,
|
|
)
|
|
|
|
return JSONResponse(status_code=HTTPStatus.CREATED)
|
|
|
|
|
|
class RemoveCustomPricing(BaseModel):
|
|
course_id: UUID4
|
|
|
|
|
|
@router.delete('/<id>/custompricing', compress=True)
|
|
def remove_custom_pricing(id: str, custom_princing: RemoveCustomPricing):
|
|
key = KeyPair(
|
|
f'CUSTOM_PRICING#ORG#{id}',
|
|
f'COURSE#{custom_princing.course_id}',
|
|
)
|
|
|
|
if course_layer.delete_item(key):
|
|
return JSONResponse(status_code=HTTPStatus.OK)
|
|
|
|
|
|
class CoursConflictError(BadRequestError): ...
|
|
|
|
|
|
class CourseNotFoundError(BadRequestError): ...
|