Move OrderedDict import where it is actually used
[kamaki] / kamaki / utils.py
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_addresses(addresses, margin):
35     for address in addresses:
36         if address['id'] == 'public':
37             net = 'public'
38         else:
39             net = '%s/%s' % (address['id'], address['name'])
40         print '%s:' % net.rjust(margin + 4)
41
42         ether = address.get('mac', None)
43         if ether:
44             print '%s: %s' % ('ether'.rjust(margin + 8), ether)
45
46         firewall = address.get('firewallProfile', None)
47         if firewall:
48             print '%s: %s' % ('firewall'.rjust(margin + 8), firewall)
49
50         for ip in address.get('values', []):
51             key = 'inet' if ip['version'] == 4 else 'inet6'
52             print '%s: %s' % (key.rjust(margin + 8), ip['addr'])
53
54
55 def print_dict(d, exclude=()):
56     if not d:
57         return
58     margin = max(len(key) for key in d) + 1
59     
60     for key, val in sorted(d.items()):
61         if key in exclude:
62             continue
63         
64         if key == 'addresses':
65             print '%s:' % 'addresses'.rjust(margin)
66             print_addresses(val.get('values', []), margin)
67             continue
68         elif key == 'servers':
69             val = ', '.join(str(x) for x in val['values'])
70         elif isinstance(val, dict):
71             if val.keys() == ['values']:
72                 val = val['values']
73             print '%s:' % key.rjust(margin)
74             for key, val in val.items():
75                 print '%s: %s' % (key.rjust(margin + 4), val)
76             continue
77         
78         print '%s: %s' % (key.rjust(margin), val)
79
80
81 def print_items(items, title=('id', 'name')):
82     for item in items:
83         print ' '.join(str(item.pop(key)) for key in title if key in item)
84         if item:
85             print_dict(item)
86             print