[アップデート]AWS Service Referenceに条件キーとリソース情報が追加されました

import requests
import json
import sys
import os

def get_services_list(use_cache=True):
    """AWSサービス一覧を取得する（キャッシュ対応）"""
    cache_dir = "aws_service_cache"
    cache_file = f"{cache_dir}/index.json"
    
    # キャッシュディレクトリがなければ作成
    if not os.path.exists(cache_dir):
        os.makedirs(cache_dir)
    
    # キャッシュがあり、使用する設定なら読み込む
    if use_cache and os.path.exists(cache_file):
        print(f"キャッシュからサービス一覧を読み込み中...")
        with open(cache_file, 'r') as f:
            return json.load(f)
    
    # キャッシュがないか使用しない設定ならダウンロード
    print(f"サービス一覧をダウンロード中...")
    try:
        response = requests.get("http://servicereference.us-east-1.amazonaws.com/")
        services = response.json()
        
        # キャッシュに保存
        with open(cache_file, 'w') as f:
            json.dump(services, f)
        
        return services
    except Exception as e:
        print(f"サービス一覧の取得に失敗しました: {e}")
        # キャッシュがあれば、エラー時はキャッシュを使用
        if os.path.exists(cache_file):
            print(f"キャッシュからサービス一覧を読み込み中...")
            with open(cache_file, 'r') as f:
                return json.load(f)
        return []

def get_service_data(service_name, service_url, use_cache=True):
    """サービス詳細を取得する（キャッシュ対応）"""
    cache_dir = "aws_service_cache"
    cache_file = f"{cache_dir}/{service_name}.json"
    
    # キャッシュがあり、使用する設定なら読み込む
    if use_cache and os.path.exists(cache_file):
        with open(cache_file, 'r') as f:
            return json.load(f)
    
    # キャッシュがないか使用しない設定ならダウンロード
    try:
        response = requests.get(service_url)
        service_data = response.json()
        
        # キャッシュに保存
        with open(cache_file, 'w') as f:
            json.dump(service_data, f)
        
        return service_data
    except Exception as e:
        print(f"  エラー: {e}")
        # キャッシュがあれば、エラー時はキャッシュを使用
        if os.path.exists(cache_file):
            with open(cache_file, 'r') as f:
                return json.load(f)
        return {}

def find_resource_actions(resource_type, use_cache=True):
    """指定されたリソースタイプを使用するAWSアクションを検索する"""
    print(f"{resource_type}リソースを使用するアクションを検索中...")
    
    # サービス一覧を取得
    services = get_services_list(use_cache)
    if not services:
        return []
    
    # 結果を保存するリスト
    found_actions = []
    
    # 各サービスを処理
    total = len(services)
    for i, service_item in enumerate(services, 1):
        service_name = service_item["service"]
        service_url = service_item["url"]
        
        print(f"処理中: {service_name} ({i}/{total})")
        
        # サービス詳細を取得
        service_data = get_service_data(service_name, service_url, use_cache)
        
        # Actionsキーがない場合はスキップ
        if "Actions" not in service_data:
            continue
            
        # 指定されたリソースを使用するアクションを検索
        for action in service_data["Actions"]:
            if "Resources" in action:
                for resource in action["Resources"]:
                    if resource.get("Name") == resource_type:
                        # 見つかったアクションを追加
                        found_actions.append({
                            "service": service_name,
                            "action": action["Name"],
                            "condition_keys": resource.get("ConditionKeys", [])
                        })
                        break
    
    return found_actions

def main():
    # コマンドライン引数からリソースタイプを取得
    if len(sys.argv) < 2:
        print("使用方法: python find_aws_actions.py <リソースタイプ> [--no-cache]")
        print("例: python find_aws_actions.py subnet")
        return
    
    resource_type = sys.argv[1]
    use_cache = "--no-cache" not in sys.argv
    
    # 指定されたリソースを使用するアクションを検索
    actions = find_resource_actions(resource_type, use_cache)
    
    # 結果を表示
    print(f"\n合計 {len(actions)} 個の {resource_type} リソースを使用するアクションが見つかりました")
    
    # サービス別に整理
    by_service = {}
    for item in actions:
        service = item["service"]
        if service not in by_service:
            by_service[service] = []
        by_service[service].append(item)
    
    # 結果を表示
    for service, service_actions in sorted(by_service.items()):
        print(f"\n## {service}")
        for action in sorted(service_actions, key=lambda x: x["action"]):
            print(f"- {action['action']}")
            if action["condition_keys"]:
                print(f"  条件キー: {', '.join(action['condition_keys'])}")
    
    # 結果をJSONファイルに保存
    output_file = f"{resource_type}_actions.json"
    with open(output_file, "w") as f:
        json.dump(actions, f, indent=2)
    print(f"\n結果は {output_file} に保存されました")

if __name__ == "__main__":
    main()