# # Copyright (C) 2006-2008 Red Hat, Inc. # Copyright (C) 2006 Daniel P. Berrange # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, # MA 02110-1301 USA. # import logging import re import gtk import virtManager.uihelpers as uihelpers from virtManager.connection import vmmConnection from virtManager.baseclass import vmmGObjectUI from virtManager.delete import vmmDeleteDialog from virtManager.graphwidgets import CellRendererSparkline from virtManager import util as util # fields in the tree model data set ROW_HANDLE = 0 ROW_NAME = 1 ROW_MARKUP = 2 ROW_STATUS = 3 ROW_STATUS_ICON = 4 ROW_KEY = 5 ROW_HINT = 6 ROW_IS_CONN = 7 ROW_IS_CONN_CONNECTED = 8 ROW_IS_VM = 9 ROW_IS_VM_RUNNING = 10 ROW_COLOR = 11 ROW_INSPECTION_OS_ICON = 12 # Columns in the tree view COL_NAME = 0 COL_GUEST_CPU = 1 COL_HOST_CPU = 2 COL_DISK = 3 COL_NETWORK = 4 rcstring = """ style "toolbar-style" { #GtkToolbar::button_relief = GTK_RELIEF_NONE #GtkToolbar::shadow_type = GTK_SHADOW_NONE GtkToolbar::internal_padding = 2 } style "treeview-style" { GtkTreeView::indent_expanders = 0 } class "GtkToolbar" style "toolbar-style" class "GtkTreeView" style "treeview-style" """ gtk.rc_parse_string(rcstring) class vmmManager(vmmGObjectUI): def __init__(self): vmmGObjectUI.__init__(self, "vmm-manager.glade", "vmm-manager") self.delete_dialog = None self.ignore_pause = False # Mapping of VM UUID -> tree model rows to # allow O(1) access instead of O(n) self.rows = {} w, h = self.config.get_manager_window_size() self.topwin.set_default_size(w or 550, h or 550) self.prev_position = None self.vmmenu = gtk.Menu() self.vmmenushutdown = gtk.Menu() self.vmmenu_items = {} self.vmmenushutdown_items = {} self.connmenu = gtk.Menu() self.connmenu_items = {} # There seem to be ref counting issues with calling # list.get_column, so avoid it self.diskcol = None self.netcol = None self.guestcpucol = None self.hostcpucol = None self.window.signal_autoconnect({ "on_menu_view_guest_cpu_usage_activate": (self.toggle_stats_visible, COL_GUEST_CPU), "on_menu_view_host_cpu_usage_activate": (self.toggle_stats_visible, COL_HOST_CPU), "on_menu_view_disk_io_activate" : (self.toggle_stats_visible, COL_DISK), "on_menu_view_network_traffic_activate": (self.toggle_stats_visible, COL_NETWORK), "on_vm_manager_delete_event": self.close, "on_vmm_manager_configure_event": self.window_resized, "on_menu_file_add_connection_activate": self.new_conn, "on_menu_file_quit_activate": self.exit_app, "on_menu_file_close_activate": self.close, "on_vmm_close_clicked": self.close, "on_vm_open_clicked": self.show_vm, "on_vm_run_clicked": self.start_vm, "on_vm_new_clicked": self.new_vm, "on_vm_shutdown_clicked": self.poweroff_vm, "on_vm_pause_clicked": self.pause_vm_button, "on_menu_edit_details_activate": self.show_vm, "on_menu_edit_delete_activate": self.do_delete, "on_menu_host_details_activate": self.show_host, "on_vm_list_row_activated": self.show_vm, "on_vm_list_button_press_event": self.popup_vm_menu_button, "on_vm_list_key_press_event": self.popup_vm_menu_key, "on_menu_edit_preferences_activate": self.show_preferences, "on_menu_help_about_activate": self.show_about, "on_menu_help_activate": self.show_help, }) self.init_vmlist() self.init_stats() self.init_toolbar() self.init_context_menus() # XXX: Help docs useless/out of date self.widget("menu_help").hide() self.vm_selected() self.widget("vm-list").get_selection().connect("changed", self.vm_selected) self.max_disk_rate = 10.0 self.max_net_rate = 10.0 # Initialize stat polling columns based on global polling # preferences (we want signal handlers for this) for typ, init_val in [ (COL_DISK, self.config.get_stats_enable_disk_poll()), (COL_NETWORK, self.config.get_stats_enable_net_poll())]: self.enable_polling(None, None, init_val, typ) # Select first list entry vmlist = self.widget("vm-list") if len(vmlist.get_model()) != 0: vmlist.get_selection().select_iter( vmlist.get_model().get_iter_first()) # Queue up the default connection detector self.idle_emit("add-default-conn") ################## # Common methods # ################## def show(self): logging.debug("Showing manager") vis = self.is_visible() self.topwin.present() if vis: return if self.prev_position: self.topwin.move(*self.prev_position) self.prev_position = None self.emit("manager-opened") def close(self, src_ignore=None, src2_ignore=None): logging.debug("Closing manager") if not self.is_visible(): return self.prev_position = self.topwin.get_position() self.topwin.hide() self.emit("manager-closed") return 1 def _cleanup(self): self.close() self.rows = None self.diskcol = None self.guestcpucol = None self.hostcpucol = None self.netcol = None if self.delete_dialog: self.delete_dialog.cleanup() self.delete_dialog = None self.vmmenu.destroy() self.vmmenu = None self.vmmenu_items = None self.vmmenushutdown.destroy() self.vmmenushutdown = None self.vmmenushutdown_items = None self.connmenu.destroy() self.connmenu = None self.connmenu_items = None def is_visible(self): return bool(self.topwin.flags() & gtk.VISIBLE) def set_startup_error(self, msg): self.widget("vm-notebook").set_current_page(1) self.widget("startup-error-label").set_text(msg) ################ # Init methods # ################ def init_stats(self): self.add_gconf_handle( self.config.on_vmlist_guest_cpu_usage_visible_changed( self.toggle_guest_cpu_usage_visible_widget)) self.add_gconf_handle( self.config.on_vmlist_host_cpu_usage_visible_changed( self.toggle_host_cpu_usage_visible_widget)) self.add_gconf_handle( self.config.on_vmlist_disk_io_visible_changed( self.toggle_disk_io_visible_widget)) self.add_gconf_handle( self.config.on_vmlist_network_traffic_visible_changed( self.toggle_network_traffic_visible_widget)) # Register callbacks with the global stats enable/disable values # that disable the associated vmlist widgets if reporting is disabled self.add_gconf_handle( self.config.on_stats_enable_disk_poll_changed(self.enable_polling, COL_DISK)) self.add_gconf_handle( self.config.on_stats_enable_net_poll_changed(self.enable_polling, COL_NETWORK)) self.widget("menu_view_stats_guest_cpu").set_active( self.config.is_vmlist_guest_cpu_usage_visible()) self.widget("menu_view_stats_host_cpu").set_active( self.config.is_vmlist_host_cpu_usage_visible()) self.widget("menu_view_stats_disk").set_active( self.config.is_vmlist_disk_io_visible()) self.widget("menu_view_stats_network").set_active( self.config.is_vmlist_network_traffic_visible()) def init_toolbar(self): self.widget("vm-new").set_icon_name("vm_new") self.widget("vm-open").set_icon_name("icon_console") uihelpers.build_shutdown_button_menu(self.widget("vm-shutdown"), self.poweroff_vm, self.reboot_vm, self.destroy_vm, self.save_vm) tool = self.widget("vm-toolbar") util.safe_set_prop(tool, "icon-size", gtk.ICON_SIZE_LARGE_TOOLBAR) for c in tool.get_children(): c.set_homogeneous(False) def init_context_menus(self): def build_icon(name): return gtk.image_new_from_icon_name(name, gtk.ICON_SIZE_MENU) def build_stock(name): return gtk.image_new_from_stock(name, gtk.ICON_SIZE_MENU) icon_name = self.config.get_shutdown_icon_name() shutdownmenu_icon = build_icon(icon_name) reboot_icon = build_icon(icon_name) shutdown_icon = build_icon(icon_name) destroy_icon = build_icon(icon_name) run_icon = build_stock(gtk.STOCK_MEDIA_PLAY) pause_icon = build_stock(gtk.STOCK_MEDIA_PAUSE) save_icon = build_stock(gtk.STOCK_SAVE) resume_icon = build_stock(gtk.STOCK_MEDIA_PAUSE) delete_icon = build_stock(gtk.STOCK_DELETE) def add_to_menu(menu, items, idx, text, icon, cb): item = gtk.ImageMenuItem(text) if icon: item.set_image(icon) item.show() if cb: item.connect("activate", cb) menu.add(item) items[idx] = item def add_vm_menu(idx, text, icon, cb): add_to_menu(self.vmmenu, self.vmmenu_items, idx, text, icon, cb) def add_shutdown_menu(idx, text, icon, cb): add_to_menu(self.vmmenushutdown, self.vmmenushutdown_items, idx, text, icon, cb) def add_conn_menu(idx, text, icon, cb): add_to_menu(self.connmenu, self.connmenu_items, idx, text, icon, cb) def add_sep(menu, items, idx): sep = gtk.SeparatorMenuItem() sep.show() menu.add(sep) items[idx] = sep # Build VM context menu add_vm_menu("run", _("_Run"), run_icon, self.start_vm) add_vm_menu("pause", _("_Pause"), pause_icon, self.pause_vm) add_vm_menu("resume", _("R_esume"), resume_icon, self.resume_vm) add_vm_menu("shutdown", _("_Shut Down"), shutdownmenu_icon, None) self.vmmenu_items["shutdown"].set_submenu(self.vmmenushutdown) add_shutdown_menu("reboot", _("_Reboot"), reboot_icon, self.reboot_vm) add_shutdown_menu("poweroff", _("_Shut Down"), shutdown_icon, self.poweroff_vm) add_shutdown_menu("forcepoweroff", _("_Force Off"), destroy_icon, self.destroy_vm) add_sep(self.vmmenushutdown, self.vmmenushutdown_items, "sep") add_shutdown_menu("save", _("Sa_ve"), save_icon, self.save_vm) add_sep(self.vmmenu, self.vmmenu_items, "hsep1") add_vm_menu("clone", _("_Clone..."), None, self.open_clone_window) add_vm_menu("migrate", _("_Migrate..."), None, self.migrate_vm) add_vm_menu("delete", _("_Delete"), delete_icon, self.do_delete) add_sep(self.vmmenu, self.vmmenu_items, "hsep2") add_vm_menu("open", gtk.STOCK_OPEN, None, self.show_vm) self.vmmenu.show() # Build connection context menu add_conn_menu("create", gtk.STOCK_NEW, None, self.new_vm) add_conn_menu("connect", gtk.STOCK_CONNECT, None, self.open_conn) add_conn_menu("disconnect", gtk.STOCK_DISCONNECT, None, self.close_conn) add_sep(self.connmenu, self.connmenu_items, "hsep1") add_conn_menu("delete", gtk.STOCK_DELETE, None, self.do_delete) add_sep(self.connmenu, self.connmenu_items, "hsep2") add_conn_menu("details", _("D_etails"), None, self.show_host) self.connmenu.show() def init_vmlist(self): vmlist = self.widget("vm-list") self.widget("vm-notebook").set_show_tabs(False) # Handle, name, markup, status, status icon name, key/uuid, hint, # is conn, is conn connected, is vm, is vm running, fg color, # inspection icon model = gtk.TreeStore(object, str, str, str, str, str, str, bool, bool, bool, bool, gtk.gdk.Color, gtk.gdk.Pixbuf) vmlist.set_model(model) util.tooltip_wrapper(vmlist, ROW_HINT, "set_tooltip_column") vmlist.set_headers_visible(True) if hasattr(vmlist, "set_level_indentation"): vmlist.set_level_indentation(-15) nameCol = gtk.TreeViewColumn(_("Name")) nameCol.set_expand(True) nameCol.set_spacing(6) statusCol = nameCol vmlist.append_column(nameCol) status_icon = gtk.CellRendererPixbuf() status_icon.set_property("stock-size", gtk.ICON_SIZE_DND) statusCol.pack_start(status_icon, False) statusCol.add_attribute(status_icon, 'icon-name', ROW_STATUS_ICON) statusCol.add_attribute(status_icon, 'visible', ROW_IS_VM) inspection_os_icon = gtk.CellRendererPixbuf() statusCol.pack_start(inspection_os_icon, False) statusCol.add_attribute(inspection_os_icon, 'pixbuf', ROW_INSPECTION_OS_ICON) statusCol.add_attribute(inspection_os_icon, 'visible', ROW_IS_VM) name_txt = gtk.CellRendererText() nameCol.pack_start(name_txt, True) nameCol.add_attribute(name_txt, 'markup', ROW_MARKUP) nameCol.add_attribute(name_txt, 'foreground-gdk', ROW_COLOR) nameCol.set_sort_column_id(COL_NAME) def make_stats_column(title, datafunc, is_visible, colnum): col = gtk.TreeViewColumn(title) col.set_min_width(140) txt = gtk.CellRendererText() img = CellRendererSparkline() img.set_property("xpad", 6) img.set_property("ypad", 12) img.set_property("reversed", True) col.pack_start(img, True) col.pack_start(txt, False) col.add_attribute(img, 'visible', ROW_IS_VM) col.add_attribute(txt, 'visible', ROW_IS_CONN) col.set_cell_data_func(img, datafunc, None) col.set_visible(is_visible) col.set_sort_column_id(colnum) vmlist.append_column(col) return col self.guestcpucol = make_stats_column(_("CPU usage"), self.guest_cpu_usage_img, self.config.is_vmlist_guest_cpu_usage_visible(), COL_GUEST_CPU) self.hostcpucol = make_stats_column(_("Host CPU usage"), self.host_cpu_usage_img, self.config.is_vmlist_host_cpu_usage_visible(), COL_HOST_CPU) self.diskcol = make_stats_column(_("Disk I/O"), self.disk_io_img, self.config.is_vmlist_disk_io_visible(), COL_DISK) self.netcol = make_stats_column(_("Network I/O"), self.network_traffic_img, self.config.is_vmlist_network_traffic_visible(), COL_NETWORK) model.set_sort_func(COL_NAME, self.vmlist_name_sorter) model.set_sort_func(COL_GUEST_CPU, self.vmlist_guest_cpu_usage_sorter) model.set_sort_func(COL_HOST_CPU, self.vmlist_host_cpu_usage_sorter) model.set_sort_func(COL_DISK, self.vmlist_disk_io_sorter) model.set_sort_func(COL_NETWORK, self.vmlist_network_usage_sorter) model.set_sort_column_id(COL_NAME, gtk.SORT_ASCENDING) ################## # Helper methods # ################## def current_row(self): vmlist = self.widget("vm-list") selection = vmlist.get_selection() active = selection.get_selected() treestore, treeiter = active if treeiter != None: return treestore[treeiter] return None def current_vm(self): row = self.current_row() if not row or row[ROW_IS_CONN]: return None return row[ROW_HANDLE] def current_conn(self): row = self.current_row() if not row: return None handle = row[ROW_HANDLE] if row[ROW_IS_CONN]: return handle else: return handle.conn def current_vmuuid(self): vm = self.current_vm() if vm is None: return None return vm.get_uuid() def current_conn_uri(self, default_selection=False): vmlist = self.widget("vm-list") model = vmlist.get_model() conn = self.current_conn() if conn is None and default_selection: # Nothing selected, use first connection row for row in model: if row[ROW_IS_CONN]: conn = row[ROW_HANDLE] break if conn: return conn.get_uri() return None #################### # Action listeners # #################### def window_resized(self, ignore, event): # Sometimes dimensions change when window isn't visible if not self.is_visible(): return self.config.set_manager_window_size(event.width, event.height) def exit_app(self, src_ignore=None, src2_ignore=None): self.emit("action-exit-app") def new_conn(self, src_ignore=None): self.emit("action-show-connect") def new_vm(self, src_ignore=None): self.emit("action-show-create", self.current_conn_uri()) def show_about(self, src_ignore): self.emit("action-show-about") def show_help(self, src_ignore): self.emit("action-show-help", None) def show_preferences(self, src_ignore): self.emit("action-show-preferences") def show_host(self, src_ignore): uri = self.current_conn_uri(default_selection=True) self.emit("action-show-host", uri) def show_vm(self, ignore, ignore2=None, ignore3=None): conn = self.current_conn() vm = self.current_vm() if conn is None: return if vm: self.emit("action-show-vm", conn.get_uri(), vm.get_uuid()) else: if not self.open_conn(): self.emit("action-show-host", conn.get_uri()) def open_clone_window(self, ignore1=None, ignore2=None, ignore3=None): if self.current_vmuuid(): self.emit("action-clone-domain", self.current_conn_uri(), self.current_vmuuid()) def do_delete(self, ignore=None): conn = self.current_conn() vm = self.current_vm() if vm is None: self._do_delete_conn(conn) else: self._do_delete_vm(vm) def _do_delete_conn(self, conn): if conn is None: return result = self.err.yes_no(_("This will remove the connection:\n\n%s\n\n" "Are you sure?") % conn.get_uri()) if not result: return self.emit("remove-conn", conn.get_uri()) def _do_delete_vm(self, vm): if vm.is_active(): return if not self.delete_dialog: self.delete_dialog = vmmDeleteDialog() self.delete_dialog.show(vm, self.topwin) def set_pause_state(self, state): src = self.widget("vm-pause") try: self.ignore_pause = True src.set_active(state) finally: self.ignore_pause = False def pause_vm_button(self, src): if self.ignore_pause: return do_pause = src.get_active() # Set button state back to original value: just let the status # update function fix things for us self.set_pause_state(not do_pause) if do_pause: self.pause_vm(None) else: self.resume_vm(None) def start_vm(self, ignore): vm = self.current_vm() if vm is not None: self.emit("action-run-domain", vm.conn.get_uri(), vm.get_uuid()) def reboot_vm(self, ignore): vm = self.current_vm() if vm is not None: self.emit("action-reboot-domain", vm.conn.get_uri(), vm.get_uuid()) def poweroff_vm(self, ignore): vm = self.current_vm() if vm is not None: self.emit("action-shutdown-domain", vm.conn.get_uri(), vm.get_uuid()) def destroy_vm(self, ignore): vm = self.current_vm() if vm is not None: self.emit("action-destroy-domain", vm.conn.get_uri(), vm.get_uuid()) def save_vm(self, ignore): vm = self.current_vm() if vm is not None: self.emit("action-save-domain", vm.conn.get_uri(), vm.get_uuid()) def pause_vm(self, ignore): vm = self.current_vm() if vm is not None: self.emit("action-suspend-domain", vm.conn.get_uri(), vm.get_uuid()) def resume_vm(self, ignore): vm = self.current_vm() if vm is not None: self.emit("action-resume-domain", vm.conn.get_uri(), vm.get_uuid()) def migrate_vm(self, ignore): vm = self.current_vm() if vm is not None: self.emit("action-migrate-domain", vm.conn.get_uri(), vm.get_uuid()) def close_conn(self, ignore): conn = self.current_conn() if conn.get_state() != vmmConnection.STATE_DISCONNECTED: conn.close() def open_conn(self, ignore=None): conn = self.current_conn() if conn.get_state() == vmmConnection.STATE_DISCONNECTED: conn.open() return True def _connect_error(self, conn, errmsg, tb, warnconsole): errmsg = errmsg.strip(" \n") tb = tb.strip(" \n") hint = "" show_errmsg = True if conn.is_remote(): logging.debug(conn.get_transport()) if re.search(r"nc: .* -- 'U'", tb): hint += _("The remote host requires a version of netcat/nc\n" "which supports the -U option.") show_errmsg = False elif conn.get_transport()[0] == "ssh" and re.search(r"ssh-askpass", tb): hint += _("You need to install openssh-askpass or similar\n" "to connect to this host.") show_errmsg = False else: hint += _("Verify that the 'libvirtd' daemon is running\n" "on the remote host.") elif conn.is_xen(): hint += _("Verify that:\n" " - A Xen host kernel was booted\n" " - The Xen service has been started") else: if warnconsole: hint += _("Could not detect a local session: if you are \n" "running virt-manager over ssh -X or VNC, you \n" "may not be able to connect to libvirt as a \n" "regular user. Try running as root.") show_errmsg = False elif re.search(r"libvirt-sock", tb): hint += _("Verify that the 'libvirtd' daemon is running.") show_errmsg = False msg = _("Unable to connect to libvirt.") if show_errmsg: msg += "\n\n%s" % errmsg if hint: msg += "\n\n%s" % hint msg = msg.strip("\n") details = msg details += "\n\n" details += "Libvirt URI is: %s\n\n" % conn.get_uri() details += tb self.err.show_err(msg, details, title=_("Virtual Machine Manager Connection Failure")) #################################### # VM add/remove management methods # #################################### def vm_row_key(self, vm): return vm.get_uuid() + ":" + vm.conn.get_uri() def vm_added(self, conn, vmuuid): vm = conn.get_vm(vmuuid) vm.connect("status-changed", self.vm_status_changed) vm.connect("resources-sampled", self.vm_resources_sampled) vm.connect("config-changed", self.vm_resources_sampled, True) vm.connect("inspection-changed", self.vm_inspection_changed) vmlist = self.widget("vm-list") model = vmlist.get_model() self._append_vm(model, vm, conn) def vm_removed(self, conn, vmuuid): vmlist = self.widget("vm-list") model = vmlist.get_model() parent = self.rows[conn.get_uri()].iter for row in range(model.iter_n_children(parent)): vm = model.get_value(model.iter_nth_child(parent, row), ROW_HANDLE) if vm.get_uuid() == vmuuid: model.remove(model.iter_nth_child(parent, row)) del self.rows[self.vm_row_key(vm)] break def _build_conn_hint(self, conn): hint = conn.get_uri() if conn.state == conn.STATE_DISCONNECTED: hint += " (%s)" % _("Double click to connect") return hint def _build_conn_markup(self, conn, row): name = util.xml_escape(row[ROW_NAME]) text = name if conn.state == conn.STATE_DISCONNECTED: text += " - " + _("Not Connected") elif conn.state == conn.STATE_CONNECTING: text += " - " + _("Connecting...") markup = "%s" % text return markup def _build_conn_color(self, conn): color = None if conn.state == conn.STATE_DISCONNECTED: # Color code #5b5b5b color = gtk.gdk.Color(23296, 23296, 23296) return color def _build_vm_markup(self, row): domtext = ("%s" % util.xml_escape(row[ROW_NAME])) statetext = "%s" % row[ROW_STATUS] return domtext + "\n" + statetext def _build_vm_row(self, vm): row = [] row.insert(ROW_HANDLE, vm) row.insert(ROW_NAME, vm.get_name()) row.insert(ROW_MARKUP, "") row.insert(ROW_STATUS, vm.run_status()) row.insert(ROW_STATUS_ICON, vm.run_status_icon_name()) row.insert(ROW_KEY, vm.get_uuid()) row.insert(ROW_HINT, util.xml_escape(vm.get_description())) row.insert(ROW_IS_CONN, False) row.insert(ROW_IS_CONN_CONNECTED, True) row.insert(ROW_IS_VM, True) row.insert(ROW_IS_VM_RUNNING, vm.is_active()) row.insert(ROW_COLOR, None) row.insert(ROW_INSPECTION_OS_ICON, self.get_inspection_icon_pixbuf(vm, 16, 16)) row[ROW_MARKUP] = self._build_vm_markup(row) return row def _append_vm(self, model, vm, conn): row_key = self.vm_row_key(vm) if row_key in self.rows: return row = self._build_vm_row(vm) parent = self.rows[conn.get_uri()].iter _iter = model.append(parent, row) path = model.get_path(_iter) self.rows[row_key] = model[path] # Expand a connection when adding a vm to it self.widget("vm-list").expand_row(model.get_path(parent), False) def _append_conn(self, model, conn): row = [] row.insert(ROW_HANDLE, conn) row.insert(ROW_NAME, conn.get_pretty_desc_inactive(False)) row.insert(ROW_MARKUP, self._build_conn_markup(conn, row)) row.insert(ROW_STATUS, ("%s" % conn.get_state_text())) row.insert(ROW_STATUS_ICON, None) row.insert(ROW_KEY, conn.get_uri()) row.insert(ROW_HINT, self._build_conn_hint(conn)) row.insert(ROW_IS_CONN, True) row.insert(ROW_IS_CONN_CONNECTED, conn.state != conn.STATE_DISCONNECTED) row.insert(ROW_IS_VM, False) row.insert(ROW_IS_VM_RUNNING, False) row.insert(ROW_COLOR, self._build_conn_color(conn)) row.insert(ROW_INSPECTION_OS_ICON, None) _iter = model.append(None, row) path = model.get_path(_iter) self.rows[conn.get_uri()] = model[path] return _iter def add_conn(self, engine_ignore, conn): # Make sure error page isn't showing self.widget("vm-notebook").set_current_page(0) if conn.get_uri() in self.rows: return conn.connect("vm-added", self.vm_added) conn.connect("vm-removed", self.vm_removed) conn.connect("resources-sampled", self.conn_resources_sampled) conn.connect("state-changed", self.conn_state_changed) conn.connect("connect-error", self._connect_error) # add the connection to the treeModel vmlist = self.widget("vm-list") row = self._append_conn(vmlist.get_model(), conn) vmlist.get_selection().select_iter(row) # Try to make sure that 2 row descriptions don't collide connrows = [] descs = [] for row in self.rows.values(): if row[ROW_IS_CONN]: connrows.append(row) for row in connrows: descs.append(row[ROW_NAME]) for row in connrows: conn = row[ROW_HANDLE] name = row[ROW_NAME] if descs.count(name) <= 1: continue newname = conn.get_pretty_desc_inactive(False, True) self.conn_resources_sampled(conn, newname) def remove_conn(self, engine_ignore, uri): model = self.widget("vm-list").get_model() parent = self.rows[uri].iter if parent is None: return child = model.iter_children(parent) while child is not None: del self.rows[self.vm_row_key(model.get_value(child, ROW_HANDLE))] model.remove(child) child = model.iter_children(parent) model.remove(parent) del self.rows[uri] ############################# # State/UI updating methods # ############################# def vm_status_changed(self, vm, oldstatus, newstatus): ignore = newstatus ignore = oldstatus parent = self.rows[vm.conn.get_uri()].iter vmlist = self.widget("vm-list") model = vmlist.get_model() missing = True for row in range(model.iter_n_children(parent)): _iter = model.iter_nth_child(parent, row) if model.get_value(_iter, ROW_KEY) == vm.get_uuid(): missing = False break if missing: self._append_vm(model, vm, vm.conn) # Update run/shutdown/pause button states self.vm_selected() self.vm_resources_sampled(vm) def vm_resources_sampled(self, vm, config_changed=False): vmlist = self.widget("vm-list") model = vmlist.get_model() if self.vm_row_key(vm) not in self.rows: return row = self.rows[self.vm_row_key(vm)] row[ROW_NAME] = vm.get_name() row[ROW_STATUS] = vm.run_status() row[ROW_STATUS_ICON] = vm.run_status_icon_name() row[ROW_IS_VM_RUNNING] = vm.is_active() row[ROW_MARKUP] = self._build_vm_markup(row) if config_changed: row[ROW_HINT] = util.xml_escape(vm.get_description()) model.row_changed(row.path, row.iter) def vm_inspection_changed(self, vm): vmlist = self.window.get_widget("vm-list") model = vmlist.get_model() if self.vm_row_key(vm) not in self.rows: return row = self.rows[self.vm_row_key(vm)] row[ROW_INSPECTION_OS_ICON] = \ self.get_inspection_icon_pixbuf(vm, 16, 16) model.row_changed(row.path, row.iter) def get_inspection_icon_pixbuf(self, vm, w, h): # libguestfs gives us the PNG data as a string. png_data = vm.inspection.icon if png_data == None: return None try: pb = gtk.gdk.PixbufLoader(image_type="png") pb.set_size(w, h) pb.write(png_data) pb.close() return pb.get_pixbuf() except: return None def conn_state_changed(self, conn): self.conn_resources_sampled(conn) self.vm_selected() def conn_resources_sampled(self, conn, newname=None): vmlist = self.widget("vm-list") model = vmlist.get_model() row = self.rows[conn.get_uri()] if newname: row[ROW_NAME] = newname row[ROW_MARKUP] = self._build_conn_markup(conn, row) row[ROW_STATUS] = ("%s" % conn.get_state_text()) row[ROW_IS_CONN_CONNECTED] = conn.state != conn.STATE_DISCONNECTED row[ROW_COLOR] = self._build_conn_color(conn) row[ROW_HINT] = self._build_conn_hint(conn) if conn.get_state() in [vmmConnection.STATE_DISCONNECTED, vmmConnection.STATE_CONNECTING]: # Connection went inactive, delete any VM child nodes parent = self.rows[conn.get_uri()].iter if parent is not None: child = model.iter_children(parent) while child is not None: del self.rows[self.vm_row_key(model.get_value(child, ROW_HANDLE))] model.remove(child) child = model.iter_children(parent) self.max_disk_rate = max(self.max_disk_rate, conn.disk_io_max_rate()) self.max_net_rate = max(self.max_net_rate, conn.network_traffic_max_rate()) model.row_changed(row.path, row.iter) def change_run_text(self, can_restore): if can_restore: text = _("_Restore") else: text = _("_Run") strip_text = text.replace("_", "") self.vmmenu_items["run"].get_child().set_label(text) self.widget("vm-run").set_label(strip_text) def vm_selected(self, ignore=None): conn = self.current_conn() vm = self.current_vm() show_open = bool(vm) show_details = bool(vm) host_details = bool(len(self.rows)) delete = bool((vm and vm.is_runable()) or (not vm and conn)) show_run = bool(vm and vm.is_runable()) is_paused = bool(vm and vm.is_paused()) if is_paused: show_pause = bool(vm and vm.is_unpauseable()) else: show_pause = bool(vm and vm.is_pauseable()) show_shutdown = bool(vm and vm.is_stoppable()) if vm and vm.managedsave_supported: self.change_run_text(vm.hasSavedImage()) self.widget("vm-open").set_sensitive(show_open) self.widget("vm-run").set_sensitive(show_run) self.widget("vm-shutdown").set_sensitive(show_shutdown) self.set_pause_state(is_paused) self.widget("vm-pause").set_sensitive(show_pause) self.widget("menu_edit_details").set_sensitive(show_details) self.widget("menu_host_details").set_sensitive(host_details) self.widget("menu_edit_delete").set_sensitive(delete) def popup_vm_menu_key(self, widget_ignore, event): if gtk.gdk.keyval_name(event.keyval) != "Menu": return False vmlist = self.widget("vm-list") treeselection = vmlist.get_selection() model, _iter = treeselection.get_selected() self.popup_vm_menu(model, _iter, event) return True def popup_vm_menu_button(self, widget, event): if event.button != 3: return False tup = widget.get_path_at_pos(int(event.x), int(event.y)) if tup == None: return False path = tup[0] model = widget.get_model() _iter = model.get_iter(path) self.popup_vm_menu(model, _iter, event) return False def popup_vm_menu(self, model, _iter, event): if model.iter_parent(_iter) != None: # Popup the vm menu vm = model.get_value(_iter, ROW_HANDLE) destroy = vm.is_destroyable() run = vm.is_runable() stop = vm.is_stoppable() paused = vm.is_paused() ro = vm.is_read_only() self.vmmenu_items["run"].set_sensitive(run) self.vmmenu_items["shutdown"].set_sensitive(stop) self.vmmenu_items["pause"].set_property("visible", not paused) self.vmmenu_items["pause"].set_sensitive(stop) self.vmmenu_items["resume"].set_property("visible", paused) self.vmmenu_items["resume"].set_sensitive(paused) self.vmmenu_items["migrate"].set_sensitive(stop) self.vmmenu_items["clone"].set_sensitive(not ro) self.vmmenu_items["delete"].set_sensitive(run) self.vmmenushutdown_items["poweroff"].set_sensitive(stop) self.vmmenushutdown_items["reboot"].set_sensitive(stop) self.vmmenushutdown_items["forcepoweroff"].set_sensitive(destroy) self.vmmenushutdown_items["save"].set_sensitive(destroy) self.vmmenu.popup(None, None, None, 0, event.time) else: # Pop up connection menu conn = model.get_value(_iter, ROW_HANDLE) disconn = (conn.get_state() == vmmConnection.STATE_DISCONNECTED) conning = (conn.get_state() == vmmConnection.STATE_CONNECTING) self.connmenu_items["create"].set_sensitive(not disconn) self.connmenu_items["disconnect"].set_sensitive(not (disconn or conning)) self.connmenu_items["connect"].set_sensitive(disconn) self.connmenu_items["delete"].set_sensitive(disconn) self.connmenu.popup(None, None, None, 0, event.time) ################# # Stats methods # ################# def vmlist_name_sorter(self, model, iter1, iter2): return cmp(model.get_value(iter1, ROW_NAME), model.get_value(iter2, ROW_NAME)) def vmlist_guest_cpu_usage_sorter(self, model, iter1, iter2): obj1 = model.get_value(iter1, ROW_HANDLE) obj2 = model.get_value(iter2, ROW_HANDLE) return cmp(obj1.guest_cpu_time_percentage(), obj2.guest_cpu_time_percentage()) def vmlist_host_cpu_usage_sorter(self, model, iter1, iter2): obj1 = model.get_value(iter1, ROW_HANDLE) obj2 = model.get_value(iter2, ROW_HANDLE) return cmp(obj1.host_cpu_time_percentage(), obj2.host_cpu_time_percentage()) def vmlist_disk_io_sorter(self, model, iter1, iter2): obj1 = model.get_value(iter1, ROW_HANDLE) obj2 = model.get_value(iter2, ROW_HANDLE) return cmp(obj1.disk_io_rate(), obj2.disk_io_rate()) def vmlist_network_usage_sorter(self, model, iter1, iter2): obj1 = model.get_value(iter1, ROW_HANDLE) obj2 = model.get_value(iter2, ROW_HANDLE) return cmp(obj1.network_traffic_rate(), obj2.network_traffic_rate()) def enable_polling(self, ignore1, ignore2, conf_entry, userdata): if userdata == COL_DISK: widgn = "menu_view_stats_disk" elif userdata == COL_NETWORK: widgn = "menu_view_stats_network" widget = self.widget(widgn) tool_text = "" if conf_entry and (conf_entry == True or conf_entry.get_value().get_bool()): widget.set_sensitive(True) else: if widget.get_active(): widget.set_active(False) widget.set_sensitive(False) tool_text = _("Disabled in preferences dialog.") util.tooltip_wrapper(widget, tool_text) # RHEL5 GTK doesn't support get_label for a checkmenuitem if hasattr(widget, "get_label"): disabled_text = _(" (disabled)") current_text = widget.get_label().strip(disabled_text) if tool_text: current_text = current_text + disabled_text widget.set_label(current_text) def toggle_network_traffic_visible_widget(self, *ignore): val = self.config.is_vmlist_network_traffic_visible() self.netcol.set_visible(val) self.widget("menu_view_stats_network").set_active(val) def toggle_disk_io_visible_widget(self, *ignore): val = self.config.is_vmlist_disk_io_visible() self.diskcol.set_visible(val) self.widget("menu_view_stats_disk").set_active(val) def toggle_guest_cpu_usage_visible_widget(self, *ignore): val = self.config.is_vmlist_guest_cpu_usage_visible() self.guestcpucol.set_visible(val) self.widget("menu_view_stats_guest_cpu").set_active(val) def toggle_host_cpu_usage_visible_widget(self, *ignore): val = self.config.is_vmlist_host_cpu_usage_visible() self.hostcpucol.set_visible(val) self.widget("menu_view_stats_host_cpu").set_active(val) def toggle_stats_visible(self, src, stats_id): visible = src.get_active() set_stats = { COL_GUEST_CPU: self.config.set_vmlist_guest_cpu_usage_visible, COL_HOST_CPU: self.config.set_vmlist_host_cpu_usage_visible, COL_DISK: self.config.set_vmlist_disk_io_visible, COL_NETWORK: self.config.set_vmlist_network_traffic_visible, } set_stats[stats_id](visible) def guest_cpu_usage_img(self, column_ignore, cell, model, _iter, data): obj = model.get_value(_iter, ROW_HANDLE) if obj is None: return data = obj.guest_cpu_time_vector_limit(40) cell.set_property('data_array', data) def host_cpu_usage_img(self, column_ignore, cell, model, _iter, data): obj = model.get_value(_iter, ROW_HANDLE) if obj is None: return data = obj.host_cpu_time_vector_limit(40) cell.set_property('data_array', data) def disk_io_img(self, column_ignore, cell, model, _iter, data): obj = model.get_value(_iter, ROW_HANDLE) if obj is None: return if not hasattr(obj, "conn"): return data = obj.disk_io_vector_limit(40, self.max_disk_rate) cell.set_property('data_array', data) def network_traffic_img(self, column_ignore, cell, model, _iter, data): obj = model.get_value(_iter, ROW_HANDLE) if obj is None: return if not hasattr(obj, "conn"): return data = obj.network_traffic_vector_limit(40, self.max_net_rate) cell.set_property('data_array', data) vmmGObjectUI.type_register(vmmManager) vmmManager.signal_new(vmmManager, "action-show-connect", []) vmmManager.signal_new(vmmManager, "action-show-vm", [str, str]) vmmManager.signal_new(vmmManager, "action-show-about", []) vmmManager.signal_new(vmmManager, "action-show-host", [str]) vmmManager.signal_new(vmmManager, "action-show-preferences", []) vmmManager.signal_new(vmmManager, "action-show-create", [str]) vmmManager.signal_new(vmmManager, "action-suspend-domain", [str, str]) vmmManager.signal_new(vmmManager, "action-resume-domain", [str, str]) vmmManager.signal_new(vmmManager, "action-run-domain", [str, str]) vmmManager.signal_new(vmmManager, "action-shutdown-domain", [str, str]) vmmManager.signal_new(vmmManager, "action-reboot-domain", [str, str]) vmmManager.signal_new(vmmManager, "action-destroy-domain", [str, str]) vmmManager.signal_new(vmmManager, "action-save-domain", [str, str]) vmmManager.signal_new(vmmManager, "action-connect", [str]) vmmManager.signal_new(vmmManager, "action-show-help", [str]) vmmManager.signal_new(vmmManager, "action-migrate-domain", [str, str]) vmmManager.signal_new(vmmManager, "action-clone-domain", [str, str]) vmmManager.signal_new(vmmManager, "action-exit-app", []) vmmManager.signal_new(vmmManager, "manager-closed", []) vmmManager.signal_new(vmmManager, "manager-opened", []) vmmManager.signal_new(vmmManager, "remove-conn", [str]) vmmManager.signal_new(vmmManager, "add-default-conn", [])