virtui-manager 1.1.5__py3-none-any.whl → 1.3.0__py3-none-any.whl

vmanager/vm_queries.py

@@ -13,8 +13,9 @@ from .libvirt_utils import (
         _get_disabled_disks_elem,
         get_overlay_backing_path,
         _get_backing_chain_elem,
+        get_host_domain_capabilities,
         )
-#from utils import log_function_call
+from .constants import StatusText
 
 def _parse_domain_xml_by_hash(xml_hash: str, xml_content: str) -> ET.Element | None:
     """
@@ -108,18 +109,18 @@ def get_status(domain, state=None):
         try:
             state, _ = domain.state()
         except libvirt.libvirtError:
-            return 'Unknown'
+            return StatusText.UNKNOWN
 
     if state == libvirt.VIR_DOMAIN_RUNNING:
-        return 'Running'
+        return StatusText.RUNNING
     elif state == libvirt.VIR_DOMAIN_PAUSED:
-        return 'Paused'
+        return StatusText.PAUSED
     elif state == libvirt.VIR_DOMAIN_PMSUSPENDED:
-        return 'PMSuspended'
+        return StatusText.PMSUSPENDED
     elif state == libvirt.VIR_DOMAIN_BLOCKED:
-        return 'Blocked'
+        return StatusText.BLOCKED
     else:
-        return 'Stopped'
+        return StatusText.STOPPED
 
 @lru_cache(maxsize=16)
 def get_vm_description(domain):
@@ -565,6 +566,7 @@ def get_all_vm_overlay_usage(conn: libvirt.virConnect) -> dict[str, list[str]]:
     """
     Scans all VMs and returns a mapping of backing file path to a list of VM names
     that use it via an overlay (checked via metadata).
+    Optimized to fetch VM XMLs in parallel.
     """
     backing_to_vms_map = {}
     if not conn:
@@ -575,30 +577,43 @@ def get_all_vm_overlay_usage(conn: libvirt.virConnect) -> dict[str, list[str]]:
     except libvirt.libvirtError:
         return backing_to_vms_map
 
-    for domain in domains:
+    def process_domain_overlay_usage(domain):
+        """Helper to process a single domain for overlay usage."""
         try:
             _, root = _get_domain_root(domain)
             if root is not None:
                 # Check all disks to see if they are overlays in metadata
                 disks = get_vm_disks_info(conn, root)
                 vm_name = domain.name()
+                overlay_mappings = []
                 for disk in disks:
                     path = disk.get('path')
                     if path:
                         backing_path = get_overlay_backing_path(root, path)
                         if backing_path:
-                            if backing_path not in backing_to_vms_map:
-                                backing_to_vms_map[backing_path] = []
-                            if vm_name not in backing_to_vms_map[backing_path]:
-                                backing_to_vms_map[backing_path].append(vm_name)
+                            overlay_mappings.append((backing_path, vm_name))
+                return overlay_mappings
         except Exception:
-            continue
+            pass
+        return []
+
+    # Use ThreadPoolExecutor for parallel processing
+    with concurrent.futures.ThreadPoolExecutor(max_workers=20) as executor:
+        results = executor.map(process_domain_overlay_usage, domains)
+
+    for overlay_mappings in results:
+        for backing_path, vm_name in overlay_mappings:
+            if backing_path not in backing_to_vms_map:
+                backing_to_vms_map[backing_path] = []
+            if vm_name not in backing_to_vms_map[backing_path]:
+                backing_to_vms_map[backing_path].append(vm_name)
 
     return backing_to_vms_map
 
 def get_all_vm_nvram_usage(conn: libvirt.virConnect) -> dict[str, list[str]]:
     """
     Scans all VMs and returns a mapping of NVRAM file path to a list of VM names.
+    Optimized to fetch VM XMLs in parallel.
     """
     nvram_to_vms_map = {}
     if not conn:
@@ -609,7 +624,8 @@ def get_all_vm_nvram_usage(conn: libvirt.virConnect) -> dict[str, list[str]]:
     except libvirt.libvirtError:
         return nvram_to_vms_map
 
-    for domain in domains:
+    def process_domain_nvram_usage(domain):
+        """Helper to process a single domain for NVRAM usage."""
         try:
             _, root = _get_domain_root(domain)
             if root is not None:
@@ -617,13 +633,22 @@ def get_all_vm_nvram_usage(conn: libvirt.virConnect) -> dict[str, list[str]]:
                 if nvram_elem is not None:
                     nvram_path = nvram_elem.text
                     if nvram_path:
-                        vm_name = domain.name()
-                        if nvram_path not in nvram_to_vms_map:
-                            nvram_to_vms_map[nvram_path] = []
-                        if vm_name not in nvram_to_vms_map[nvram_path]:
-                            nvram_to_vms_map[nvram_path].append(vm_name)
+                        return nvram_path, domain.name()
         except Exception:
-            continue
+            pass
+        return None, None
+
+    # Use ThreadPoolExecutor for parallel processing
+    with concurrent.futures.ThreadPoolExecutor(max_workers=20) as executor:
+        results = executor.map(process_domain_nvram_usage, domains)
+
+    for nvram_path, vm_name in results:
+        if nvram_path and vm_name:
+            if nvram_path not in nvram_to_vms_map:
+                nvram_to_vms_map[nvram_path] = []
+            if vm_name not in nvram_to_vms_map[nvram_path]:
+                nvram_to_vms_map[nvram_path].append(vm_name)
+
     return nvram_to_vms_map
 
 
@@ -644,7 +669,9 @@ def get_supported_machine_types(conn, domain):
         arch = arch_elem.get('arch') if arch_elem is not None else 'x86_64' # default
 
         # Get capabilities
-        caps_xml = conn.getCapabilities()
+        caps_xml = get_host_domain_capabilities(conn)
+        if not caps_xml:
+            return []
         caps_root = ET.fromstring(caps_xml)
 
         # Find machines for that arch
@@ -1016,19 +1043,34 @@ def check_for_spice_vms(conn):
     """
     Checks if any VM uses Spice graphics.
     Returns a message if a Spice VM is found, otherwise None.
+    Optimized to fetch VM XMLs in parallel.
     """
     if not conn:
         return None
     try:
         all_domains = conn.listAllDomains(0) or []
-        for domain in all_domains:
+    except libvirt.libvirtError:
+        return None
+
+    def check_domain_for_spice(domain):
+        """Helper to check a single domain for Spice graphics."""
+        try:
             _, root = _get_domain_root(domain)
             if root is not None:
                 graphics_info = get_vm_graphics_info(root)
                 if graphics_info.get("type") == "spice":
-                    return "Some VMs use Spice graphics. 'Web Console' is only available for VNC."
-    except libvirt.libvirtError:
-        pass
+                    return True
+        except Exception:
+            pass
+        return False
+
+    # Use ThreadPoolExecutor for parallel processing
+    with concurrent.futures.ThreadPoolExecutor(max_workers=20) as executor:
+        results = executor.map(check_domain_for_spice, all_domains)
+        # Check if any domain uses Spice
+        if any(results):
+            return "Some VMs use Spice graphics. 'Web Console' is only available for VNC."
+
     return None
 
 def get_all_network_usage(conn: libvirt.virConnect) -> dict[str, list[str]]:

vmanager/vm_service.py

@@ -1297,7 +1297,7 @@ class VMService:
                 pass
         return None
 
-    def find_domains_by_uuids(self, active_uris: list[str], vm_uuids: list[str]) -> dict[str, libvirt.virDomain]:
+    def find_domains_by_uuids(self, active_uris: list[str], vm_uuids: list[str], check_validity: bool = True) -> dict[str, libvirt.virDomain]:
         """Finds and returns a dictionary of domain objects from a list of UUIDs."""
         self._update_target_uris(active_uris)
 
@@ -1327,11 +1327,14 @@ class VMService:
 
             valid = False
             if domain:
-                try:
-                    domain.info() # Check if domain is still valid
+                if check_validity:
+                    try:
+                        domain.info() # Check if domain is still valid
+                        valid = True
+                    except libvirt.libvirtError:
+                        valid = False
+                else:
                     valid = True
-                except libvirt.libvirtError:
-                    valid = False
 
             if valid:
                 found_domains[uuid] = domain

vmanager/vmanager.css

@@ -28,6 +28,7 @@ Select:focus {
 Toast.-inprogress {
     background: $primary;
     color: $text;
+    border-left: tall $primary;
 }
 
 /* --- Base Dialog & Container Styles --- */
@@ -635,3 +636,48 @@ NetworkListItem { height: auto; margin-bottom: 0; }
 .net-mode { width: 15%; }
 #net-active-check { width: 25%; }
 #net-autostart-check { width: 25%; }
+
+/* Host Dashboard Styles */
+#host-dashboard-dialog {
+    width: 80;
+    height: auto;
+    border: round $primary;
+    background: $surface;
+    padding: 1 2;
+}
+
+.info-section-dashboard {
+    background: $panel;
+    margin-bottom: 1;
+    border: round $primary;
+    padding: 0 1;
+    height: auto;
+}
+
+.section-title-dashboard {
+    text-style: bold;
+    background: $primary;
+    color: $text;
+    width: 100%;
+    margin-bottom: 1;
+}
+
+.host-info-grid {
+    grid-size-columns: 2;
+    grid-gutter: 1;
+    height: auto;
+}
+
+.usage-row {
+    height: 3;
+    align: center middle;
+}
+
+.usage-row Label {
+    width: auto;
+    margin-right: 2;
+}
+
+#mem-bar, #cpu-alloc-bar, #mem-alloc-bar {
+    width: 1fr;
+}