File size: 3,150 Bytes
243f395 f2a79fa 3dbe011 c6054f0 3dbe011 c6054f0 f2a79fa da5737b 243f395 c6054f0 f2a79fa 3dbe011 f2a79fa 3dbe011 f2a79fa 3dbe011 f2a79fa 3dbe011 f2a79fa 3dbe011 6f1250c 3dbe011 6f1250c f2a79fa 3dbe011 f2a79fa 6f1250c c6054f0 da5737b 6f1250c da5737b f2a79fa da5737b f2a79fa 3dbe011 f2a79fa 3dbe011 c6054f0 3dbe011 f2a79fa 3dbe011 f2a79fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 |
import json
import time
from http import HTTPStatus
from aws_lambda_powertools import Logger
from aws_lambda_powertools.event_handler import content_types, Response
from aws_lambda_powertools.utilities.typing import LambdaContext
from pydantic import BaseModel, ValidationError
from src.utilities.type_hints import input_floatlist, input_floatlist2
from src.utilities.utilities import base64_decode
logger = Logger()
class BBoxWithPointInput(BaseModel):
bbox: input_floatlist
points: input_floatlist2
duration_run: float = 0
message: str = ""
request_id: str = ""
def get_response(status: int, start_time: float, request_id: str, output: BBoxWithPointInput = None) -> str:
"""
Return a response for frontend clients.
Args:
status: status response
start_time: request start time (float)
request_id: str
output: dict we embed into our response
Returns:
dict: response
"""
messages = {200: "ok", 422: "validation error", 500: "internal server error"}
body = f"{messages[status]}, request_id: {request_id}."
if status == 200:
output.duration_run = time.time() - start_time
output.message = messages[status]
output.request_id = request_id
body = output.model_dump_json()
response = {
"statusCode": status,
"headers": {
"Content-Type": content_types.APPLICATION_JSON if status == 200 else content_types.TEXT_PLAIN
},
"body": body,
"isBase64Encoded": False
}
logger.info(f"response type:{type(response)} => {response}.")
return json.dumps(response)
def lambda_handler(event: dict, context: LambdaContext):
logger.info(f"start with aws_request_id:{context.aws_request_id}.")
start_time = time.time()
if "version" in event:
logger.info(f"event version: {event['version']}.")
try:
logger.info(f"event:{json.dumps(event)}...")
logger.info(f"context:{context}...")
try:
body = event["body"]
except Exception as e_constants1:
logger.error(f"e_constants1:{e_constants1}.")
body = event
logger.info(f"body: {type(body)}, {body}...")
if isinstance(body, str):
body_decoded_str = base64_decode(body)
logger.info(f"body_decoded_str: {type(body_decoded_str)}, {body_decoded_str}...")
body = json.loads(body_decoded_str)
logger.info(f"body:{body}...")
try:
bbox_points = BBoxWithPointInput(bbox=body["bbox"], points=body["points"])
logger.info(f"validation ok, bbox_points:{bbox_points}...")
response = get_response(HTTPStatus.OK.value, start_time, context.aws_request_id, bbox_points)
except ValidationError as ve:
logger.error(f"validation error:{ve}.")
response = get_response(422, start_time, context.aws_request_id)
except Exception as e:
logger.error(f"exception:{e}.")
response = get_response(500, start_time, context.aws_request_id)
logger.info(f"response_dumped:{response}...")
return response
|