astakos.py 3.33 KB
Newer Older
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
# Copyright 2011-2012 GRNET S.A. All rights reserved.
#
# 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
from kamaki.cli.commands import _command_init, errors, _optional_json
37
from kamaki.cli.command_tree import CommandTree
38
from kamaki.cli.errors import CLIBaseUrlError
39
from kamaki.cli.utils import print_dict
40

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


45
class _user_init(_command_init):
46

47
    @errors.generic.all
48
    @errors.user.load
49
    def _run(self):
50
51
52
        if getattr(self, 'auth_base', False):
            self.client = self.auth_base
        else:
53
54
            token = self.config.get('user', 'token')\
                or self.config.get('astakos', 'token')\
55
                or self.config.get('global', 'token')
56
57
            base_url = self.config.get('user', 'url')\
                or self.config.get('astakos', 'url')
58
59
60
61
            if not base_url:
                raise CLIBaseUrlError(service='astakos')
            self.client = AstakosClient(base_url=base_url, token=token)

62
        self._set_log_params()
63
        self._update_max_threads()
64

65
    def main(self):
66
        self._run()
67

Stavros Sachtouris's avatar
Stavros Sachtouris committed
68

69
@command(user_cmds)
70
class user_authenticate(_user_init, _optional_json):
71
72
73
74
75
76
77
    """Authenticate a user
    Get user information (e.g. unique account name) from token
    Token should be set in settings:
    *  check if a token is set    /config get token
    *  permanently set a token    /config set token <token>
    Token can also be provided as a parameter
    """
78

79
80
81
82
    @staticmethod
    def _print_access(r):
        print_dict(r['access'])

83
    @errors.generic.all
84
    @errors.user.authenticate
85
86
    def _run(self, custom_token=None):
        super(self.__class__, self)._run()
87
        r = self.client.authenticate(custom_token)
88
        self._print(r, self._print_access)
89
90
91

    def main(self, custom_token=None):
        self._run(custom_token)