summaryrefslogtreecommitdiffstats
path: root/testapi/testapi-client/testapiclient/command.py
blob: 62a8b8637bcbfad1557296147e8ff10d5c5e1c33 (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
from cliff import command
from testapiclient import url_parse


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 show(self, request, response):
        print ' '.join([request,
                        'success' if response.status_code < 300
                        else 'failed: {}'.format(response.text)])


class Lister(command.Command):

    @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

    def show(self, response):
        print response.json() if response.status_code < 300 else response.text


class ShowOne(command.Command):
    def show(self, response):
        print response.json() if response.status_code < 300 else response.text