summaryrefslogtreecommitdiffstats
path: root/testapi/testapi-client/testapiclient/utils/command.py
blob: 9614acf9d57e6236e40fef79169c90eaa8f76a86 (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
39
40
41
42
43
44
45
46
47
48
49
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 get_parser(self, prog_name):
        parser = super(Command, self).get_parser(prog_name)
        parser.add_argument('-u',
                            type=str,
                            help='Username for authentication')
        parser.add_argument('-p',
                            type=str,
                            help='Password for authentication')
        return parser

    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)))