gerrit.onap Code Review - sdc/sdc-distribution-client.git/blob

   1 #  Licensed under the Apache License, Version 2.0 (the "License"); you may
   2 #  not use this file except in compliance with the License. You may obtain
   3 #  a copy of the License at
   4 #
   5 #       http://www.apache.org/licenses/LICENSE-2.0
   6 #
   7 #  Unless required by applicable law or agreed to in writing, software
   8 #  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
   9 #  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  10 #  License for the specific language governing permissions and limitations
  11 #  under the License.
  12
  13 """Output formatters using PyYAML.
  14 """
  15
  16 import yaml
  17
  18 from . import base
  19 from cliff import columns
  20
  21
  22 class YAMLFormatter(base.ListFormatter, base.SingleFormatter):
  23
  24     def add_argument_group(self, parser):
  25         pass
  26
  27     def emit_list(self, column_names, data, stdout, parsed_args):
  28         items = []
  29         for item in data:
  30             items.append(
  31                 {n: (i.machine_readable()
  32                      if isinstance(i, columns.FormattableColumn)
  33                      else i)
  34                  for n, i in zip(column_names, item)}
  35             )
  36         yaml.safe_dump(items, stream=stdout, default_flow_style=False)
  37
  38     def emit_one(self, column_names, data, stdout, parsed_args):
  39         for key, value in zip(column_names, data):
  40             dict_data = {
  41                 key: (value.machine_readable()
  42                       if isinstance(value, columns.FormattableColumn)
  43                       else value)
  44             }
  45             yaml.safe_dump(dict_data, stream=stdout, default_flow_style=False)