Statistics
| Branch: | Tag: | Revision:

root / kamaki / utils.py @ c4ab2af9

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

    
66

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

    
75
        ether = address.get('mac', None)
76
        if ether:
77
            print '%s: %s' % ('ether'.rjust(margin + 8), ether)
78

    
79
        firewall = address.get('firewallProfile', None)
80
        if firewall:
81
            print '%s: %s' % ('firewall'.rjust(margin + 8), firewall)
82

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

    
87

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

    
113

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