IaC Generatorで使用されるリソーススキャンをboto3から使ってAWSリソース一覧をCSVで出力してみた

import time
import boto3
import json
import polars as pl
from functools import wraps
import time

def wait_until_complete(timeout=3600, interval=10):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            start_time = time.time()
            while True:
                response = func(*args, **kwargs)
                elapsed_time = int(time.time() - start_time)

                if response["Status"] == "COMPLETE":
                    print(f"{elapsed_time=} [sec] finished!! (status={response['Status']})")
                    return response

                if timeout and elapsed_time >= timeout:
                    raise TimeoutError(f"Timeout after {timeout} seconds")

                print(f"{elapsed_time} [sec] waiting... (status={response['Status']})")
                time.sleep(interval)
        return wrapper
    return decorator

def main():

    client = boto3.client("cloudformation")

    # スキャン開始
    response = client.start_resource_scan()
    ResourceScanId = response["ResourceScanId"]
    print(f"{ResourceScanId=}")

    @wait_until_complete()
    def check_scan_status(client, ResourceScanId):
        return client.describe_resource_scan(ResourceScanId=ResourceScanId)
    response = check_scan_status(client=client, ResourceScanId=ResourceScanId)

    # リソース取得
    data = []
    paginator = client.get_paginator("list_resource_scan_resources")
    for response in paginator.paginate(ResourceScanId=ResourceScanId):
        resources = response.get("Resources", [])
        resources = [{**i, "ResourceIdentifier": json.dumps(i["ResourceIdentifier"])} for i in resources]
        data.extend(resources)

    # csv出力
    df = pl.DataFrame(data=data)
    df.write_csv("resource_scan.csv")

if __name__ == "__main__":
    main()

ResourceType,ResourceIdentifier,ManagedByStack
AWS::Athena::WorkGroup,"{""Name"": ""primary""}",false
AWS::CloudFront::CachePolicy,"{""Id"": ""xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx""}",false
AWS::CloudFront::CachePolicy,"{""Id"": ""xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx""}",false
AWS::CloudFront::CachePolicy,"{""Id"": ""xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx""}",false
AWS::CloudFront::CachePolicy,"{""Id"": ""xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx""}",false
AWS::CloudFront::CachePolicy,"{""Id"": ""xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx""}",false
AWS::CloudFront::CachePolicy,"{""Id"": ""xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx""}",false
...