126 lines
3.4 KiB
Python
126 lines
3.4 KiB
Python
from config import *
|
|
from functions.fnet import *
|
|
import base64
|
|
import time
|
|
import lxc
|
|
|
|
def get_version():
|
|
vlxc = lxc.version
|
|
vlibvirt = str(libvirt.getVersion()/1000000) #1000000 * major + 1000 * minor + release
|
|
vhype = hype_version
|
|
return vlxc, vlibvirt, vhype
|
|
|
|
def get_vm_infos(vm_name):
|
|
dom = conn.lookupByName(vm_name)
|
|
vm_info = dom.guestInfo()
|
|
vm_infos=[]
|
|
vm_infos=[vm_info['os.id'],vm_info['os.pretty-name'],vm_info['os.kernel-release']]
|
|
return vm_infos
|
|
|
|
|
|
def set_memory(vm_name,memory_new):
|
|
dom = conn.lookupByName(vm_name)
|
|
dom.shutdown()
|
|
time.sleep(3)
|
|
alive=0
|
|
while alive < 5:
|
|
if dom.isActive():
|
|
time.sleep(3)
|
|
alive+=1
|
|
else:
|
|
alive=6
|
|
if dom.isActive():
|
|
dom.destroy()
|
|
dom.setMaxMemory(memory_new)
|
|
dom.setMemoryFlags(memory_new)
|
|
dom.create()
|
|
time.sleep(3)
|
|
alive=0
|
|
while alive < 3:
|
|
if dom.isActive():
|
|
alive=4
|
|
else:
|
|
time.sleep(3)
|
|
alive+=1
|
|
|
|
def set_vcpu(vm_name,vcpu_new):
|
|
dom = conn.lookupByName(vm_name)
|
|
dom.shutdown()
|
|
time.sleep(3)
|
|
alive=0
|
|
while alive < 5:
|
|
if dom.isActive():
|
|
time.sleep(3)
|
|
alive+=1
|
|
else:
|
|
alive=6
|
|
if dom.isActive():
|
|
dom.destroy()
|
|
dom.setVcpusFlags(vcpu_new,libvirt.VIR_DOMAIN_AFFECT_CONFIG | libvirt.VIR_DOMAIN_VCPU_MAXIMUM)
|
|
dom.setVcpusFlags(vcpu_new,libvirt.VIR_DOMAIN_AFFECT_CONFIG | libvirt.VIR_DOMAIN_VCPU_CURRENT)
|
|
dom.create()
|
|
time.sleep(3)
|
|
alive=0
|
|
while alive < 3:
|
|
if dom.isActive():
|
|
alive=4
|
|
else:
|
|
time.sleep(3)
|
|
alive+=1
|
|
|
|
def get_info_vm(vm_name):
|
|
dom = conn.lookupByName(vm_name)
|
|
state, maxmem, mem, cpus, cput = dom.info()
|
|
return state, maxmem, mem, cpus, cput
|
|
|
|
def attach_net(vm_name,net_name):
|
|
dom = conn.lookupByName(vm_name)
|
|
network_xml = f'<interface type="network"><source network="{net_name}"/><model type="virtio"/></interface>'
|
|
dom.attachDeviceFlags(network_xml, libvirt.VIR_DOMAIN_AFFECT_CONFIG | libvirt.VIR_DOMAIN_AFFECT_LIVE)
|
|
|
|
def detach_net(vm_name, net_name, net_mac,net_int):
|
|
dom = conn.lookupByName(vm_name)
|
|
# network_xml = f'<interface type="network"><mac address="{net_mac}"/><source network="{net_name}"/><model type="virtio"/></interface>'
|
|
network_xml = f'<interface type="bridge"><mac address="{net_mac}"/><source network="{net_name}" bridge="{net_int}"/></interface>'
|
|
print(network_xml)
|
|
dom.detachDeviceFlags(network_xml, libvirt.VIR_DOMAIN_AFFECT_CONFIG | libvirt.VIR_DOMAIN_AFFECT_LIVE)
|
|
|
|
def get_screenshot(vm_name):
|
|
dom = conn.lookupByName(vm_name)
|
|
stream = conn.newStream()
|
|
imageType = dom.screenshot(stream,0)
|
|
file = "tmp_screen_" + dom.name()
|
|
fileHandler = open(file, 'wb')
|
|
streamBytes = stream.recv(262120)
|
|
while streamBytes != b'':
|
|
fileHandler.write(streamBytes)
|
|
streamBytes = stream.recv(262120)
|
|
fileHandler.close()
|
|
stream.finish()
|
|
with open(file, "rb") as f:
|
|
data = base64.b64encode(f.read())
|
|
os.remove(file)
|
|
return data
|
|
|
|
def get_autostart(vm_name):
|
|
dom = conn.lookupByName(vm_name)
|
|
return dom.autostart()
|
|
|
|
def set_autostart(vm_name):
|
|
dom = conn.lookupByName(vm_name)
|
|
dom.setAutostart(1)
|
|
|
|
def unset_autostart(vm_name):
|
|
dom = conn.lookupByName(vm_name)
|
|
dom.setAutostart(0)
|
|
|
|
"""
|
|
def get_net_vm(vm_name):
|
|
netws, nu2 = get_virt_int()
|
|
vm_net_list=[]
|
|
for net in netws:
|
|
if vm_name in net[1][0]:
|
|
vm_net_list.append(net[0][0][0])
|
|
return vm_net_list
|
|
"""
|