forked from Bitmessage/virtpool
144 lines
4.1 KiB
Python
144 lines
4.1 KiB
Python
#!/usr/bin/python3
|
|
"""Agent for setting up accounts for sysdeploy clients"""
|
|
# pylint: disable=subprocess-run-check,too-few-public-methods
|
|
import os
|
|
import sys
|
|
|
|
import cherrypy
|
|
import libvirt
|
|
import json
|
|
|
|
import uuid
|
|
from jinja2 import Environment, FileSystemLoader
|
|
PATH = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
TEMPLATE_ENVIRONMENT = Environment(
|
|
autoescape=False,
|
|
loader=FileSystemLoader(os.path.join(PATH, 'templates')),
|
|
trim_blocks=False)
|
|
|
|
def render_template(template_filename, context):
|
|
return TEMPLATE_ENVIRONMENT.get_template(template_filename).render(context)
|
|
|
|
|
|
class Client():
|
|
"""Main manager class"""
|
|
uris = [
|
|
#'qemu+ssh://pool@reserve.homedevops/system',
|
|
#'qemu+ssh://root@check2.homedevops/system',
|
|
#'qemu+ssh://root@check3.homedevops/system'
|
|
'qemu:///system'
|
|
]
|
|
|
|
def __init__(self):
|
|
self.status = {}
|
|
self.connections = {}
|
|
for i in self.uris:
|
|
self.connections[i] = libvirt.open(i)
|
|
|
|
@cherrypy.expose
|
|
def getavailable(self):
|
|
out = ""
|
|
for n, c in self.connections.items():
|
|
# cmap = c.getCPUMap()
|
|
# out += "CPUs: {}".format(str(cmap[0]))
|
|
# out += "Available: {}".format(str(cmap[1]))
|
|
nodeinfo = c.getInfo()
|
|
mem = nodeinfo[1]
|
|
cpus = nodeinfo[2]
|
|
usedmem = 0
|
|
usedcpus = 0
|
|
|
|
domain_ids = c.listDomainsID()
|
|
for did in domain_ids:
|
|
dom = c.lookupByID(did)
|
|
state, maxmem, mem2, cpus, cput = dom.info()
|
|
usedmem += maxmem/1024
|
|
usedcpus += cpus
|
|
|
|
out += "{}: {}/{} CPUs free,{}MiB/{}MiB memory free<br/>\n".format(
|
|
n,
|
|
cpus - usedcpus, cpus,
|
|
str((mem - usedmem)),
|
|
str(mem))
|
|
return out
|
|
|
|
@cherrypy.expose
|
|
def create(
|
|
self,
|
|
name,
|
|
iso_source,
|
|
img_source,
|
|
domain_type="kvm",
|
|
memory_unit="MiB",
|
|
memory=1024,
|
|
current_memory= 1024,
|
|
vcpu=2,
|
|
arch="x86_64",
|
|
machine="pc-q35-4.2",
|
|
offset="utc",
|
|
emulator="/usr/bin/qemu-system-x86_64"
|
|
):
|
|
|
|
""" create a VM with given specs
|
|
boot it and return the ID for future ref
|
|
"""
|
|
|
|
#TODO: Check host resources(cpus, mem, etc.) before creating a VM
|
|
|
|
param= {
|
|
"name": name,
|
|
"uuid": str(uuid.uuid4()),
|
|
"iso_source": iso_source,
|
|
"img_source": img_source,
|
|
"type": domain_type,
|
|
"memory_unit": memory_unit,
|
|
"memory": memory,
|
|
"current_memory": current_memory,
|
|
"vcpu":vcpu,
|
|
"arch": arch,
|
|
"machine": machine,
|
|
"offset": offset,
|
|
"emulator": emulator,
|
|
}
|
|
|
|
context= {
|
|
'args': param
|
|
}
|
|
|
|
xml= render_template("config.tmpl.xml", context)
|
|
#print(xml)
|
|
|
|
host= self.connections[self.uris[0]]
|
|
if host:
|
|
try:
|
|
vm= host.createXML(xml)
|
|
return json.dumps({"uuid":vm.UUIDString(), "id":vm.ID(), "status":1})
|
|
except Exception as e:
|
|
return json.dumps({"err":str(e), "status":0})
|
|
return "No host available"
|
|
|
|
|
|
ROOT = Client()
|
|
CONF = os.path.join(os.path.dirname(__file__), 'site.conf')
|
|
|
|
if __name__ == '__main__':
|
|
# cherrypy.config.update(CONF)
|
|
cherrypy.server.socket_host = '127.0.0.1'
|
|
cherrypy.server.socket_port = 8080
|
|
ENGINE = cherrypy.engine
|
|
# cherrypy.process.plugins.Daemonizer(engine).subscribe()
|
|
# cherrypy.process.plugins.DropPrivileges(
|
|
# ENGINE, uid='cherrypy', gid='cherrypy').subscribe()
|
|
cherrypy.tree.mount(Client())
|
|
if hasattr(ENGINE, "signal_handler"):
|
|
ENGINE.signal_handler.subscribe()
|
|
if hasattr(ENGINE, "console_control_handler"):
|
|
ENGINE.console_control_handler.subscribe()
|
|
try:
|
|
ENGINE.start()
|
|
except Exception: # pylint: disable=broad-except
|
|
sys.exit(1)
|
|
else:
|
|
ENGINE.block()
|