File size: 2,966 Bytes
243f395 f2a79fa 3dbe011 c6054f0 f2a79fa 28ea6eb f2a79fa da5737b 243f395 28ea6eb c6054f0 f2a79fa 3dbe011 f2a79fa 3dbe011 f2a79fa 3dbe011 f2a79fa 28ea6eb 6f1250c 28ea6eb f2a79fa 3dbe011 f2a79fa 6f1250c c6054f0 da5737b 6f1250c da5737b f2a79fa da5737b f2a79fa 3dbe011 f2a79fa 28ea6eb c6054f0 28ea6eb 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 |
import json
import time
from http import HTTPStatus
from aws_lambda_powertools import Logger
from aws_lambda_powertools.utilities.typing import LambdaContext
from pydantic import BaseModel, ValidationError
from src.utilities.constants import CUSTOM_RESPONSE_MESSAGES
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
message: 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
"""
duration_run = time.time() - start_time
if output and status == 200:
output.message = f"{CUSTOM_RESPONSE_MESSAGES[status]} - duration_run: {duration_run}, request_id: {request_id}."
return output.model_dump_json()
elif status == 200:
raise KeyError("missing BBoxWithPointInput...")
return json.dumps({
"statusCode": status,
"message": f"{CUSTOM_RESPONSE_MESSAGES[status]} - duration_run: {duration_run}, request_id: {request_id}."
})
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(HTTPStatus.UNPROCESSABLE_ENTITY.value, start_time, context.aws_request_id)
except Exception as e:
logger.error(f"exception:{e}.")
response = get_response(HTTPStatus.INTERNAL_SERVER_ERROR.value, start_time, context.aws_request_id)
logger.info(f"response_dumped:{response}...")
return response
|