Statistics
| Branch: | Tag: | Revision:

root / kamaki / cli / commands / __init__.py @ 5a673575

History | View | Annotate | Download (3.7 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
import logging
35

    
36
sendlog = logging.getLogger('clients.send')
37
recvlog = logging.getLogger('clients.recv')
38

    
39

    
40
class _command_init(object):
41

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

    
52
    def _safe_progress_bar(self, msg, arg='progress_bar'):
53
        """Try to get a progress bar, but do not raise errors"""
54
        try:
55
            progress_bar = self.arguments[arg]
56
            gen = progress_bar.get_generator(msg)
57
        except Exception:
58
            return (None, None)
59
        return (progress_bar, gen)
60

    
61
    def _safe_progress_bar_finish(self, progress_bar):
62
        try:
63
            progress_bar.finish()
64
        except Exception:
65
            pass
66

    
67
    def __getitem__(self, argterm):
68
        """
69
        :param argterm: (str) the name/label of an argument in self.arguments
70

71
        :returns: the value of the corresponding Argument (not the argument
72
            object)
73

74
        :raises KeyError: if argterm not in self.arguments of this object
75
        """
76
        return self.arguments[argterm].value
77

    
78
    def __setitem__(self, argterm, arg):
79
        """Install an argument as argterm
80
        If argterm points to another argument, the other argument is lost
81

82
        :param argterm: (str)
83

84
        :param arg: (Argument)
85
        """
86
        if not hasattr(self, 'arguments'):
87
            self.arguments = {}
88
        self.arguments[argterm] = arg
89

    
90
    def get_argument_object(self, argterm):
91
        """
92
        :param argterm: (str) the name/label of an argument in self.arguments
93

94
        :returns: the arument object
95

96
        :raises KeyError: if argterm not in self.arguments of this object
97
        """
98
        return self.arguments[argterm]
99

    
100
    def get_argument(self, argterm):
101
        """
102
        :param argterm: (str) the name/label of an argument in self.arguments
103

104
        :returns: the value of the arument object
105

106
        :raises KeyError: if argterm not in self.arguments of this object
107
        """
108
        return self[argterm]