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
|
#!/usr/bin/python -tt
# by skvidal
# gplv2+
# print out the ACTUAL freemem - not overcommitted value
import sys
import ansible.runner
import os
from optparse import OptionParser
def parse_args(args):
parser = OptionParser(version = "1.0")
parser.add_option('--host', default=[], action='append',
help="hosts to act on, defaults to ALL")
parser.add_option('--timeout', default=30, type='int',
help='set the wait timeout for func commands')
parser.add_option('--hosts-from-file', default=None, dest="hostfile",
help="read list of hosts from this file, if '-' read from stdin")
(opts, args) = parser.parse_args(args)
if opts.hostfile:
hosts = []
if opts.hostfile == '-':
hosts = sys.stdin.readlines()
else:
hosts = open(opts.hostfile, 'r').readlines()
for hn in hosts:
hn = hn.strip()
if hn.startswith('#'):
continue
hn = hn.replace('\n', '')
opts.host.append(hn)
if not opts.host:
opts.host = ["virthost*"]
return opts, args, parser
opts, args, parser = parse_args(sys.argv[1:])
hosts ='*'
if opts.host:
hosts = ';'.join(opts.host)
if os.geteuid() == 0:
login = 'root'
else:
login = os.getlogin()
# get results of nodeinfo phymemory
# "phymemory": "24018",
errors = []
# Setup some dictionaries for storing intermediate results
mem_per_host = {}
mem_used_in_vm = {}
cpu_per_host = {}
cpu_used_in_vm = {}
# We end up running ansible twice here. These are the common arguments.
# We'll use two different commands of the 'virt' ansible module.
ansible_args = dict(
pattern=hosts,
module_name='virt',
forks=25,
transport='paramiko',
timeout=10,
remote_user=login,
)
ans = ansible.runner.Runner(module_args='command=nodeinfo', **ansible_args)
res = ans.run()
for hn in sorted(res['contacted']):
if 'failed' in res['contacted'][hn] and res['contacted'][hn]['failed']:
continue
mem_per_host[hn] = int(res['contacted'][hn]['phymemory'])
cpu_per_host[hn] = int(res['contacted'][hn]['cpus'])
ans = ansible.runner.Runner(module_args='command=info', **ansible_args)
res = ans.run()
for hn in sorted(res['contacted']):
mem_used = 0
cpu_used = 0
if 'failed' in res['contacted'][hn] and res['contacted'][hn]['failed']:
errors.append('Failed to contact/run virt lookups on %s' % hn)
continue
for vm in sorted(res['contacted'][hn]):
info = res['contacted'][hn][vm]
if vm == 'Domain-0':
continue
elif type(info) != dict:
continue
mem_used += int(info.get('maxMem', 0))/1024
cpu_used += info.get('nrVirtCpu', 0)
mem_used_in_vm[hn] = mem_used
cpu_used_in_vm[hn] = cpu_used
for hn in sorted(mem_per_host):
freemem = mem_per_host[hn] - mem_used_in_vm[hn]
freecpu = cpu_per_host[hn] - cpu_used_in_vm[hn]
print '%s:\t%s/%s mem(unused/total)\t%s/%s cpus(unused/total)' % (
hn, freemem, mem_per_host[hn], freecpu, cpu_per_host[hn])
for err in errors:
print >> sys.stderr, err
|
