A function that handles pagination on a DynamoDB query operation, returning an iterator

dynamodb_query_iterator.py

import boto3
from boto3.dynamodb.conditions import Key

def query_dynamodb_table(table, params):
    """
    Perform a Query operation and handles pagination automatically,
    returning an iterator.
    This will work for Scan operations too.
    Tested on python 3.6
    """
    response = table.query(**params)

    for item in response['Items']:
        yield item

    while 'LastEvaluatedKey' in response:
        response = table.query(**{
            **params, 'ExclusiveStartKey': response['LastEvaluatedKey']
        })

        for item in response['Items']:
            yield item

table = boto3.resource('dynamodb').Table('ddb_table_name')
params = {'KeyConditionExpression': Key('PK').eq('my_partition_key')}

for item in query_dynamodb_table(table, params):
    print(item)