Infrajam, the simplest assistant for engineering docs

def generate_infra_diagram():

    with Diagram("Infra Diagram", show=False) as diag:
        dns = Route53("DNS")
        cdn = CloudFront("CDN")
        storage = S3("Persistent Storage")
        metadata_db = Dynamodb("Image Metadata")
        lambda_func = Lambda("On-the-fly Manipulations")
        api_gateway = APIGateway("API Gateway")

        dns >> cdn >> storage
        cdn >> api_gateway >> lambda_func
        lambda_func >> metadata_db

        diag.metadata = {
            "AmazonRoute53": {
                "count": 1,
                "hosted_zones": 1,
                "queries_per_month": 0  # Placeholder, adjust as needed
            },
            "AmazonCloudFront": {
                "count": 1,
                "data_transfer_gb_per_month": 0  # Placeholder, adjust as needed
            },
            "AmazonS3": {
                "count": 1,
                "storage_gb": 0  # Placeholder, adjust as needed
            },
            "AmazonDynamoDB": {
                "count": 1,
                "read_capacity_units": 0,  # Placeholder, adjust as needed
                "write_capacity_units": 0,  # Placeholder, adjust as needed
                "storage_gb": 0  # Placeholder, adjust as needed
            },