Statistics
| Branch: | Tag: | Revision:

root / kamaki / cli / commands / __init__.py @ 0d4a6d0a

History | View | Annotate | Download (4.4 kB)

1
# Copyright 2011-2012 GRNET S.A. All rights reserved.
2
#
3
# Redistribution and use in source and binary forms, with or
4
# without modification, are permitted provided that the following
5
# conditions are met:
6
#
7
#   1. Redistributions of source code must retain the above
8
#      copyright notice, this list of conditions and the following
9
#      disclaimer.
10
#
11
#   2. Redistributions in binary form must reproduce the above
12
#      copyright notice, this list of conditions and the following
13
#      disclaimer in the documentation and/or other materials
14
#      provided with the distribution.
15
#
16
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
17
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
20
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
24
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
26
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27
# POSSIBILITY OF SUCH DAMAGE.
28
#
29
# The views and conclusions contained in the software and
30
# documentation are those of the authors and should not be
31
# interpreted as representing official policies, either expressed
32
# or implied, of GRNET S.A.command
33

    
34
from kamaki.logger import get_logger
35

    
36
log = get_logger(__name__)
37

    
38

    
39
class _command_init(object):
40

    
41
    def __init__(self, arguments={}):
42
        if hasattr(self, 'arguments'):
43
            arguments.update(self.arguments)
44
        self.arguments = dict(arguments)
45
        try:
46
            self.config = self['config']
47
            #self.config = self.get_argument('config')
48
        except KeyError:
49
            pass
50

    
51
    def _set_log_params(self):
52
        try:
53
            self.client.LOG_TOKEN, self.client.LOG_DATA = (
54
                self['config'].get('global', 'log_token') == 'on',
55
                self['config'].get('global', 'log_data') == 'on')
56
        except Exception as e:
57
            log.warning('Failed to read custom log settings:'
58
                '%s\n defaults for token and data logging are off' % e)
59

    
60
    def _update_max_threads(self):
61
        try:
62
            max_threads = int(self['config'].get('global', 'max_threads'))
63
            assert max_threads > 0
64
            self.client.MAX_THREADS = max_threads
65
        except Exception as e:
66
            log.warning('Failed to read custom thread settings: '
67
                '%s, use default max threads (%s)' % (
68
                    e, self.client.MAX_THREADS))
69

    
70
    def _safe_progress_bar(self, msg, arg='progress_bar'):
71
        """Try to get a progress bar, but do not raise errors"""
72
        try:
73
            progress_bar = self.arguments[arg]
74
            gen = progress_bar.get_generator(msg)
75
        except Exception:
76
            return (None, None)
77
        return (progress_bar, gen)
78

    
79
    def _safe_progress_bar_finish(self, progress_bar):
80
        try:
81
            progress_bar.finish()
82
        except Exception:
83
            pass
84

    
85
    def __getitem__(self, argterm):
86
        """
87
        :param argterm: (str) the name/label of an argument in self.arguments
88

89
        :returns: the value of the corresponding Argument (not the argument
90
            object)
91

92
        :raises KeyError: if argterm not in self.arguments of this object
93
        """
94
        return self.arguments[argterm].value
95

    
96
    def __setitem__(self, argterm, arg):
97
        """Install an argument as argterm
98
        If argterm points to another argument, the other argument is lost
99

100
        :param argterm: (str)
101

102
        :param arg: (Argument)
103
        """
104
        if not hasattr(self, 'arguments'):
105
            self.arguments = {}
106
        self.arguments[argterm] = arg
107

    
108
    def get_argument_object(self, argterm):
109
        """
110
        :param argterm: (str) the name/label of an argument in self.arguments
111

112
        :returns: the arument object
113

114
        :raises KeyError: if argterm not in self.arguments of this object
115
        """
116
        return self.arguments[argterm]
117

    
118
    def get_argument(self, argterm):
119
        """
120
        :param argterm: (str) the name/label of an argument in self.arguments
121

122
        :returns: the value of the arument object
123

124
        :raises KeyError: if argterm not in self.arguments of this object
125
        """
126
        return self[argterm]