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
|
#!/usr/bin/env python
import libvirt
import yaml
import multiprocessing
import click
import sys
SETTING = "/root/cobbler/settings"
power_action_map = {
"on": "create",
"off": "destroy",
"status": "state"
}
def get_virt_host(setting_file):
with open(setting_file) as fd:
try:
settings = yaml.load(fd)
return settings['server']
except Exception:
raise RuntimeError("Can't get server ip from %s" % SETTING)
def get_libvit_connection(user, passwd):
# def request_cred(credentials, user_data):
# for credential in credentials:
# if credential[0] == libvirt.VIR_CRED_AUTHNAME:
# credential[4] = user
# elif credential[0] == libvirt.VIR_CRED_PASSPHRASE:
# credential[4] = passwd
# return 0
# auth = [[libvirt.VIR_CRED_AUTHNAME, libvirt.VIR_CRED_PASSPHRASE],
# request_cred, None]
server = get_virt_host(SETTING)
conn = libvirt.open('qemu+tcp://%s/system' % server)
return conn
def libvirt_function(domain, action, rc):
function = getattr(domain, power_action_map.get(action))
state, reason = domain.state()
if function.__name__ == "create":
if state == libvirt.VIR_DOMAIN_RUNNING:
rc.value = 1
else:
rc.value = function()
elif function.__name__ == "destroy":
if state == libvirt.VIR_DOMAIN_SHUTOFF:
rc.value = 1
else:
rc.value = function()
elif function.__name__ == "state":
rc.value = state
def power_action(action, hostname, user, passwd):
conn = get_libvit_connection(user, passwd)
domain = conn.lookupByName(hostname)
rc = multiprocessing.Value('i')
p = multiprocessing.Process(target=libvirt_function,
args=(domain, action, rc,))
p.start()
p.join()
print rc.value
return rc.value
@click.command()
@click.option("--action")
@click.option("--hostname")
@click.option("--user")
@click.option("--passwd")
def cli(action, hostname, user, passwd):
power_action(action, hostname, user, passwd)
def no_cli():
opt = {}
for line in sys.stdin.readlines():
try:
line = line.strip()
name, value = line.split("=")
opt.update({name: value})
except Exception:
continue
try:
if opt["action"] and opt["hostname"]:
power_action(opt["action"], opt["hostname"],
opt["user"], opt["passwd"])
else:
raise RuntimeError("Invalid argument, \
action: {0}, hostname: {1}, user: {2}, passwd: {3}".format(
opt.get("action", None), opt.get("hostname", None),
opt.get("user", None), opt.get("action", None)))
def main():
if len(sys.argv) > 1:
cli()
else:
no_cli()
if __name__ == '__main__':
main()
|
