Source code for lago.plugins.output

#
# Copyright 2014 Red Hat, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
#
# Refer to the README and COPYING files for full details of the license
#

from __future__ import absolute_import
"""
About OutFormatPlugins

An OutFormatPlugin is used to format the output of the commands that extract
information from the perfixes, like status.
"""

import collections
import json
import yaml
from abc import (abstractmethod, ABCMeta)
import copy
from operator import itemgetter

from six import with_metaclass

from . import Plugin


[docs]class OutFormatPlugin(with_metaclass(ABCMeta, Plugin)): def __init__(self): pass
[docs] @abstractmethod def format(self, info_dict): """ Execute any actions given the arguments Args: info_dict (dict): information to reformat Returns: str: String representing the formatted info """ pass
[docs]class DefaultOutFormatPlugin(OutFormatPlugin): indent_unit = ' '
[docs] def format(self, info_obj, indent=''): formatted_lines = [] if isinstance(info_obj, list): if indent: formatted_lines.append('') for elem in info_obj: value_str = self.format(elem) formatted_lines.append(indent + value_str) elif isinstance(info_obj, collections.Mapping): for key in sorted(info_obj.keys()): value = info_obj[key] if isinstance(value, collections.Mapping): if not value: continue formatted_lines.append('%s[%s]:' % (indent, str(key))) value_str = self.format( info_obj=value, indent=indent + self.indent_unit, ) if value_str: formatted_lines.append(value_str) elif value not in (None, ''): formatted_lines.append( indent + str(key) + ': ' + self.format( info_obj=value, indent=indent + self.indent_unit, ) ) else: formatted_lines.append(str(info_obj)) return '\n'.join(formatted_lines)
[docs]class JSONOutFormatPlugin(OutFormatPlugin):
[docs] def format(self, info_dict): return json.dumps( info_dict, sort_keys=True, indent=4, )
[docs]class YAMLOutFormatPlugin(OutFormatPlugin):
[docs] def format(self, info_dict): return yaml.dump(info_dict, default_flow_style=False)
[docs]class FlatOutFormatPlugin(OutFormatPlugin):
[docs] def format(self, info_dict, delimiter='/'): """ This formatter will take a data structure that represent a tree and will print all the paths from the root to the leaves in our case it will print each value and the keys that needed to get to it, for example: vm0: net: lago memory: 1024 will be output as: vm0/net/lago vm0/memory/1024 Args: info_dict (dict): information to reformat delimiter (str): a delimiter for the path components Returns: str: String representing the formatted info """ def dfs(father, path, acc): if isinstance(father, list): for child in father: dfs(child, path, acc) elif isinstance(father, collections.Mapping): for child in sorted(father.items(), key=itemgetter(0)), : dfs(child, path, acc) elif isinstance(father, tuple): path = copy.copy(path) path.append(father[0]) dfs(father[1], path, acc) else: # join the last key with it's value path[-1] = '{}: {}'.format(path[-1], str(father)) acc.append(delimiter.join(path)) result = [] dfs(info_dict.get('Prefix') or info_dict, [], result) return '\n'.join(result)