kintoneで設定したスケジュールにあわせてlambda(python)からSQSへメッセージを送る

2016/07/18

EC2の起動停止をそろそろ手動でやるのも疲れてきそうなのと、やはり停止するのを忘れてしまってもったいないので。

SQSを挟むほどの規模でもないのですが、慣れておきたかったのでSQSを使ってみています。
今回はSQSへメッセージを送るところまでを記載しています。
SQSからメッセージを受信してEC2の起動停止をlambdaで実行して結果をslackへ通知するのは次回のブログに書きます。

kintoneでアプリ作成

kintone-ec2-config

こんな項目を持ったアプリを作成します。

フォームはこんな感じです。

APIトークンを発行します

アプリの設定で詳細設定を展開して[APIトークン]をクリックします。

APIトークンの生成をします。
今回はデータを読むだけなのでアクセス権は閲覧のみにします。

保存した後、[設定完了]をクリックします。
※これをしないと生成したAPIトークンが反映しません。

ターミナルなどで確認します。

$ curl -H "X-Cybozu-API-Token: YOUR_TOKEN" "https://YOUR_DOMAIN.cybozu.com/k/v1/record.json?app=24&id=1"

1 2	$ curl -H "X-Cybozu-API-Token: YOUR_TOKEN" "https://YOUR_DOMAIN.cybozu.com/k/v1/record.json?app=24&id=1"

YOUR_TOKEN, YOUR_DOMAINは適宜変更します。

データのレスポンスが返ってくればOKです。

Terminalで見づらいのでPOSTMANで見ます。

祝日対応

やりながら祝日も対応しなければと思い、上記のkintoneの曜日に「祝」を足しました。

なんかいいAPIないかなーと探したのですが見つけられなかったのでGoogleカレンダーAPIを使いました。

from apiclient.discovery import build

API_KEY = 'XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX'
CALENDAR_ID = 'ja.japanese#holiday@group.v.calendar.google.com'

def getHolidays():

    try:
        service = build(serviceName='calendar', version='v3', developerKey=API_KEY)
        events = service.events().list(calendarId=CALENDAR_ID).execute()

        holidays = []
        for item in sorted(events['items'], key=lambda item: item['start']['date']):
            holidays.append(item['start']['date'])

        return holidays

    except Exception as e:
        return e

from apiclient.discovery import build

API_KEY = 'XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX'

CALENDAR_ID = 'ja.japanese#holiday@group.v.calendar.google.com'

def getHolidays():

try:

service = build(serviceName='calendar', version='v3', developerKey=API_KEY)

events = service.events().list(calendarId=CALENDAR_ID).execute()

holidays = []

for item in sorted(events['items'], key=lambda item: item['start']['date']):

holidays.append(item['start']['date'])

return holidays

except Exception as e:

return e

これをlambdaファンクションと同じディレクトリに格納しておきます。

kintoneからデータ取得してSQSへエンキューするlambda

GitHubはこちらです。

kintoneのレコードから起動対象か停止対象かをチェックします。

def checkValidate(type_str,record):
    field_name = ''
    field_value = ''
    if type_str == 'start':
        field_name = 'c_start_check'
        field_value = u'開始'
    elif type_str == 'stop':
        field_name = 'c_stop_check'
        field_value = u'停止'
    check_item = record[field_name]['value']
    if len(check_item) > 0:
        if check_item[0] == field_value:
            return True
    return False

def checkDate(type_str,record,holidays,date_str,week_day_str):
    field_name = ''
    if type_str == 'start':
        field_name = 'c_start_week_day'
    elif type_str == 'stop':
        field_name = 'c_stop_week_day'
    week_day_array = record[field_name]['value']
    if len(week_day_array) > 0:
        if u'祝' in week_day_array and date_str in holidays:
            return True
        elif week_day_str in week_day_array:
            if u'祝' in week_day_array:
                return True
            else:
                if date_str not in holidays:
                    return True
    return False

def checkItem(type_str,record,time_str):
    field_name = ''
    if type_str == 'start':
        field_name = 't_start_time'
    elif type_str == 'stop':
        field_name = 't_stop_time'
    check_time = record[field_name]['value']
    if time_str[0:4] == check_time[0:4]:
        return True
    return False

def checkValidate(type_str,record):

field_name = ''

field_value = ''

if type_str == 'start':

field_name = 'c_start_check'

field_value = u'開始'

elif type_str == 'stop':

field_name = 'c_stop_check'

field_value = u'停止'

check_item = record[field_name]['value']

if len(check_item) > 0:

if check_item[0] == field_value:

return True

return False

def checkDate(type_str,record,holidays,date_str,week_day_str):

field_name = ''

if type_str == 'start':

field_name = 'c_start_week_day'

elif type_str == 'stop':

field_name = 'c_stop_week_day'

week_day_array = record[field_name]['value']

if len(week_day_array) > 0:

if u'祝' in week_day_array and date_str in holidays:

return True

elif week_day_str in week_day_array:

if u'祝' in week_day_array:

return True

else:

if date_str not in holidays:

return True

return False

def checkItem(type_str,record,time_str):

field_name = ''

if type_str == 'start':

field_name = 't_start_time'

elif type_str == 'stop':

field_name = 't_stop_time'

check_time = record[field_name]['value']

if time_str[0:4] == check_time[0:4]:

return True

return False

起動停止対象の場合辞書型のメッセージを配列に格納します。

def getQueueList():
    now = datetime.now(pytz.timezone('Asia/Tokyo'))
    date_str = now.strftime('%Y-%m-%d')
    time_str = now.strftime("%H:%M")
    week_day_str = getWeekDayStr(now)
    headers = {HEADERS_KEY:HEADERS_VALUE}
    response_record = requests.get(KINTONE_URL,headers=headers)
    record_data = json.loads(response_record.text)
    records=record_data['records']
    holidays = get_holiday.getHolidays()
    queue_list = []
    for record in records:
        ec2_name = record['s_ec2_name_tag']['value']
        for type_str in ['start','stop']:
            if checkValidate(type_str,record):
                if checkDate(type_str,record,holidays,date_str,week_day_str):
                    if checkItem(type_str,record,time_str):
                        queue_list.append({'type':type_str,'ec2_name':ec2_name,'datetime':date_str + ' ' + time_str})
    return queue_list

def getQueueList():

now = datetime.now(pytz.timezone('Asia/Tokyo'))

date_str = now.strftime('%Y-%m-%d')

time_str = now.strftime("%H:%M")

week_day_str = getWeekDayStr(now)

headers = {HEADERS_KEY:HEADERS_VALUE}

response_record = requests.get(KINTONE_URL,headers=headers)

record_data = json.loads(response_record.text)

records=record_data['records']

holidays = get_holiday.getHolidays()

queue_list = []

for record in records:

ec2_name = record['s_ec2_name_tag']['value']

for type_str in ['start','stop']:

if checkValidate(type_str,record):

if checkDate(type_str,record,holidays,date_str,week_day_str):

if checkItem(type_str,record,time_str):

queue_list.append({'type':type_str,'ec2_name':ec2_name,'datetime':date_str + ' ' + time_str})

return queue_list

SQSにメッセージをエンキューします。

def set_enqueue(queue_list):
    try:
        sqs_client = boto3.client('sqs')
        for queue in queue_list:
            response = sqs_client.send_message(
                QueueUrl = SQS_URL,
                MessageBody = json.dumps(queue)
            )
    except Exception as e:
        logger.info(e)

def set_enqueue(queue_list):

try:

sqs_client = boto3.client('sqs')

for queue in queue_list:

response = sqs_client.send_message(

QueueUrl = SQS_URL,

MessageBody = json.dumps(queue)

)

except Exception as e:

logger.info(e)

実行してます。

def handler(event, context):
    try:
        queue_list = getQueueList()
        if len(queue_list) > 0:
            set_enqueue(queue_list)

    except Exception as e:
       logger.info(e)
       raise e