PythonでAWS DynamoDBのCRUDを書いてみる のバックアップ(No.3) - 闘うITエンジニアの覚え書き

book.js

import os
import boto3
import json
import decimal

dynamodb = None
if os.environ and os.environ["DYNAMO_REGION"] and os.environ["DYNAMO_END_POINT"]:
    dynamodb = boto3.resource('dynamodb', region_name=os.environ["DYNAMO_REGION"], endpoint_url=os.environ["DYNAMO_END_POINT"])
else:
    dynamodb = boto3.resource('dynamodb', dynamo_opt)

# テーブル名
TABLE_NAME = "Books"

# 一覧
def list(event, context):
    res = dynamodb.Table(TABLE_NAME).scan()
    return { 'body' : str(res) }

# 一意検索
def get(event, context):
    res = dynamodb.Table(TABLE_NAME).get_item(
        Key={
            "id" : event["pathParameters"]["id"]
        }   
    )   
    return { 'body' : str(res) }

# 登録
def create(event, context):
    req = json.loads(event["body"]) if type(event["body"]) == str else event["body"]
    res = dynamodb.Table(TABLE_NAME).put_item(
        Item={
            "id": req["id"],
            "isbn": req["isbn"],
            "title": req["title"],
            "price": decimal.Decimal(req["price"])
        }   
    )
    return { 'body' : str(res) }

# 更新
def update(event, context):
    req = json.loads(event["body"]) if type(event["body"]) == str else event["body"]
    res = dynamodb.Table(TABLE_NAME).update_item(
        Key={
            'id': event["pathParameters"]["id"]
        },  
        UpdateExpression="set isbn=:i, title=:t, price=:p",
        ExpressionAttributeValues={
            ':i': req["isbn"],
            ':t': req["title"],
            ':p': decimal.Decimal(req["price"])
        },  
        ReturnValues="UPDATED_NEW"
    )
    return { 'body' : str(res) }

# 削除
def delete(event, context):
    res = dynamodb.Table(TABLE_NAME).delete_item(
        Key={
            'id': event["pathParameters"]["id"]
        }
    )
    return { 'body' : str(res) }

template-local.yml

AWSTemplateFormatVersion : '2010-09-09'
Transform: AWS::Serverless-2016-10-31
Description: Test serverless application.

Resources:

  BookListFunction:
    Type: AWS::Serverless::Function
    Properties:
      Handler: book.list
      Policies: AmazonDynamoDBFullAccess
      Runtime: python3.6
      MemorySize: 128 
      Timeout: 5
      Environment:
        Variables:
          DYNAMO_END_POINT: http://10.20.30.40:8000
          DYNAMO_REGION: localhost
      Events:
        List:
          Type: Api 
          Properties:
            Path: /book
            Method: get 

  BookGetFunction:
    Type: AWS::Serverless::Function
    Properties:
      Handler: book.get
      Policies: AmazonDynamoDBFullAccess
      Runtime: python3.6
      MemorySize: 128 
      Timeout: 5
      Environment:
        Variables:
          DYNAMO_END_POINT: http://10.20.30.40:8000
          DYNAMO_REGION: localhost
      Events:
        Get:
          Type: Api 
          Properties:
            Path: /book/{id}
            Method: get 

  BookCreateFunction:
    Type: AWS::Serverless::Function
    Properties:
      Handler: book.create
      Policies: AmazonDynamoDBFullAccess
      Runtime: python3.6
      MemorySize: 128 
      Timeout: 5
      Environment:
        Variables:
          DYNAMO_END_POINT: http://10.20.30.40:8000
          DYNAMO_REGION: localhost
      Events:
        Create:
          Type: Api 
          Properties:
            Path: /book
            Method: post

  BookUpdateFunction:
    Type: AWS::Serverless::Function
    Properties:
      Handler: book.update
      Policies: AmazonDynamoDBFullAccess
      Runtime: python3.6
      MemorySize: 128
      Timeout: 5
      Environment:
        Variables:
          DYNAMO_END_POINT: http://10.20.30.40:8000
          DYNAMO_REGION: localhost
      Events:
        Update:
          Type: Api
          Properties:
            Path: /book/{id}
            Method: put

  BookDeleteFunction:
    Type: AWS::Serverless::Function
    Properties:
      Handler: book.delete
      Policies: AmazonDynamoDBFullAccess
      Runtime: python3.6
      MemorySize: 128
      Timeout: 5
      Environment:
        Variables:
          DYNAMO_END_POINT: http://10.20.30.40:8000
          DYNAMO_REGION: localhost
      Events:
        Delete:
          Type: Api
          Properties:
            Path: /book/{id}
            Method: delete