마법의 말, 경비 절감

선인님의 참고 등

이번에 하고 싶은 일

평일 or 주말 or 공휴일

Lambda에서 외부 모듈을 사용할 때는 zip으로 코드 전체 업로드

cd [適当なディレクトリ]
sudo pip install --upgrade google-api-python-client -t ./

# 同じディレクトリ内でLambdaで実行するメインファイルを書く
vim hoge.py

import boto3
import datetime
import sys

from apiclient.discovery import build


# https://console.developers.google.com/project ここらへんから
API_KEY = '[Googleの開発APIキー]'
CALENDAR_ID = 'ja.japanese#[email protected]'

# 年末年始とか会社の休みは自動起動しない(YYYY-MM-DDで列挙)。
company_holiday_list = []


# 関数名(ここではlambda_handler)と、ファイル名(ここではhoge.py)を
# LambdaのHandler名に設定する ex.) hoge.lambda_handler
def lambda_handler(event, context):
    client = boto3.client('ec2')

    # タグ名「AutoShutdown」がAUTOだと自動起動/停止
    # タグ名「AutoShutdown」がONだと自動停止のみ行う
    query_start = [
        {'Name': 'tag:AutoShutdown', "Values": ['AUTO']},
        {'Name': 'instance-state-name', "Values": ['stopped']}
    ]
    query_stop = [
        {'Name': 'tag:AutoShutdown', "Values": ['ON', 'AUTO']},
        {'Name': 'instance-state-name', "Values": ['running']}
    ]

    service = build(serviceName='calendar', version='v3', developerKey=API_KEY)
    events = service.events().list(calendarId=CALENDAR_ID).execute()
    holiday_list = []

    for item in events['items']:
        holiday_list.append(item['start']['date'])

    holiday_list.extend(company_holiday_list)

    # 発火元event名の確認
    try:
        # 自動起動の場合は土日祝日を除く平日のみ動作させる
        if '[Event sourceのARN(自動起動用ScheduleEvent)をコピー]' in event['resources']:
            if not str(datetime.date.today()) in holiday_list:
                client.start_instances(InstanceIds=get_instanceid(query_start))

        elif '[Event sourceのARN(自動停止用ScheduleEvent)をコピー]' in event['resources']:
            client.stop_instances(InstanceIds=get_instanceid(query_stop))

        elif '[Event sourceのARN(自動停止X分前告知用ScheduleEvent)をコピー]' in event['resources'] \
                and (not str(datetime.date.today()) in holiday_list):
            # 5分とか10分前にSlack通知するような処理をここに書くとより親切(自動停止爆死予防)

    except Exception as e:
        # エラー処理。Slackにスタックトレース投げるとか。

    print("SUCCESS: task succeeded")
    return


def get_instanceid(query):
    client = boto3.client('ec2')
    response = client.describe_instances(Filters=query)

    ec2_count = len(response['Reservations'])
    ec2_list = []

    if not ec2_count == 0:
        for i in range(0, ec2_count):
            ec2_list.append(response['Reservations'][i]['Instances'][0]['InstanceId'])
        return ec2_list
    else:
        print("SUCCESS: specified hosts is None")
        sys.exit()

zip -r ~/hoge.zip .
aws lambda update-function-code --function-name [Lambda作成時に設定したfunction名] --zip-file fileb://~/hoge.zip

일단 보충

[Event source ARN]을 각 환경으로 다시 작성하십시오. m (__) m
이하의 화상으로 말하는 ARN의 부분을 말과 코피페입니다.