vm cache: shorten vm description by stripping .xva suffix

With install tests the description becomes really long, use chars
sparingly.

Signed-off-by: Yann Dirson <[email protected]>

Author: ydirson

lib/common.py

@@ -1,6 +1,7 @@
 import getpass
 import inspect
 import logging
+import sys
 import time
 import traceback
 from enum import Enum
@@ -75,6 +76,20 @@ def safe_split(text, sep=','):
     """ A split function that returns an empty list if the input string is empty. """
     return text.split(sep) if len(text) > 0 else []
 
+def strip_prefix(string, prefix):
+    if sys.version_info >= (3, 9):
+        return string.removeprefix(prefix)
+    if string.startswith(prefix):
+        return string[len(prefix):]
+    return string
+
+def strip_suffix(string, prefix):
+    if sys.version_info >= (3, 9):
+        return string.removesuffix(prefix)
+    if string.endswith(prefix):
+        return string[:-1 - len(prefix)]
+    return string
+
 def setup_formatted_and_mounted_disk(host, sr_disk, fs_type, mountpoint):
     if fs_type == 'ext4':
         option_force = '-F'

lib/host.py

@@ -8,7 +8,7 @@
 
 import lib.commands as commands
 
-from lib.common import _param_get, safe_split, to_xapi_bool, wait_for, wait_for_not
+from lib.common import _param_get, safe_split, strip_suffix, to_xapi_bool, wait_for, wait_for_not
 from lib.common import prefix_object_name
 from lib.netutil import wrap_ip
 from lib.sr import SR
@@ -203,7 +203,7 @@ def xo_server_reconnect(self):
     def import_vm(self, uri, sr_uuid=None, use_cache=False):
         if use_cache:
             assert sr_uuid, "A SR UUID is necessary to use import cache"
-            cache_key = f"[Cache for {uri}]"
+            cache_key = f"[Cache for {strip_suffix(uri, '.xva')}]"
             # Look for an existing cache VM
             vm_uuids = safe_split(self.xe('vm-list', {'name-description': cache_key}, minimal=True), ',')