スイッチIoTで服薬管理 〜M5Stack × AWS〜 [Part1] | 服薬カウンター（2/4）

import json
import boto3
from boto3.dynamodb.conditions import Key
from datetime import datetime, timedelta, timezone, date

dynamo = boto3.resource('dynamodb')
table = dynamo.Table('medication-management-table')

def lambda_handler(event, context):
    statusCode = 200
    responseBody = {}

    # ルートキーの取得
    routeKey = event['routeKey']

    try:
        if routeKey == 'PUT /items/{year}/{month}/{day}':
        # 服薬情報登録のPUT APIがコールされた場合

            # パスパラメータ取得
            year = event['pathParameters']['year']         # 年
            month = event['pathParameters']['month']       # 月
            day = event['pathParameters']['day']           # 日

            # パーティションキー（yyyymmdd）の文字列を生成
            yyyymmdd = year + month.zfill(2) + day.zfill(2)
            # ソートキー（entry_datetime）の文字列を生成
            JST = timezone(timedelta(hours=+9), 'JST')
            entry_datetime = datetime.now(JST).strftime('%Y-%m-%d %H:%M:%S')

            # put_itemメソッドでデータ登録
            data = {
                'yyyymmdd' : yyyymmdd,
                'entry_datetime' : entry_datetime
            }
            table.put_item(Item=data)

            responseBody = json.dumps('put ok')

        elif routeKey == 'DELETE /items/{year}/{month}/{day}':
        # 服薬情報削除のDELETE APIがコールされた場合

            # パスパラメータ取得
            year = event['pathParameters']['year']         # 年
            month = event['pathParameters']['month']       # 月
            day = event['pathParameters']['day']           # 日

            # パーティションキー（yyyymmdd）の文字列を生成
            yyyymmdd = year + month.zfill(2) + day.zfill(2)

            # パーティションキー（yyyymmdd）を検索キーとして降順で検索
            data = table.query(
                KeyConditionExpression = Key('yyyymmdd').eq(yyyymmdd),
                ScanIndexForward = False
            )

            # 先頭データのソートキー（entry_datetime）を取得
            entry_datetime = ''
            if data['Count'] > 0:
                entry_datetime = data['Items'][0]['entry_datetime']

            if entry_datetime:
                # 取得したパーティションキーとソートキーをターゲットとして、
                # delete_itemメソッドでデータ削除
                data = {
                    'yyyymmdd' : yyyymmdd,
                    'entry_datetime' : entry_datetime
                }
                res = table.delete_item(Key=data)

            responseBody = json.dumps('delete ok')

        elif routeKey == 'GET /items/{year}/{month}/{day}/{offset}':
        # 服薬情報取得のGET APIがコールされた場合

            # パスパラメータ取得
            year = int(event['pathParameters']['year'])         # 年
            month = int(event['pathParameters']['month'])       # 月
            day = int(event['pathParameters']['day'])           # 日
            offset = int(event['pathParameters']['offset'])     # オフセット

            # 日付計算（オフセット分を加算する）
            key_date = date(year, month, day)
            if offset != 0:
                key_date = key_date + timedelta(days=offset) 

            # YYYYMMDDの文字列に変換
            yyyymmdd = key_date.strftime('%Y%m%d')

            # queryメソッドで指定日付のデータ検索
            # 「ScanIndexForward = False」をつけて降順とする
            data = table.query(
                KeyConditionExpression = Key('yyyymmdd').eq(yyyymmdd),
                ScanIndexForward = False
            )

            # データ件数を取得
            count = data['Count']
            # 先頭データのentry_datetimeを最新登録日付として取得
            lastest_entry = 'None'
            if count > 0:
                lastest_entry = data['Items'][0]['entry_datetime']

            # レスポンスボディに取得結果をJSON形式でセット
            res = {
                'display_date' : key_date.strftime('%Y/%m/%d'),     # 表示日付
                'count' : count,                                    # 件数
                'lastest_entry' : lastest_entry                     # 最新登録日付
            }
            responseBody = json.dumps(res)

    except Exception as e:
        # エラーが起きた時の処理
        print(e)
        statusCode = 500
        responseBody = json.dumps('エラー発生')

    # レスポンス返却
    return {
        'statusCode' : statusCode,
        'body' : responseBody,
        'headers' : {
            'Content-Type': 'application/json'
        }
    }