summaryrefslogtreecommitdiffstats
path: root/testapi/testapi-client/testapiclient/utils/command.py
blob: c99a3f37a3e30519eba41d57b341bfcde997e5e2 (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
50
51
52
53
54
55
56
57
import abc
import logging

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

from testapiclient import utils
from testapiclient.utils import url_parse


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 filter_by_name(url, parsed_args):
        def query_url():
            return url_parse.query_join(url, name=parsed_args.name)

        return query_url() if parsed_args.name else url

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