batch_get_bnumbers.py

def batch_get_bnumbers(dynamo_client, bnumbers):
    keys = [
        { 'bnumber': { 'S': bnum } }
        for bnum in bnumbers
    ]

    response = dynamo_client.batch_get_item(
        RequestItems={
            'storage-migration-status' : {
                'Keys': keys
            }
        }
    )
    
    for row in response['Responses']['storage-migration-status']:
        yield deserialize_row(row)