update url

This commit is contained in:
2025-10-03 21:58:35 -03:00
parent 5ae2128dee
commit b9ac0f1dae
4 changed files with 85 additions and 30 deletions

View File

@@ -1,4 +1,6 @@
from http import HTTPStatus
from io import BytesIO
from typing import Any
from aws_lambda_powertools import Logger
from aws_lambda_powertools.event_handler.api_gateway import Router
@@ -6,12 +8,14 @@ from aws_lambda_powertools.event_handler.exceptions import (
BadRequestError,
NotFoundError,
)
from layercake.dateutils import now
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
from pydantic import UUID4, BaseModel
from python_multipart import parse_form
from api_gateway import JSONResponse
from boto3clients import dynamodb_client
from config import COURSE_TABLE
from form_data import parse
logger = Logger(__name__)
router = Router()
@@ -27,45 +31,53 @@ def get_course(course_id: str):
class Cert(BaseModel):
exp_interval: int
rawfile: bytes
exp_interval: int | None = None
rawfile: bytes | None = None
def model_dump(self, **kwargs) -> dict[str, Any]:
return super().model_dump(
exclude={'rawfile'},
exclude_none=True,
**kwargs,
)
class FormData(BaseModel):
class Course(BaseModel):
id: UUID4
name: str
access_period: int
cert: Cert | None = None
@router.post('/<course_id>')
@router.put('/<course_id>')
def edit_course(course_id: str):
event = router.current_event
if not event.decoded_body:
raise BadRequestError('Invalid request body')
ret = {'id': course_id}
body = BytesIO(event.decoded_body.encode())
def on_field(field):
field_name = field.field_name.decode().split('.')
if len(field_name) > 1:
field_name, subfield_name = field_name
print(field_name, subfield_name)
else:
field_name, *_ = field_name
ret[field_name] = field.value
parse_form(
event.headers, # type: ignore
body,
on_field=on_field,
on_file=on_field,
course = Course.model_validate(
{'id': course_id} | parse(event.headers, body),
)
now_ = now()
# print(ret.keys())
data = FormData.model_validate(ret)
print(data)
return {}
with dyn.transact_writer() as transact:
transact.update(
key=KeyPair(str(course.id), '0'),
update_expr='SET #name = :name, access_period = :access_period, \
cert = :cert, updated_at = :updated_at',
expr_attr_names={
'#name': 'name',
},
expr_attr_values={
':name': course.name,
':cert': course.cert.model_dump() if course.cert else None,
':access_period': course.access_period,
':updated_at': now_,
},
cond_expr='attribute_exists(sk)',
exc_cls=BadRequestError,
)
return JSONResponse(HTTPStatus.NO_CONTENT)