summaryrefslogtreecommitdiffstats
path: root/moonclient/moonclient/subjects.py
blob: 8203cd9263e4861f4301fa2aff366702750e1ac4 (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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
# Copyright 2015 Open Platform for NFV Project, Inc. and its contributors
# This software is distributed under the terms and conditions of the 'Apache-2.0'
# license which can be found in the file 'LICENSE' in this package distribution
# or at 'http://www.apache.org/licenses/LICENSE-2.0'.

import logging

from cliff.lister import Lister
from cliff.command import Command
import getpass


class SubjectsList(Lister):
    """List all subjects."""

    log = logging.getLogger(__name__)

    def get_parser(self, prog_name):
        parser = super(SubjectsList, self).get_parser(prog_name)
        parser.add_argument(
            '--intraextension',
            metavar='<intraextension-uuid>',
            help='IntraExtension UUID',
        )
        return parser

    def take_action(self, parsed_args):
        if not parsed_args.intraextension:
            parsed_args.intraextension = self.app.intraextension
        data = self.app.get_url(self.app.url_prefix+"/intra_extensions/{}/subjects".format(parsed_args.intraextension),
                                authtoken=True)
        return (
            ("id", "name", "Keystone ID"),
            ((_uuid, data[_uuid]["name"], data[_uuid]["keystone_id"]) for _uuid in data)
        )


class SubjectsAdd(Command):
    """add a new subject."""

    log = logging.getLogger(__name__)

    def get_parser(self, prog_name):
        parser = super(SubjectsAdd, self).get_parser(prog_name)
        parser.add_argument(
            'subject_name',
            metavar='<subject-name>',
            help='Subject name',
        )
        parser.add_argument(
            '--intraextension',
            metavar='<intraextension-uuid>',
            help='IntraExtension UUID',
        )
        parser.add_argument(
            '--description',
            metavar='<description-str>',
            help='Subject description',
        )
        parser.add_argument(
            '--password',
            metavar='<password-str>',
            help='Password for subject (if not given, user will be prompted for one)',
        )
        parser.add_argument(
            '--email',
            metavar='<email-str>',
            help='Email for the user',
        )
        return parser

    def take_action(self, parsed_args):
        if not parsed_args.intraextension:
            parsed_args.intraextension = self.app.intraextension
        if not parsed_args.password:
            parsed_args.password = getpass.getpass("Password for user {}:".format(parsed_args.subject_name))
        data = self.app.get_url(self.app.url_prefix+"/intra_extensions/{}/subjects".format(parsed_args.intraextension),
                                post_data={
                                    "subject_name": parsed_args.subject_name,
                                    "subject_description": parsed_args.description,
                                    "subject_password": parsed_args.password,
                                    "subject_email": parsed_args.email
                                    },
                                authtoken=True)
        return (
            ("id", "name", "Keystone ID"),
            ((_uuid, data[_uuid]["name"], data[_uuid]["keystone_id"]) for _uuid in data)
        )


class SubjectsDelete(Command):
    """Delete a subject."""

    log = logging.getLogger(__name__)

    def get_parser(self, prog_name):
        parser = super(SubjectsDelete, self).get_parser(prog_name)
        parser.add_argument(
            'subject_id',
            metavar='<subject-uuid>',
            help='Subject UUID',
        )
        parser.add_argument(
            '--intraextension',
            metavar='<intraextension-uuid>',
            help='IntraExtension UUID',
        )
        return parser

    def take_action(self, parsed_args):
        if not parsed_args.intraextension:
            parsed_args.intraextension = self.app.intraextension
        self.app.get_url(self.app.url_prefix+"/intra_extensions/{}/subjects/{}".format(
            parsed_args.intraextension,
            parsed_args.subject_id
            ),
            method="DELETE",
            authtoken=True
        )