Statistics
| Branch: | Tag: | Revision:

root / image_creator / util.py @ 22a6d232

History | View | Annotate | Download (2.5 kB)

1
# Copyright 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.
33

    
34
import pbs
35
from clint.textui import puts, puts_err, colored, progress
36

    
37

    
38
def get_command(command):
39
    def find_sbin_command(command, exception):
40
        search_paths = ['/usr/local/sbin', '/usr/sbin', '/sbin']
41
        for fullpath in map(lambda x: "%s/%s" % (x, command), search_paths):
42
            if os.path.exists(fullpath) and os.access(fullpath, os.X_OK):
43
                return pbs.Command(fullpath)
44
        raise exception
45

    
46
    try:
47
        return pbs.__getattr__(command)
48
    except pbs.CommadNotFount as e:
49
        return find_sbin_command(command, e)
50

    
51

    
52
def error(msg):
53
    puts_err(colored.red("Error: %s\n" % msg))
54

    
55

    
56
def warn(msg):
57
    puts_err(colored.yellow("Warning: %s" % msg))
58

    
59

    
60
def success(msg):
61
    puts(colored.green(msg))
62

    
63

    
64
def progress_generator(label='', n=100):
65
    position = 0
66
    for i in progress.bar(range(n), label):
67
        if i < position:
68
            continue
69
        position = yield
70
    yield  # suppress the StopIteration exception
71

    
72

    
73
# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :