sheppy/athq-vm-management
1
0
Fork 0
mirror of https://github.com/FAUSheppy/athq-vm-management synced 2025-12-06 05:41:35 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: various improvments

Browse Source
This commit is contained in:
Johnathen Sheppard
2024-08-04 15:20:19 +00:00
parent 6e8cb97512
commit 2bb888c661
1 changed files with 40 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
snapshot-backups.py
View File

@@ -2,10 +2,17 @@
import libvirt import libvirt
import os import os
import time
import sys import sys
import subprocess import subprocess
import datetime
import pwd
import grp
BASE_DIR = "/home/backup-atlantis-array/" BASE_DIR = "/home/backup-atlantis-array/"
BACKUP_USER_NAME = "backup-atlantis-array"
BACKUP_USER = pwd.getpwnam(BACKUP_USER_NAME).pw_uid
BACKUP_GROUP = grp.getgrnam(BACKUP_USER_NAME).gr_gid
def list_running_vms(conn): def list_running_vms(conn):
running_vms = [] running_vms = []
@@ -18,7 +25,7 @@ def get_default_network_xml(conn):
network = conn.networkLookupByName('default') network = conn.networkLookupByName('default')
return network.XMLDesc() return network.XMLDesc()
if __name__ == "__main"__: if __name__ == "__main__":
# connect to libvirt daemon # # connect to libvirt daemon #
conn = libvirt.open('qemu:///system') conn = libvirt.open('qemu:///system')
@@ -27,52 +34,70 @@ if __name__ == "__main"__:
sys.exit(1) sys.exit(1)
# build date str # # build date str #
date_str = datetime.datetime.now().stftime("%y-%m-%d") date_str = datetime.datetime.now().strftime("%y-%m-%d")
# get running vms # # get running vms #
running_vms = list_running_vms(conn) running_vms = list_running_vms(conn)
# dump network # # dump network #
default_network_xml = get_default_network_xml(conn) default_network_xml = get_default_network_xml(conn)
with open(os.path.join(BASE_DIR, "network-{}.xml".format(date_str)), "w") as f: network_dir = os.path.join(BASE_DIR, "network")
os.makedirs(network_dir, exist_ok=True)
network_path = os.path.join(network_dir, "network-{}.xml".format(date_str))
with open(network_path, "w") as f:
f.write(default_network_xml) f.write(default_network_xml)
os.chown(network_dir, BACKUP_USER, BACKUP_GROUP)
os.chown(network_path, BACKUP_USER, BACKUP_GROUP)
# debug output # # debug output #
print("Doing:") print("Doing:", [vm.name() for vm in running_vms])
print([ " " + vm.name() + "\n" for vm in running_vms])
for vm in running_vms: for vm in running_vms:
# shut down VM # # shut down VM #
print("Next:", vm.name()) print("Next:", vm.name())
if not vm.name() == "signal":
continue
vm.shutdown() vm.shutdown()
grace_time = 1
while vm.state()[0] != libvirt.VIR_DOMAIN_SHUTOFF:
if grace_time > 120:
print("Exeeded maximum shutdown wait for {}".format(vm.name()), file=sys.stderr)
sys.exit(1)
print("\rWaiting for '{}' to shutdown ({}/120s)".format(vm.name(), grace_time))
grace_time += 1
time.sleep(1)
# create the backup dir for domain # create the backup dir for domain
safe_dir_xml = os.path.join(BASE_DIR, vm.name()) safe_dir_xml = os.path.join(BASE_DIR, "{}-xml".format(vm.name()))
safe_dir_img = os.path.join(BASE_DIR, vm.name()) safe_dir_img = os.path.join(BASE_DIR, "{}-img".format(vm.name()))
os.makedirs(safe_dir_xml, exists_ok=True) os.makedirs(safe_dir_xml, exist_ok=True)
os.makedirs(safe_dir_img, exists_ok=True) os.makedirs(safe_dir_img, exist_ok=True)
os.chown(safe_dir_xml, BACKUP_USER) os.chown(safe_dir_xml, BACKUP_USER, BACKUP_GROUP)
os.chown(safe_dir_img, BACKUP_USER) os.chown(safe_dir_img, BACKUP_USER, BACKUP_GROUP)
# take xml dump # # take xml dump #
path_xml = os.path.join(safe_dir_xml, "{}-{}.xml".format(vm.name(), date_str)) path_xml = os.path.join(safe_dir_xml, "{}-{}.xml".format(vm.name(), date_str))
with open(path_xml, "w") as f: with open(path_xml, "w") as f:
f.write(vm.XMLDesc()) f.write(vm.XMLDesc())
os.chown(path_xml, BACKUP_USER) os.chown(path_xml, BACKUP_USER, BACKUP_GROUP)
# take img # # take img #
image_src_path = "/var/lib/libvirt/images/{}".format(vm.name()) image_src_path = "/var/lib/libvirt/images/{}".format(vm.name())
path_img = os.path.join(safe_dir_img, "{}-{}.img".format(vm.name(), date_str)) path_img = os.path.join(safe_dir_img, "{}-{}.img".format(vm.name(), date_str))
p = subprocess.run(["virt-sparsify", image_src_path, "--compress", path_img]) p = subprocess.run(["virt-sparsify", image_src_path, "--compress", path_img])
if p.return_code != 0: if p.returncode != 0:
print("ERROR: virt-sparsify failed for {}".format(vm.name()), file=sys.stderr) print("ERROR: virt-sparsify failed for {}".format(vm.name()), file=sys.stderr)
vm.start() vm.start()
sys.exit(1) sys.exit(1)
os.chown(path_img, BACKUP_USER) os.chown(path_img, BACKUP_USER, BACKUP_GROUP)
# start vm # # start vm #
vm.start() vm.create()
# wait for atlantis-array to collect the backup # wait for atlantis-array to collect the backup
i = 0 i = 0