MySQL にアクセスする Lambda 関数 (Python3)

python3 で MySQL にアクセスする AWS の Lambda 関数を書いてみました。
ライブラリーは、pymysql を使います。

mysql_scan.py

# -*- coding: utf-8 -*-
#
#   mysql_scan.py
#
#                   Jan/27/2018
# --------------------------------------------------------------------
import  sys
import  json
#
from mysql_scan_exec import mysql_scan_exec_proc
# --------------------------------------------------------------------
def mysql_scan_handler(event, context):
    sys.stderr.write("*** mysql_scan_handler *** start ***\n")
#
    table_name = "No table_name"
#
    sys.stderr.write("Received event: " + json.dumps(event, indent=2) + "\n")
    if ('table_name' in event):
        sys.stderr.write("table_name = " + event['table_name'] + "\n")
        table_name = event['table_name']
#
    version = "Jan/27/2018 PM 18:46"
    sys.stderr.write("version: " + version + "\n")
#
    sys.stderr.write("*** mysql_scan_handler *** ccc ***\n")
#
    rvalue = mysql_scan_exec_proc(table_name)
#
    sys.stderr.write("*** mysql_scan_handler *** end ***\n")
#
    return rvalue
# --------------------------------------------------------------------

mysql_scan_exec.py

# -*- coding: utf-8 -*-
#
#   mysql_scan_exec.py
#
#                   Jan/28/2018
# --------------------------------------------------------------------
import  os
import  sys
import  json
sys.path.append(os.path.join(os.path.abspath(os.path.dirname(__file__)),'api'))
import pymysql
#
from mysql_config import mysql_config_proc
# --------------------------------------------------------------------
def mysql_scan_exec_proc(table_name):
    rvalue = []
    connection = mysql_config_proc()
    cursor = connection.cursor()
#
    sql_str="select * from " + table_name
    sys.stderr.write("sql_str = " + sql_str + "\n")
    cursor.execute (sql_str)
    rows = cursor.fetchall ()
#
    cursor.close()
    connection.close()
#
    return rows
# --------------------------------------------------------------------

mysql_config.py

# -*- coding: utf-8 -*-
#
#   mysql_config.py
#
#                   Jan/28/2018
# ---------------------------------------------------------------
import os
import sys
sys.path.append(os.path.join(os.path.abspath(os.path.dirname(__file__)),'api'))

import pymysql
# ---------------------------------------------------------------
def mysql_config_proc():
#
    host_aa='test_aa.abcdefghijkl.ap-northeast-1.rds.amazonaws.com'
    data_base = 'city'
    user_aa ='scott'
    password_aa = 'tiger123'
#
    connection = pymysql.connect(host_aa, user=user_aa, \
        passwd=password_aa, db=data_base, \
        cursorclass=pymysql.cursors.DictCursor, connect_timeout=5)
    return connection
#
# ---------------------------------------------------------------

ラムダ関数の作成手順
api というフォルダーに pymysql を保存

mkdir api
pip install PyMySQL -t ./api

ラムダ関数を作成するスクリプト
morning_exec_role というロールが作られているとします。

function_create.sh

#
FUNCTION='mysql_scan'
ZIP_FILE=$FUNCTION".zip"
#
zip -r $ZIP_FILE $FUNCTION".py" mysql_config.py mysql_scan_exec.py api
#
#
aws lambda create-function \
    --function-name $FUNCTION \
    --runtime python3.6 \
    --role arn:aws:iam::123456789012:role/morning_exec_role \
    --handler $FUNCTION.mysql_scan_handler \
    --zip-file fileb://$ZIP_FILE \
    --region ap-northeast-1
#

Lambda 関数を更新するスクリプト
(何かを変更した時に使います。）

function_update.sh

#
FUNCTION='mysql_scan'
ZIP_FILE=$FUNCTION".zip"
#
zip -r $ZIP_FILE $FUNCTION".py" mysql_config.py mysql_scan_exec.py api
#
#
aws lambda update-function-code \
    --function-name $FUNCTION \
    --zip-file fileb://$ZIP_FILE
#