diff --git a/client.py b/client.py index abbba82..51d56fc 100644 --- a/client.py +++ b/client.py @@ -33,9 +33,10 @@ class RPCClient(): pass if __name__ == "__main__": + import random client= RPCClient() print("Creating VM...") - name= "CentOS-8-Demo" + name= "CentOS-8-Demo-{}-{}".format(random.sample("abcdefghijklmn", 1)[0], random.sample(range(10),1)[0]) iso_src= "/home/coolguy/Downloads/CentOS-8.2.2004-x86_64-boot.iso" img_src= "/var/lib/libvirt/images/vol.img" diff --git a/virtpool.py b/virtpool.py index 1a4c3d9..b16d77c 100644 --- a/virtpool.py +++ b/virtpool.py @@ -7,6 +7,9 @@ import sys import cherrypy import libvirt import json +import threading +from multiprocessing.pool import ThreadPool +pool = ThreadPool(processes=1) import uuid from jinja2 import Environment, FileSystemLoader @@ -17,6 +20,8 @@ TEMPLATE_ENVIRONMENT = Environment( loader=FileSystemLoader(os.path.join(PATH, 'templates')), trim_blocks=False) +#TEMPLATE_ENVIRONMENT.globals['STATIC_PREFIX'] = '/templates/' + def render_template(template_filename, context): return TEMPLATE_ENVIRONMENT.get_template(template_filename).render(context) @@ -111,13 +116,89 @@ class Client(): 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}) + async_result = pool.apply_async(self.__create_vm, (host, xml)) + return async_result.get() return "No host available" + + def __create_vm(self, host, xml): + 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}) + + @cherrypy.expose + def gethost(self): + """ + It'll show relevant info about all available hosts + """ + + host_data= [] + + for i, each in enumerate(self.uris): + tmp= {} + c= self.connections[each] + nodeinfo = c.getInfo() + mem = nodeinfo[1] + cpus = nodeinfo[2] + usedmem = 0 + usedcpus = 0 + + tmp["sn"]= i+1 + tmp["hostname"]= c.getHostname() + tmp["status"]= "Alive" if c.isAlive() else "Dead" + tmp["uri"]= c.getURI() + tmp["cpu"]= {} + tmp["mem"]= {} + + domain_ids = c.listDomainsID() + for did in domain_ids: + dom = c.lookupByID(did) + state, maxmem, mem2, cpus2, cput = dom.info() + usedmem += maxmem/1024 + usedcpus += cpus2 + + tmp["cpu"]["total"]= cpus + tmp["cpu"]["free"]= cpus - usedcpus + tmp["mem"]["total"]= mem + tmp["mem"]["free"]= int(mem - usedmem) + + host_data.append(tmp) + + return render_template("host.html", {"host_data": host_data}) + + @cherrypy.expose + def getguest(self): + """ + It'll show host-wise available guests and detailed info about them + """ + + guest_data= [] + cnt=0 + for i, each in enumerate(self.uris): + c= self.connections[each] + for did in c.listDomainsID(): + dom= c.lookupByID(did) + tmp= { + "sn": cnt+1, + "guestname": dom.name(), + "hostname": c.getHostname(), + "id_": dom.ID(), + "mem": int(dom.info()[1]/1024), + "cpu": dom.info()[3], + "status": "Active" if dom.isActive() else "Inactive", + "uuid": dom.UUIDString() + } + + guest_data.append(tmp) + cnt+=1 + + return render_template("guest.html", {"guest_data": guest_data}) + + + def manage_host(self,host_id): + pass ROOT = Client() CONF = os.path.join(os.path.dirname(__file__), 'site.conf') @@ -130,7 +211,15 @@ if __name__ == '__main__': # cherrypy.process.plugins.Daemonizer(engine).subscribe() # cherrypy.process.plugins.DropPrivileges( # ENGINE, uid='cherrypy', gid='cherrypy').subscribe() - cherrypy.tree.mount(Client()) + cherrypy.tree.mount(Client(), + '/', + config={ + '/':{ + 'tools.staticdir.on': True, + 'tools.staticdir.dir': PATH+"/templates" + } + } + ) if hasattr(ENGINE, "signal_handler"): ENGINE.signal_handler.subscribe() if hasattr(ENGINE, "console_control_handler"):