adjust paths in vm config when --restore-root is used

Author: richardstephens

libvirtnbdbackup/restore/files.py

@@ -31,16 +31,20 @@
 from libvirtnbdbackup.exceptions import RestoreError
 
 
-def restore(args: Namespace, vmConfig: str, virtClient: virt.client) -> None:
+def restore(
+    args: Namespace, vmConfig: str, virtClient: virt.client, restConfig: bytes
+) -> bytes:
     """Notice user if backed up vm had loader / nvram"""
     config = vmconfig.read(vmConfig)
     info = virtClient.getDomainInfo(config)
+    restored_files = {}
 
     for setting, val in info.items():
         f = lib.getLatest(args.input, f"*{os.path.basename(val)}*", -1)
         if args.restore_root is not None:
             _, _, val_as_relative = os.path.splitroot(val)
             val = os.path.join(args.restore_root, val_as_relative)
+            restored_files[setting] = os.path.abspath(val)
         if lib.exists(args, val):
             logging.info(
                 "File [%s]: for boot option [%s] already exists, skipping.",
@@ -53,6 +57,10 @@ def restore(args: Namespace, vmConfig: str, virtClient: virt.client) -> None:
             "Restoring configured file [%s] for boot option [%s]", val, setting
         )
         lib.copy(args, f[0], val)
+    if restConfig != b"" and args.adjust_config is True:
+        return vmconfig.apply_paths(restConfig, restored_files)
+    else:
+        return restConfig
 
 
 def verify(args: Namespace, dataFiles: List[str]) -> bool:

libvirtnbdbackup/restore/vmconfig.py

@@ -25,6 +25,8 @@
 from libvirtnbdbackup.virt import xml
 from libvirtnbdbackup.virt import disktype
 
+from typing import Dict
+
 
 def read(ConfigFile: str) -> str:
     """Read saved virtual machine config'"""
@@ -138,3 +140,11 @@ def restore(
         lib.copy(args, vmConfig, targetFile)
         logging.info("Copied original vm config to [%s]", targetFile)
         logging.info("Note: virtual machine config must be adjusted manually.")
+
+
+def apply_paths(config: bytes, restored_files: Dict[str, str]):
+    tree = xml.asTree(config)
+    os_config = tree.find("os")
+    for flag, val in restored_files.items():
+        os_config.find(flag).text = val
+    return xml.ElementTree.tostring(tree, encoding="utf8", method="xml")

virtnbdrestore

@@ -278,7 +278,7 @@ def main() -> None:
             logging.error("Disk restore failed: [%s]", errmsg)
             sys.exit(1)
 
-        files.restore(args, ConfigFile, virtClient)
+        restConfig = files.restore(args, ConfigFile, virtClient, restConfig)
         vmconfig.restore(args, ConfigFile, restConfig, args.config_file)
         virtClient.refreshPool(args.output)
         if args.define is True: