import boto3

import config

client = boto3.client('dynamodb')

def update_table(write_capacity):
    """Update dynamo_table

    Args: write_capacity (int): new write capacity for all resources
    """
    return client.update_table(
        TableName=config.TABLE_NAME,
        ProvisionedThroughput={
            'ReadCapacityUnits': 5,
            'WriteCapacityUnits': write_capacity
        },
        GlobalSecondaryIndexUpdates=[
            {
                'Update': {
                    'IndexName': config.INDEX_NAME,
                    'ProvisionedThroughput': {
                        'ReadCapacityUnits': 5,
                        'WriteCapacityUnits': write_capacity
                    }
                },
            }
        ]
    )