Statistics
| Branch: | Tag: | Revision:

root / kamaki / utils.py @ 486f00d3

History | View | Annotate | Download (4.1 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
try:
35
    from collections import OrderDict
36
except ImportError:
37
    class OrderedDict(dict):
38
        """An ordered dict implementation for Python versions prior to 2.7"""
39
    
40
        def __init__(self):
41
            dict.__init__(self)
42
            self._keys = []
43
    
44
        def __delitem__(self, key):
45
            dict.__delitem__(self, key)
46
            self._keys.remove(key)
47
    
48
        def __iter__(self):
49
            return iter(self._keys)
50
    
51
        def __repr__(self):
52
            return repr(self.items())
53
    
54
        def __setitem__(self, key, value):
55
            if key not in self:
56
                self._keys.append(key)
57
            dict.__setitem__(self, key, value)
58
    
59
        def keys(self):
60
            return self._keys
61
    
62
        def iteritems(self):
63
            for key in self._keys:
64
                yield key, self[key]
65
    
66
        def items(self):
67
            return list(self.iteritems())
68

    
69

    
70
def print_addresses(addresses, margin):
71
    for address in addresses:
72
        if address['id'] == 'public':
73
            net = 'public'
74
        else:
75
            net = '%s/%s' % (address['id'], address['name'])
76
        print '%s:' % net.rjust(margin + 4)
77

    
78
        ether = address.get('mac', None)
79
        if ether:
80
            print '%s: %s' % ('ether'.rjust(margin + 8), ether)
81

    
82
        firewall = address.get('firewallProfile', None)
83
        if firewall:
84
            print '%s: %s' % ('firewall'.rjust(margin + 8), firewall)
85

    
86
        for ip in address.get('values', []):
87
            key = 'inet' if ip['version'] == 4 else 'inet6'
88
            print '%s: %s' % (key.rjust(margin + 8), ip['addr'])
89

    
90

    
91
def print_dict(d, exclude=()):
92
    if not d:
93
        return
94
    margin = max(len(key) for key in d) + 1
95
    
96
    for key, val in sorted(d.items()):
97
        if key in exclude:
98
            continue
99
        
100
        if key == 'addresses':
101
            print '%s:' % 'addresses'.rjust(margin)
102
            print_addresses(val.get('values', []), margin)
103
            continue
104
        elif key == 'servers':
105
            val = ', '.join(str(x) for x in val['values'])
106
        elif isinstance(val, dict):
107
            if val.keys() == ['values']:
108
                val = val['values']
109
            print '%s:' % key.rjust(margin)
110
            for key, val in val.items():
111
                print '%s: %s' % (key.rjust(margin + 4), val)
112
            continue
113
        
114
        print '%s: %s' % (key.rjust(margin), val)
115

    
116

    
117
def print_items(items, title=('id', 'name')):
118
    for item in items:
119
        print ' '.join(str(item.pop(key)) for key in title if key in item)
120
        if item:
121
            print_dict(item)
122
            print