Class StreamedYAMLFormatter

awscli/formatter.py:154–194

class StreamedYAMLFormatter(Formatter):
    def __init__(self, args, yaml_dumper=None):
        super(StreamedYAMLFormatter, self).__init__(args)
        self._yaml_dumper = yaml_dumper
        if yaml_dumper is None:
            self._yaml_dumper = YAMLDumper()

    def __call__(self, command_name, response, stream=None):
        if stream is None:
            stream = self._get_default_stream()
        compat.set_preferred_output_encoding(stream)
        response_stream = self._get_response_stream(response)
        for response in response_stream:
            try:
                # For YAML it is ambiguous as to whether the output from the
                # stream is N responses in 1 list or N lists each with 1
                # response. We go with the latter so we can reuse our YAML
                # dumper
                self._yaml_dumper.dump([response], stream)
            except OSError:
                # If the reading end of our stdout stream has closed the file
                # we can just exit.
                return
            finally:
                # flush is needed to avoid the "close failed in file object
                # destructor" in python2.x. See:
                # http://bugs.python.org/issue11380).
                self._flush_stream(stream)

    def _get_response_stream(self, response):
        if is_response_paginated(response):
            return compat.imap(
                self._get_transformed_response_for_output, response
            )
        else:
            output = self._get_transformed_response_for_output(response)
            if output == {}:
                # The operation did not have an output so return an empty list
                # as the stream so nothing gets printed out.
                return []
            return [output]


class TableFormatter(FullyBufferedFormatter):