astakos.py 6.16 KB
Newer Older
1
# Copyright 2011-2013 GRNET S.A. All rights reserved.
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
#
# Redistribution and use in source and binary forms, with or
# without modification, are permitted provided that the following
# conditions are met:
#
#   1. Redistributions of source code must retain the above
#      copyright notice, this list of conditions and the following
#      disclaimer.
#
#   2. Redistributions in binary form must reproduce the above
#      copyright notice, this list of conditions and the following
#      disclaimer in the documentation and/or other materials
#      provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#
# The views and conclusions contained in the software and
# documentation are those of the authors and should not be
# interpreted as representing official policies, either expressed
# or implied, of GRNET S.A.command

Stavros Sachtouris's avatar
Stavros Sachtouris committed
34
from kamaki.cli import command
35
from kamaki.clients.astakos import AstakosClient
36
37
from kamaki.cli.commands import (
    _command_init, errors, _optional_json, addLogSettings)
38
from kamaki.cli.command_tree import CommandTree
39
40
from kamaki.cli.errors import CLIBaseUrlError, CLIError
from kamaki.cli.utils import print_dict, ask_user
41

42
43
user_cmds = CommandTree('user', 'Astakos API commands')
_commands = [user_cmds]
Stavros Sachtouris's avatar
Stavros Sachtouris committed
44
45


46
class _user_init(_command_init):
47

48
49
50
51
52
53
54
55
    def _write_main_token(self, token):
        tokens = self.config.get_cloud(self.cloud, 'token').split()
        if token in tokens:
            tokens.remove(token)
        tokens.insert(0, token)
        self.config.set_cloud(self.cloud, 'token', ' '.join(tokens))
        self.config.write()

56
    @errors.generic.all
57
    @errors.user.load
58
    @addLogSettings
59
    def _run(self):
60
61
62
63
        if getattr(self, 'cloud', False):
            base_url = self._custom_url('astakos')
            if base_url:
                token = self._custom_token('astakos')\
64
                    or self.config.get_cloud(self.cloud, 'token')
65
                token = token.split()[0] if ' ' in token else token
66
67
68
69
                self.client = AstakosClient(base_url=base_url, token=token)
                return
        else:
            self.cloud = 'default'
70
71
        if getattr(self, 'auth_base', False):
            self.client = self.auth_base
72
73
            return
        raise CLIBaseUrlError(service='astakos')
74

75
    def main(self):
76
        self._run()
77

Stavros Sachtouris's avatar
Stavros Sachtouris committed
78

79
@command(user_cmds)
80
class user_authenticate(_user_init, _optional_json):
81
82
83
    """Authenticate a user
    Get user information (e.g. unique account name) from token
    Token should be set in settings:
84
85
    *  check if a token is set    /config get cloud.default.token
    *  permanently set a token    /config set cloud.default.token <token>
86
    Token can also be provided as a parameter
87
    (In case of another named cloud, use its name instead of default)
88
    """
89

90
91
92
93
    @staticmethod
    def _print_access(r):
        print_dict(r['access'])

94
    @errors.generic.all
95
    @errors.user.authenticate
96
    def _run(self, custom_token=None):
97
98
99
        token_bu = self.client.token
        try:
            r = self.client.authenticate(custom_token)
100
101
102
            if (token_bu != self.client.token and
                    ask_user('Permanently save token as cloud.%s.token ?' % (
                        self.cloud))):
103
                self._write_main_token(self.client.token)
104
105
106
107
        except Exception:
            #recover old token
            self.client.token = token_bu
            raise
108
        self._print(r, self._print_access)
109
110

    def main(self, custom_token=None):
111
        super(self.__class__, self)._run()
112
        self._run(custom_token)
113
114
115
116


@command(user_cmds)
class user_list(_user_init, _optional_json):
Stavros Sachtouris's avatar
Stavros Sachtouris committed
117
    """List all authenticated users"""
118
119
120
121
122
123
124
125

    @errors.generic.all
    def _run(self, custom_token=None):
        self._print(self.client.list_users())

    def main(self):
        super(self.__class__, self)._run()
        self._run()
126
127
128


@command(user_cmds)
129
130
class user_get(_user_init, _optional_json):
    """Get session user"""
131
132
133

    @errors.generic.all
    def _run(self):
134
        self._print(self.client.user_info(), print_dict)
135
136
137
138

    def main(self):
        super(self.__class__, self)._run()
        self._run()
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165


@command(user_cmds)
class user_set(_user_init, _optional_json):
    """Set session user by id
    To enrich your options, authenticate more users:
    /user authenticate <other user token>
    To list authenticated users
    /user list
    To get the current session user
    /user get
    """

    @errors.generic.all
    def _run(self, uuid):
        for user in self.client.list_users():
            if user.get('id', None) in (uuid,):
                ntoken = user['auth_token']
                if ntoken == self.client.token:
                    print('%s (%s) is already the session user' % (
                        self.client.user_term('name'),
                        self.client.user_term('id')))
                    return
                self.client.token = user['auth_token']
                print('Session user set to %s (%s)' % (
                        self.client.user_term('name'),
                        self.client.user_term('id')))
166
167
168
                if ask_user('Permanently make %s the main user?' % (
                        self.client.user_term('name'))):
                    self._write_main_token(self.client.token)
169
170
171
172
173
174
175
176
177
                return
        raise CLIError(
            'User with UUID %s not authenticated in current session' % uuid,
            details=[
                'To authenticate a user', '  /user authenticate <t0k3n>'])

    def main(self, uuid):
        super(self.__class__, self)._run()
        self._run(uuid)