lambda_reviews.py

import os
import logging
import json
import boto3
from botocore.exceptions import ClientError
from boto3.dynamodb.conditions import Key

logger = logging.getLogger()
logger.setLevel(logging.INFO)

dynamodb_client = boto3.client("dynamodb")

if os.getenv("AWS_SAM_LOCAL"):
    dynamodb_client = boto3.client('dynamodb', endpoint_url='http://docker.for.mac.localhost:8000/')

table_name = os.environ["TABLE_NAME"]

logger.info("Using table {}".format(table_name))


def create_response(status_code, body):
    response = {
        "statusCode": status_code,
        "headers": {"Content-Type": "application/json"},
        "body": json.dumps(body) or ""
    }
    logger.info("Created response: {}".format(response))
    return response


# WIP
def handler_get_all(event, context):
    logger.info("Received event {}".format(event))
    # Get profile_id from authentication header
    profile_id = "33"

    try:
        result = dynamodb_client.query(
            KeyConditionExpression=Key("profile_id").eq(profile_id),
            TableName=table_name
        )
        logger.info("Get Item result: {}".format(result))
        return create_response(200, result["Items"])
    except ClientError as e:
        logger.info("Get Item failed: {}".format(e))
        return create_response(500, e.response["Error"]["Message"])


def handler_get(event, context):
    logger.info("Received event {}".format(event))

    try:
        result = dynamodb_client.get_item(
            TableName=table_name,
            Key={"id": {"S": event["pathParameters"]["reviewId"]}}
        )
        logger.info("Get Item result: {}".format(result))
        return create_response(200, result["Item"])
    except ClientError as e:
        logger.info("Get Item failed: {}".format(e))
        return create_response(500, e.response["Error"]["Message"])


def handler_put(event, context):
    logger.info("Received event {}".format(event))
    item = json.loads(event["body"])['item']

    try:
        result = dynamodb_client.put_item(
            TableName=table_name,
            Item=item
        )
        logger.info("Get Item result: {}".format(result))
        return create_response(200, result)
    except ClientError as e:
        logger.info("Get Item failed: {}".format(e))
        return create_response(500, e.response["Error"]["Message"])


def handler_delete(event, context):
    logger.info("Received event {}".format(event))

    try:
        result = dynamodb_client.delete_item(
            TableName=table_name,
            Key={"id": {"S": event["pathParameters"]["reviewId"]}}
        )
        logger.info("Delete Item result: {}".format(result))
        return create_response(200, result)
    except ClientError as e:
        logger.info("Delete Item failed: {}".format(e))
        return create_response(500, e.response["Error"]["Message"])