Handles responses from boto3 client methods that can or can't be paginated and yields the results.

client_handler.py

"""Boto3 client response handler."""
import boto3


class ClientHandler:
    """Boto3 client response handler."""
    __slots__ = ["client", "method", "key", "paginator"]

    def __init__(self, client, method, key):
        """Override."""
        self.client = client
        self.method = method
        self.key = key

        if client.can_paginate(method):
            self.paginator = self.client.get_paginator(self.method)

    def generate_paginated(self, **arguments):
        """Generate values from the paginated response."""
        for response in self.paginator.paginate(**arguments):
            for value in response[self.key]:
                yield value

    def generate_unpaginated(self, **arguments):
        """Generate values from the unpaginated response."""
        response = getattr(self.client, self.method)(**arguments)

        for value in response[self.key]:
            yield value

    def generate(self, **arguments):
        """Get a generated response."""
        return self.generate_paginated(**arguments) if self.paginator \
            else self.generate_unpaginated(**arguments)