1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
# -*- coding: utf-8 -*-
from __future__ import print_function
class Prefix:
def __init__(self, prefix, output, oneline):
self.prefix = prefix
self.output = output
self.empty = oneline
def pprint(self, data):
self.output("%s%s" % (self.prefix, data))
if not self.empty:
self.prefix = ' ' * len(self.prefix)
self.empty = True
class Printer:
labels = {
'hostname': 'Hostname',
'arch': 'Architecture',
'os': 'Operating System',
'addresses': 'IPs',
'ports': 'Ports',
'vserver': 'VServer',
'services': 'Services',
'maintainers': 'Maintainers',
'description': 'Description'
}
ignore = ['vserver-host', 'groups']
def __init__(self, data, oneline, verbose, nospaces):
self.data = data
self.oneline = oneline
self.verbose = verbose
self.nospaces = nospaces
def _sort_with_list(self, list, sort):
def helper(value):
if sort is None:
return value
(key, full_key) = value
if key in sort:
return sort.index(key)
return "%d %s" % (len(sort), key)
return sorted(list, key=helper)
def print_addresses(self, value, pprint, prefix, filter_key):
self._print_group_by(value, pprint, prefix, filter_key,
lambda ip: 'vserver' not in ip and ip['interface'] or None,
lambda ip: '%s/%s' % (ip['address'], ip['netmask']))
def print_vserver(self, value, pprint, prefix, filter_key):
if value == 'guest':
pprint('guest running on %s' % self.data['vserver-host'])
else:
pprint(value)
def print_ports(self, value, pprint, prefix, filter_key):
def _print_port(port):
if port['proto'] in ['tcp6', 'udp6']:
return '(%s) [%s]:%s' % (port['proto'].replace('6', ''),
port['ip'], port['port'])
return '(%s) %s:%s' % (port['proto'], port['ip'], port['port'])
self._print_group_by(value, pprint, prefix, filter_key,
lambda port: 'process' in port and port['process'] or 'UNKNOWN',
_print_port,
sort=['sshd', 'nrpe', 'munin-node'])
def _print(self, value, pprint, prefix='', filter_key=None, sort=None):
try:
value = value.strip().splitlines()
except:
pass
if isinstance(value, dict):
self._print_dict(value, pprint, prefix, filter_key, sort)
elif isinstance(value, list):
for v in value:
pprint(v)
elif value is None:
if self.verbose:
pprint('')
else:
pprint(value)
def _print_dict(self, value, pprint, prefix, filter_key, sort):
keys = self._sort_with_list(
[(key, full_key) for key in value.keys()
for full_key in [self._get_full_key(prefix, key)]
if full_key not in self.ignore or
filter_key is not None and filter_key.startswith(full_key)],
sort)
maxlength = max(map(len, map(lambda (key, full_key): self._get_label(key, full_key), keys)))
if not self.nospaces and filter_key is None and prefix == '':
pprint()
for (key, full_key) in keys:
if filter_key is None or filter_key.startswith(full_key):
new_filter_key = filter_key
if full_key == filter_key:
new_filter_key = None
new_pprint = pprint
if filter_key is None:
label = "%s: " % self._get_label(key, full_key)
if prefix == '':
if self.nospaces:
label = label.rjust(maxlength+4)
else:
label = (label + " ").rjust(maxlength+6)
else:
label = label.ljust(maxlength+2)
p = Prefix(label, pprint, self.oneline)
new_pprint = p.pprint
try:
method = getattr(self, 'print_%s' % full_key.replace('.', '_'))
method(value[key], new_pprint, full_key, new_filter_key)
except AttributeError:
self._print(value[key], new_pprint, full_key, new_filter_key)
if not self.nospaces and filter_key is None and prefix == '':
pprint()
def _print_group_by(self, value, pprint, prefix, filter_key, group, format, sort=None):
d = {}
for e in value:
key = group(e)
if key is not None:
d.setdefault(key, []).append(format(e))
self._print(d, pprint, prefix, filter_key, sort)
def _get_full_key(self, prefix, key):
if prefix == '':
return key
return "%s.%s" % (prefix, key)
def _get_label(self, key, full_key):
if full_key in self.labels:
return self.labels[full_key]
return key
def info(self, key):
self._print(self.data, print, filter_key=key,
sort=['hostname', 'arch', 'os', 'addresses', 'ports'])
|