lambda modify aurora serverless acu

acu_master.py

import json
import boto3

client = boto3.client('rds')

def modify_acu(cluster_id, min_acu, max_acu):
    resp = client.modify_db_cluster(
        DBClusterIdentifier=cluster_id,
        ApplyImmediately=True,
        ServerlessV2ScalingConfiguration={
            'MinCapacity': min_acu,
            'MaxCapacity': max_acu
        }
    )
    return resp

def lambda_handler(event, context):
    cluster_id = event.get('cluster_id', None)
    min_acu = event.get('min_acu', 0)
    max_acu = event.get('max_acu', 0)
    if cluster_id is None or min_acu == 0 or max_acu == 0:
        return  {
            'statusCode': 401,
            'body': json.dumps('Parameter error.')
        }
    
    resp = modify_acu(cluster_id, min_acu, max_acu)
    return {
        'statusCode': 200,
        'body': json.dumps('Cluster updated')
    }