get_composite_keys

get_composite_keys.py

def get_composite_keys(table_name: str) -> list:
    if table_name in table_composite_keys:
        return table_composite_keys[table_name]

    composite_keys = list()
    desc = dynamodb_client.describe_table(TableName=table_name)
    key_schema = desc['Table']['KeySchema']
    for k in key_schema:
        if '#' in k['AttributeName']:
            composite_keys.append(k['AttributeName'])

    table_composite_keys[table_name] = composite_keys

    return composite_keys