virtui-manager 1.1.6__py3-none-any.whl → 1.4.0__py3-none-any.whl

vmanager/modals/host_stats.py

@@ -0,0 +1,199 @@
+"""
+Host Stats modals
+"""
+import logging
+import threading
+from textual.app import ComposeResult
+from textual.widgets import Static, Label
+from textual.reactive import reactive
+from textual.events import Click, Message
+from textual.worker import get_current_worker
+
+from ..libvirt_utils import get_host_resources, get_active_vm_allocation
+from ..utils import extract_server_name_from_uri
+
+class SingleHostStat(Static):
+    """
+    Displays stats for a single host.
+    """
+    server_name = reactive("")
+
+    class ServerLabelClicked(Message):
+        """Posted when the server label is clicked."""
+        def __init__(self, server_uri: str, server_name: str) -> None:
+            super().__init__()
+            self.server_uri = server_uri
+            self.server_name = server_name
+
+    DEFAULT_CSS = """
+    SingleHostStat {
+        layout: horizontal;
+        height: auto;
+        padding: 0 1;
+        background: $boost;
+        align-vertical: middle;
+    }
+    .stat-label {
+        color: $text;
+    }
+    """
+
+    def __init__(self, uri: str, name: str, vm_service, server_color: str = "white"):
+        super().__init__()
+        self.uri = uri
+        self.server_name = name
+        self.vm_service = vm_service
+        self.server_color = server_color
+        self.server_label = Label("", id=f"single_host_stat_label_{self.server_name.replace(' ', '_').replace('.', '_')}")
+        self.cpu_label = Label("", classes="stat-label")
+        self.mem_label = Label("", classes="stat-label")
+        self.host_res = None
+
+    def compose(self) -> ComposeResult:
+        self.server_label.styles.color = self.server_color
+        self.server_label.styles.text_style = "bold"
+        self.server_label.update(f"{self.server_name} ")
+        yield self.server_label
+        yield self.cpu_label
+        yield Label(" ")
+        yield self.mem_label
+
+    def on_click(self, event: Click) -> None:
+        """Called when the user clicks on a widget."""
+        if event.control.id == self.server_label.id:
+            self.post_message(self.ServerLabelClicked(self.uri, self.server_name))
+
+    def update_stats(self):
+        """Fetches and updates stats for this host."""
+        def _fetch_and_update():
+            try:
+                # Check cancellation before potentially expensive op
+                try:
+                    if get_current_worker().is_cancelled:
+                        return
+                except Exception:
+                    pass
+
+                conn = self.vm_service.connect(self.uri)
+                if not conn:
+                    if threading.current_thread() is threading.main_thread():
+                        self.cpu_label.update("Offline")
+                        self.mem_label.update("Offline")
+                    else:
+                        self.app.call_from_thread(self.cpu_label.update, "Offline")
+                        self.app.call_from_thread(self.mem_label.update, "Offline")
+                    return
+
+                if self.host_res is None:
+                    self.host_res = get_host_resources(conn)
+
+                current_alloc = get_active_vm_allocation(conn)
+                
+                # Check cancellation again after expensive op
+                try:
+                    if get_current_worker().is_cancelled:
+                        return
+                except Exception:
+                    pass
+
+                total_cpus = self.host_res.get('total_cpus', 1)
+                total_mem = self.host_res.get('available_memory', 1) # MB
+
+                used_cpus = current_alloc.get('active_allocated_vcpus', 0)
+                used_mem = current_alloc.get('active_allocated_memory', 0) # MB
+
+                cpu_pct = (used_cpus / total_cpus) * 100
+                mem_pct = (used_mem / total_mem) * 100
+                # Format memory string (GB if > 1024 MB)
+                def fmt_mem(mb):
+                    if mb >= 1024:
+                        return f"{mb/1024:.1f}G"
+                    return f"{mb}M"
+
+                # UI Updates need to be on main thread
+                def _update_ui():
+                    def get_status_bck(pct):
+                        if pct >= 90:
+                            return ("red")
+                        if pct >= 75:
+                            return ("orange")
+                        if pct >= 55:
+                            return ("yellow")
+                        return ("green")
+
+                    self.cpu_label.update(f"{used_cpus}/{total_cpus}CPU")
+                    self.cpu_label.styles.background = get_status_bck(cpu_pct)
+
+                    self.mem_label.update(f"{fmt_mem(used_mem)}/{fmt_mem(total_mem)}")
+                    self.mem_label.styles.background = get_status_bck(mem_pct)
+
+                if threading.current_thread() is threading.main_thread():
+                    _update_ui()
+                else:
+                    self.app.call_from_thread(_update_ui)
+
+            except Exception as e:
+                logging.error(f"Error updating host stats for {self.name}: {e}")
+                if threading.current_thread() is threading.main_thread():
+                    self.cpu_label.update("Err")
+                    self.mem_label.update("Err")
+                else:
+                    self.app.call_from_thread(self.cpu_label.update, "Err")
+                    self.app.call_from_thread(self.mem_label.update, "Err")
+
+        _fetch_and_update()
+
+class HostStats(Static):
+    """
+    Container for multiple SingleHostStat widgets.
+    """
+    DEFAULT_CSS = """
+    HostStats {
+        layout: grid;
+        grid-size: 3;
+        overflow-y: auto;
+        margin-bottom: 0;
+        margin-top: 0;
+        display: none;
+    }
+    """
+
+    def __init__(self, vm_service, get_server_color_callback):
+        super().__init__()
+        self.vm_service = vm_service
+        self.get_server_color = get_server_color_callback
+        self.active_hosts = {}
+
+    def update_hosts(self, active_uris, servers):
+        """
+        Reconciles the list of active hosts.
+        """
+        current_uris = set(active_uris)
+        existing_uris = set(self.active_hosts.keys())
+
+        # Remove stale
+        for uri in existing_uris - current_uris:
+            widget = self.active_hosts.pop(uri)
+            widget.remove()
+
+        # Add new hosts
+        for uri in current_uris - existing_uris:
+            name = self._get_server_name(uri, servers)
+            color = self.get_server_color(uri)
+            widget = SingleHostStat(uri, name, self.vm_service, color)
+            self.active_hosts[uri] = widget
+            self.mount(widget)
+            self.app.set_timer(0.5, widget.update_stats)
+
+    def _get_server_name(self, uri: str, servers) -> str:
+        """Helper to get server name from URI."""
+        if servers:
+            for s in servers:
+                if s['uri'] == uri:
+                    return s.get('name', extract_server_name_from_uri(uri))
+        return extract_server_name_from_uri(uri)
+
+    def refresh_stats(self):
+        """Triggers update on all children."""
+        for widget in self.active_hosts.values():
+            widget.update_stats()

vmanager/modals/howto_disk_modal.py

@@ -8,6 +8,7 @@ from textual.containers import Vertical, Horizontal, ScrollableContainer
 from textual.widgets import Button, Markdown
 from textual import on
 from .base_modals import BaseModal
+from ..constants import ButtonLabels
 
 class HowToDiskModal(BaseModal[None]):
     """A modal to display instructions for managing VM disks."""
@@ -25,7 +26,7 @@ class HowToDiskModal(BaseModal[None]):
             with ScrollableContainer(id="howto-disk-content"):
                 yield Markdown(content, id="howto-disk-markdown")
         with Horizontal(id="dialog-buttons"):
-            yield Button("Close", id="close-btn", variant="primary")
+            yield Button(ButtonLabels.CLOSE, id="close-btn", variant="primary")
 
     @on(Button.Pressed)
     def on_button_pressed(self, event: Button.Pressed) -> None:

vmanager/modals/howto_network_modal.py

@@ -7,6 +7,7 @@ from textual.app import ComposeResult
 from textual.containers import Vertical, Horizontal, ScrollableContainer
 from textual.widgets import Button, Markdown
 from textual import on
+from ..constants import ButtonLabels
 from .base_modals import BaseModal
 
 class HowToNetworkModal(BaseModal[None]):
@@ -25,7 +26,7 @@ class HowToNetworkModal(BaseModal[None]):
             with ScrollableContainer(id="howto-network-content"):
                 yield Markdown(content, id="howto-network-markdown")
         with Horizontal(id="dialog-buttons"):
-            yield Button("Close", id="close-btn", variant="primary")
+            yield Button(ButtonLabels.CLOSE, id="close-btn", variant="primary")
 
     @on(Button.Pressed)
     def on_button_pressed(self, event: Button.Pressed) -> None:

vmanager/modals/howto_overlay_modal.py

@@ -8,6 +8,7 @@ from textual.containers import Vertical, Horizontal, ScrollableContainer
 from textual.widgets import Button, Markdown
 from textual import on
 from .base_modals import BaseModal
+from ..constants import ButtonLabels
 
 class HowToOverlayModal(BaseModal[None]):
     """A modal to display instructions for disk overlays."""
@@ -25,7 +26,7 @@ class HowToOverlayModal(BaseModal[None]):
             with ScrollableContainer(id="howto-overlay-content"):
                 yield Markdown(content, id="howto-overlay-markdown")
         with Horizontal(id="dialog-buttons"):
-            yield Button("Close", id="close-btn", variant="primary")
+            yield Button(ButtonLabels.CLOSE, id="close-btn", variant="primary")
 
     @on(Button.Pressed)
     def on_button_pressed(self, event: Button.Pressed) -> None:

vmanager/modals/howto_ssh_modal.py

@@ -7,6 +7,7 @@ from textual.containers import Vertical, Horizontal, ScrollableContainer
 from textual.widgets import Button, Markdown
 from textual import on
 from .base_modals import BaseModal
+from ..constants import ButtonLabels
 
 class HowToSSHModal(BaseModal[None]):
     """A modal to display instructions for using an ssh-agent."""
@@ -24,7 +25,7 @@ class HowToSSHModal(BaseModal[None]):
             with ScrollableContainer(id="howto-ssh-content"):
                 yield Markdown(content, id="howto-ssh-markdown")
         with Horizontal(id="dialog-buttons"):
-            yield Button("Close", id="close-btn", variant="primary")
+            yield Button(ButtonLabels.CLOSE, id="close-btn", variant="primary")
 
     @on(Button.Pressed)
     def on_button_pressed(self, event: Button.Pressed) -> None:

vmanager/modals/howto_virtiofs_modal.py

@@ -8,6 +8,7 @@ from textual.containers import Vertical, Horizontal, ScrollableContainer
 from textual.widgets import Button, Markdown
 from textual import on
 from .base_modals import BaseModal
+from ..constants import ButtonLabels
 
 class HowToVirtIOFSModal(BaseModal[None]):
     """A modal to display instructions for using VirtIO-FS."""
@@ -25,7 +26,7 @@ class HowToVirtIOFSModal(BaseModal[None]):
             with ScrollableContainer(id="howto-virtiofs-content"):
                 yield Markdown(content, id="howto-virtiofs-markdown")
         with Horizontal(id="dialog-buttons"):
-            yield Button("Close", id="close-btn", variant="primary")
+            yield Button(ButtonLabels.CLOSE, id="close-btn", variant="primary")
 
     @on(Button.Pressed)
     def on_button_pressed(self, event: Button.Pressed) -> None:

vmanager/modals/input_modals.py

@@ -7,6 +7,7 @@ from textual.app import ComposeResult
 from textual.containers import Vertical, Horizontal
 from textual import on
 from .base_modals import BaseModal
+from ..constants import ButtonLabels, StaticText
 
 class InputModal(BaseModal[str | None]):
     """A generic modal for getting text input from the user."""
@@ -21,8 +22,8 @@ class InputModal(BaseModal[str | None]):
             yield Label(self.prompt)
             yield Input(value=self.initial_value, id="text-input", restrict=self.restrict)
             with Horizontal():
-                yield Button("OK", variant="primary", id="ok-btn")
-                yield Button("Cancel", variant="default", id="cancel-btn")
+                yield Button(ButtonLabels.OK, variant="primary", id="ok-btn")
+                yield Button(ButtonLabels.CANCEL, variant="default", id="cancel-btn")
 
     def on_button_pressed(self, event: Button.Pressed) -> None:
         if event.button.id == "ok-btn":
@@ -40,7 +41,7 @@ class AddInputDeviceModal(BaseModal[None]):
 
     def compose(self) -> ComposeResult:
         with Vertical(id="add-input-container"):
-            yield Label("Input Device")
+            yield Label(StaticText.INPUT_DEVICE)
             yield Select(
                 [(t, t) for t in self.available_types],
                 prompt="Input Type",
@@ -53,8 +54,8 @@ class AddInputDeviceModal(BaseModal[None]):
             )
             with Vertical():
                 with Horizontal():
-                    yield Button("Add", variant="primary", id="add-input", disabled=True)
-                    yield Button("Cancel", variant="default", id="cancel-input")
+                    yield Button(ButtonLabels.ADD, variant="primary", id="add-input", disabled=True)
+                    yield Button(ButtonLabels.CANCEL, variant="default", id="cancel-input")
 
     @on(Select.Changed)
     def on_select_changed(self) -> None:
@@ -82,26 +83,26 @@ class AddChannelModal(BaseModal[dict | None]):
 
     def compose(self) -> ComposeResult:
         with Vertical(id="add-channel-container"):
-            yield Label("Add Channel Device")
+            yield Label(StaticText.ADD_CHANNEL_DEVICE)
             yield Select(
                 [("unix", "unix"), ("virtio", "virtio"), ("spicevmc", "spicevmc")],
                 prompt="Channel Type",
                 id="channel-type-select",
                 value="unix"
             )
-            yield Label("Standard Target Names:")
+            yield Label(StaticText.STANDARD_TARGET_NAMES)
             yield Select(
                 [],
                 id="target-preset-select",
                 prompt="Select a standard target or type below",
                 value=Select.BLANK
             )
-            yield Label("Target Name:")
+            yield Label(StaticText.TARGET_NAME)
             yield Input(placeholder="Target Name (e.g. org.qemu.guest_agent.0)", id="target-name-input")
 
             with Horizontal():
-                yield Button("Add", variant="primary", id="add-channel-btn")
-                yield Button("Cancel", variant="default", id="cancel-channel-btn")
+                yield Button(ButtonLabels.ADD, variant="primary", id="add-channel-btn")
+                yield Button(ButtonLabels.CANCEL, variant="default", id="cancel-channel-btn")
 
     def on_mount(self) -> None:
         # Initialize presets for default type (unix)

vmanager/modals/log_modal.py

@@ -6,6 +6,7 @@ from textual.widgets import Button, Label, TextArea
 from textual.containers import Vertical, Horizontal
 
 from .base_modals import BaseModal
+from ..constants import ButtonLabels
 
 class LogModal(BaseModal[None]):
     """ Modal Screen to show Log"""
@@ -22,7 +23,7 @@ class LogModal(BaseModal[None]):
             text_area.load_text(self.log_content)
             yield text_area
         with Horizontal():
-            yield Button("Close", variant="default", id="cancel-btn", classes="Buttonpage")
+            yield Button(ButtonLabels.CLOSE, variant="default", id="cancel-btn", classes="Buttonpage")
 
     def on_mount(self) -> None:
         """Called when the modal is mounted."""

vmanager/modals/migration_modals.py

@@ -11,6 +11,7 @@ from textual.screen import ModalScreen
 from textual.widgets import Button, Static, Select, Checkbox, Label, ProgressBar
 from textual import on, work
 
+from ..constants import ErrorMessages, StaticText, ButtonLabels
 from ..vm_actions import check_server_migration_compatibility, check_vm_migration_compatibility
 from ..storage_manager import find_shared_storage_pools
 from ..utils import extract_server_name_from_uri
@@ -74,38 +75,39 @@ class MigrationModal(ModalScreen):
 
         with Vertical(id="migration-dialog",):
             with Vertical(id="migration-content-wrapper"):
-                yield Label(f"[{migration_type}] Migrate VMs: [b]{vm_names}[/b]")
-                yield Static("Select destination server:")
+                yield Label(StaticText.MIGRATE_VMS_TITLE.format(migration_type=migration_type, vm_names=vm_names))
+                yield Static(StaticText.SELECT_DESTINATION_SERVER)
                 yield Select(dest_servers, id="dest-server-select", prompt="Destination...", value=default_dest_uri, allow_blank=False)
 
-                yield Static("Migration Options:")
+                yield Static(StaticText.MIGRATION_OPTIONS)
                 with Horizontal(classes="checkbox-container"):
-                    yield Checkbox("Copy storage all", id="copy-storage-all", tooltip="Copy all disk files during migration", value=False)
-                    yield Checkbox("Unsafe migration", id="unsafe", tooltip="Perform unsafe migration (may lose data)", disabled=not self.is_live)
-                    yield Checkbox("Persistent migration", id="persistent", tooltip="Keep VM persistent on destination", value=True)
+                    yield Checkbox(StaticText.COPY_STORAGE_ALL, id="copy-storage-all", tooltip="Copy all disk files during migration", value=False)
+                    yield Checkbox(StaticText.UNSAFE_MIGRATION, id="unsafe", tooltip="Perform unsafe migration (may lose data)", disabled=not self.is_live)
+                    yield Checkbox(StaticText.PERSISTENT_MIGRATION, id="persistent", tooltip="Keep VM persistent on destination", value=True)
                 with Horizontal(classes="checkbox-container"):
-                    yield Checkbox("Compress data", id="compress", tooltip="Compress data during migration", disabled=not self.is_live)
-                    yield Checkbox("Tunnelled migration", id="tunnelled", tooltip="Tunnel migration data through libvirt daemon", disabled=not self.is_live)
-                    yield Checkbox("Custom migration", id="custom", tooltip="Use custom migration workflow", value=False)
-                yield Static("Compatibility Check Results / Migration Log:")
+                    yield Checkbox(StaticText.COMPRESS_DATA, id="compress", tooltip="Compress data during migration", disabled=not self.is_live)
+                    yield Checkbox(StaticText.TUNNELLED_MIGRATION, id="tunnelled", tooltip="Tunnel migration data through libvirt daemon", disabled=not self.is_live)
+                    yield Checkbox(StaticText.CUSTOM_MIGRATION, id="custom", tooltip="Use custom migration workflow", value=False)
+                yield Static(StaticText.COMPATIBILITY_CHECK_RESULTS)
                 yield ProgressBar(total=100, show_eta=False, id="migration-progress")
                 yield Static(id="results-log")
                 yield Grid(
                     ScrollableContainer(
-                        Static("[b]VMs [green]Ready[/] for Migration[/b]", classes="summary-title"),
+                        Static(StaticText.VMS_READY_FOR_MIGRATION, classes="summary-title"),
                         Static(id="can-migrate-list"),
                     ),
                     ScrollableContainer(
-                        Static("[b]VMs [red]Not[/] Ready for Migration[/b]", classes="summary-title"),
+                        Static(StaticText.VMS_NOT_READY_FOR_MIGRATION, classes="summary-title"),
                         Static(id="cannot-migrate-list"),
                     ),
                     id="migration-summary-grid"
                 )
 
             with Horizontal(classes="modal-buttons"):
-                yield Button("Check Compatibility", variant="primary", id="check", classes="Buttonpage")
-                yield Button("Start Migration", variant="success", id="start", disabled=True, classes="Buttonpage")
-                yield Button("Close", variant="default", id="close", disabled=False, classes="close-button")
+                yield Button(ButtonLabels.CHECK_COMPATIBILITY, variant="primary", id="check", classes="Buttonpage")
+                yield Button(ButtonLabels.START_MIGRATION, variant="success", id="start", disabled=True, classes="Buttonpage")
+                yield Button(ButtonLabels.CLOSE, variant="default", id="close", disabled=False, classes="close-button")
+
 
     def _lock_controls(self, lock: bool):
         self.query_one("#check").disabled = lock
@@ -385,7 +387,7 @@ class MigrationModal(ModalScreen):
 
         write_log("\n[bold]--- Migration process finished ---[/]")
         self.app.call_from_thread(lambda: setattr(progress_bar.styles, "display", "none"))
-        self.app.call_from_thread(self.app.refresh_vm_list)
+        self.app.call_from_thread(self.app.refresh_vm_list, force=True)
         self.app.call_from_thread(final_ui_state)
 
     @on(Checkbox.Changed, "#custom")
@@ -403,17 +405,17 @@ class MigrationModal(ModalScreen):
     def on_button_pressed(self, event: Button.Pressed):
         if event.button.id == "check":
             if not self.dest_conn:
-                self.app.show_error_message("Please select a destination server.")
+                self.app.show_error_message(ErrorMessages.SELECT_DESTINATION_SERVER)
                 return
             self._clear_log()
             self.run_compatibility_checks()
 
         elif event.button.id == "start":
             if not self.compatibility_checked:
-                self.app.show_error_message("Please run compatibility check first.")
+                self.app.show_error_message(ErrorMessages.RUN_COMPATIBILITY_CHECK_FIRST)
                 return
             if not self.checks_passed:
-                self.app.show_error_message("Cannot start migration due to compatibility errors.")
+                self.app.show_error_message(ErrorMessages.MIGRATION_COMPATIBILITY_ERRORS)
                 return
 
             self._clear_log()