Lambdaでスポットリクエストのキャンセルをしてみた

import boto3
import json
import os
from botocore.exceptions import ClientError

def lambda_handler(event, context):
    # 環境変数からリージョンを取得、設定されていない場合はデフォルト値を使用
    region = os.environ.get('AWS_REGION', 'ap-northeast-1')

    # EC2クライアントの初期化
    ec2_client = boto3.client('ec2', region_name=region)

    # 環境変数からスポットリクエストのNameタグを取得
    spot_request_name = os.environ.get('SPOT_REQUEST_NAME')

    if not spot_request_name:
        return {
            'statusCode': 400,
            'body': '環境変数 SPOT_REQUEST_NAME が設定されていません。'
        }

    try:
        # 指定されたNameタグを持つスポットリクエストを検索
        spot_requests = ec2_client.describe_spot_instance_requests(
            Filters=[
                {'Name': 'tag:Name', 'Values': [spot_request_name]},
                {'Name': 'state', 'Values': ['active', 'open']}
            ]
        )

        if not spot_requests['SpotInstanceRequests']:
            return {
                'statusCode': 404,
                'body': '指定されたNameタグを持つアクティブなスポットリクエストが見つかりませんでした。'
            }

        # 複数のスポットリクエストに対応するため、スポットリクエストIDとインスタンスIDを取得
        spot_request_ids = []
        instance_ids = []
        for request in spot_requests['SpotInstanceRequests']:
            spot_request_ids.append(request['SpotInstanceRequestId'])
            if 'InstanceId' in request:
                instance_ids.append(request['InstanceId'])

        # スポットリクエストのキャンセル
        cancel_response = ec2_client.cancel_spot_instance_requests(
            SpotInstanceRequestIds=spot_request_ids
        )

        # インスタンスの終了
        if instance_ids:
            terminate_response = ec2_client.terminate_instances(
                InstanceIds=instance_ids
            )

        # キャンセルと終了の成功を確認
        cancelled_ids = [req['SpotInstanceRequestId'] for req in cancel_response['CancelledSpotInstanceRequests']]
        terminated_ids = instance_ids if instance_ids else []

        return {
            'statusCode': 200,
            'body': json.dumps({
                'CancelledSpotRequests': cancelled_ids,
                'TerminatedInstances': terminated_ids
            })
        }

    except ClientError as e:
        # エラーハンドリング
        error_message = e.response['Error']['Message']
        return {
            'statusCode': 500,
            'body': f'エラーが発生しました: {error_message}'
        }