summaryrefslogtreecommitdiffstats
path: root/testapi/testapi-client/testapiclient/utils/command.py
blob: b9d1ce8aca2c98b80af42b857df0d7ee82fb058d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import abc
import logging

from cliff import command
from cliff import lister
from cliff import show
import six

from testapiclient import utils


class CommandMeta(abc.ABCMeta):

    def __new__(mcs, name, bases, cls_dict):
        if 'log' not in cls_dict:
            cls_dict['log'] = logging.getLogger(
                cls_dict['__module__'] + '.' + name)
        return super(CommandMeta, mcs).__new__(mcs, name, bases, cls_dict)


@six.add_metaclass(CommandMeta)
class Command(command.Command):
    def run(self, parsed_args):
        self.log.debug('run(%s)', parsed_args)
        return super(Command, self).run(parsed_args)


class Lister(Command, lister.Lister):
    @staticmethod
    def format_output(columns, data):
        return (columns,
                (utils.get_item_properties(s, columns) for s in data))


class ShowOne(Command, show.ShowOne):
    @staticmethod
    def format_output(body):
        return zip(*sorted(six.iteritems(body)))