67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
import urllib
|
|
import urllib2
|
|
import ssl
|
|
import json
|
|
from VM import VM
|
|
|
|
|
|
def get_auth(host, username, password):
|
|
postdata = urllib.urlencode({'username': username, 'password': password})
|
|
url = "https://" + host + "/api2/json/access/ticket"
|
|
req = urllib2.Request(url, postdata)
|
|
response = urllib2.urlopen(req, context=ssl._create_unverified_context())
|
|
response = json.load(response)
|
|
return response['data']['CSRFPreventionToken'], response['data']['ticket']
|
|
|
|
|
|
class PVEApi(object):
|
|
|
|
def __init__(self, host, username, password):
|
|
self.host = host
|
|
self.url = "https://" + self.host + "/api2/json"
|
|
self.csrftoken = get_auth(self.host, username, password)[0]
|
|
self.ticket = get_auth(self.host, username, password)[1]
|
|
|
|
def get_request(self, url):
|
|
req = urllib2.Request(url)
|
|
req.add_header('Cookie', 'PVEAuthCookie=' + self.ticket)
|
|
return req
|
|
|
|
def get_node_list(self):
|
|
url = self.url + '/nodes'
|
|
response = json.load(urllib2.urlopen(self.get_request(url), context=ssl._create_unverified_context()))
|
|
resources = response['data']
|
|
nodes = list()
|
|
for node in resources:
|
|
if type(node) is dict:
|
|
nodes.append(node[u'node'])
|
|
return nodes
|
|
|
|
def get_cluster_vmlist(self):
|
|
vmlist = list()
|
|
for node in self.get_node_list():
|
|
url = self.url + '/nodes/' + node + '/qemu'
|
|
response = json.load(urllib2.urlopen(self.get_request(url), context=ssl._create_unverified_context()))
|
|
resources = response['data']
|
|
for machine in resources:
|
|
name = machine[u'name']
|
|
status = machine[u'status']
|
|
uptime = machine[u'uptime']
|
|
diskread = machine[u'diskread']
|
|
diskwrite = machine[u'diskwrite']
|
|
memusage = machine[u'mem']
|
|
pid = machine[u'pid']
|
|
vmid = machine[u'vmid']
|
|
netin = machine[u'netin']
|
|
netout = machine[u'netout']
|
|
cpus = machine[u'cpus']
|
|
template = machine[u'template']
|
|
disk = machine[u'disk']
|
|
cpuusage = machine[u'cpu']
|
|
maxdisk = machine[u'maxdisk']
|
|
maxmem = machine[u'maxmem']
|
|
vm = VM(node, name, status, uptime, diskread, diskwrite, memusage, pid, vmid, netin, netout, cpus,
|
|
template, disk, cpuusage, maxdisk, maxmem)
|
|
vmlist.append(vm)
|
|
return vmlist
|