lambda自動実行

autoinvoke.py

# usage: python3 hoge.py us-east-1 20190926

import boto3
import datetime
import json
import sys


REGION = sys.argv[1]
YYYYMMDD_UTC = datetime.datetime.strptime(sys.argv[2], '%Y%m%d')
DRY_RUN = False

DAILY_LAMBDA_FUNC_NAME = "XXXXX"
HOURLY_LAMBDA_FUNC_NAME = "XXXXX"

client = boto3.client('lambda', region_name=REGION)

for hours in range(1, 25):
  payload= {
    "time": datetime.datetime.isoformat(YYYYMMDD_UTC + datetime.timedelta(hours=hours)) + 'Z'
  }
  print(f'> LAMBDA: {HOURLY_LAMBDA_FUNC_NAME}, PAYLAOD: {payload}')
  if not DRY_RUN:
    res = client.invoke(
        FunctionName=HOURLY_LAMBDA_FUNC_NAME,
        Payload=json.dumps(payload)
    )
    print(res)


payload= {
  "time": datetime.datetime.isoformat(YYYYMMDD_UTC) + 'Z'
}
print(f'> LAMBDA: {DAILY_LAMBDA_FUNC_NAME}, PAYLAOD: {payload}')
if not DRY_RUN:
  res = client.invoke(
      FunctionName=DAILY_LAMBDA_FUNC_NAME,
      Payload=json.dumps(payload)
  )
  print(res)