Statistics
| Branch: | Tag: | Revision:

root / kamaki / utils.py @ 6add54fd

History | View | Annotate | Download (3.2 kB)

1
# Copyright 2011 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
def print_dict(d, exclude=(), ident= 0):
35
    margin = max(max(len(key) for key in d) + 1, ident)
36
    for key, val in d.items():
37
        if key in exclude:
38
            continue
39
        print_str = '%s:' % unicode(key)
40
        if isinstance(val, dict):
41
            print(print_str.rjust(margin)+' {')
42
            print_dict(val, exclude = exclude, ident = margin + 6)
43
            print '}'.rjust(margin)
44
        elif isinstance(val, list):
45
            print(print_str.rjust(margin)+' [')
46
            print_list(val, exclude = exclude, ident = margin + 6)
47
            print ']'.rjust(margin)
48
        else:
49
            print print_str.rjust(margin)+' '+unicode(val)
50

    
51
def print_list(l, exclude=(), ident = 0):
52
    margin = max(max(len(item) for item in l) + 1, ident)
53
    for item in l:
54
        if item in exclude:
55
            continue
56
        if isinstance(item, dict):
57
            print('{'.rjust(margin))
58
            print_dict(item, exclude = exclude, ident = margin + 6)
59
            print '}'.rjust(margin)
60
        elif isinstance(item, list):
61
            print '['.rjust(margin)
62
            print_list(item, exclude = exclude, ident = margin + 6)
63
            print ']'.rjust(margin)
64
        else:
65
            print unicode(val).rjust(margin)
66

    
67
def print_items(items, title=('id', 'name')):
68
    for item in items:
69
        print ' '.join(unicode(item.pop(key)) for key in title if key in item)
70
        if item:
71
            print_dict(item)
72
            print
73

    
74
def format_size(size):
75
    units = ('B', 'K', 'M', 'G', 'T')
76
    size = float(size)
77
    for unit in units:
78
        if size <= 1024:
79
            break
80
        size /= 1024
81
    s = ('%.1f' % size).rstrip('.0')
82
    return s + unit
83

    
84
def dict_from_args(*args):
85
    return {args[i]:args[i+1] for i in range(0, len(args)/2)}
86