summaryrefslogtreecommitdiffstats
path: root/moonclient/moonclient/rules.py
blob: b1f7e42251fdd18162afcc93bac12f0785ba1682 (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
120
121
122
123
124
125
126
127
# 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
from cliff.show import ShowOne


class RulesList(ShowOne):
    """List all aggregation algorithms."""

    log = logging.getLogger(__name__)

    def get_parser(self, prog_name):
        parser = super(RulesList, 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("/v3/OS-MOON/intra_extensions/{}/sub_rules".format(
            parsed_args.intraextension),
            authtoken=True)
        if "sub_rules" not in data:
            raise Exception("Error in command {}: {}".format("RulesList", data))
        # TODO (dthom): a better view with a Lister
        return (
            ("sub_rules",),
            (data["sub_rules"],)
        )


class RuleAdd(ShowOne):
    """List the current aggregation algorithm."""

    log = logging.getLogger(__name__)

    def get_parser(self, prog_name):
        parser = super(RuleAdd, self).get_parser(prog_name)
        parser.add_argument(
            'relation',
            metavar='<relation-uuid>',
            help='Relation UUID',
        )
        parser.add_argument(
            'rule',
            metavar='<argument-list>',
            help='Rule list (example: admin,vm_admin,servers)',
        )
        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
        rule = parsed_args.rule.split(",")
        post = {
            "rule": rule,
            "relation": parsed_args.relation
        }
        data = self.app.get_url("/v3/OS-MOON/intra_extensions/{intraextension}/sub_rules".format(
            intraextension=parsed_args.intraextension),
            post_data=post,
            authtoken=True)
        if "sub_rules" not in data:
            raise Exception("Error in command {}: {}".format("RuleAdd", data))
        return (
            ("sub_rules",),
            (data["sub_rules"],)
        )


class RuleDelete(ShowOne):
    """Set the current aggregation algorithm."""

    log = logging.getLogger(__name__)

    def get_parser(self, prog_name):
        parser = super(RuleDelete, self).get_parser(prog_name)
        parser.add_argument(
            'relation',
            metavar='<relation-uuid>',
            help='Relation UUID',
        )
        parser.add_argument(
            'rule',
            metavar='<argument-list>',
            help='Rule list (example: admin,vm_admin,servers)',
        )
        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
        rule = "+".join(parsed_args.rule.split(","))
        data = self.app.get_url(
            "/v3/OS-MOON/intra_extensions/{intra_extensions_id}/sub_rules/{relation_name}/{rule}".format(
                intra_extensions_id=parsed_args.intraextension,
                relation_name=parsed_args.relation,
                rule=rule,
            ),
            method="DELETE",
            authtoken=True)
        if "sub_rules" not in data:
            raise Exception("Error in command {}: {}".format("RuleDelete", data))
        return (
            ("sub_rules",),
            (data["sub_rules"],)
        )