virt-manager/virtManager/clone.py
Cole Robinson 382391eadb Clean up gtk STOCK and various image usage
- Remove most use of deprecated stock icons. Without it the UI will
  be a lot more ugly in Fedora 36
- Remove deprecated ImageMenuItem usage, convert to regular MenuItem
- Remove most embedded button images

Signed-off-by: Cole Robinson <crobinso@redhat.com>
2022-02-19 15:33:42 -05:00

616 lines
20 KiB
Python

# Copyright (C) 2009, 2013 Red Hat, Inc.
# Copyright (C) 2009 Cole Robinson <crobinso@redhat.com>
#
# This work is licensed under the GNU GPLv2 or later.
# See the COPYING file in the top-level directory.
import collections
from gi.repository import Gtk
from gi.repository import Pango
import virtinst
from virtinst import Cloner
from virtinst import log
from virtinst import xmlutil
from .lib import uiutil
from .baseclass import vmmGObjectUI
from .asyncjob import vmmAsyncJob
from .storagebrowse import vmmStorageBrowser
def _get_cloneable_msg(diskinfo):
"""Is the passed path even clone-able"""
if diskinfo.get_cloneable_msg():
return diskinfo.get_cloneable_msg()
if diskinfo.disk.is_empty():
return _("No storage to clone.")
class _StorageInfo:
"""
Our wrapper class that is close to Cloner _DiskInfo content.
We track all user choices explicitly in this class, and then
serialize them into the actual cloner diskinfo once the
user clicks 'Clone' to complete the process.
"""
def __init__(self, vm, cloner, diskinfo):
self._diskinfo = diskinfo
self._orig_disk = diskinfo.disk
self._orig_vm_name = vm.get_name()
self._new_vm_name = cloner.new_guest.name
self.share_msg = diskinfo.get_share_msg()
self.cloneable_msg = _get_cloneable_msg(diskinfo)
self._manual_path = None
self._generated_path = (diskinfo.new_disk and
diskinfo.new_disk.get_source_path())
if not self._generated_path:
self._reset_generated_path()
self._is_clone_requested = diskinfo.is_clone_requested()
if self.share_msg or not self.is_cloneable():
self._is_clone_requested = False
def is_cloneable(self):
return not bool(self.cloneable_msg)
def is_clone_requested(self):
return self._is_clone_requested
def is_share_requested(self):
return not self._is_clone_requested
def warn_about_sharing(self):
return not self.share_msg
def get_target(self):
return self._orig_disk.target
def get_orig_disk_path(self):
return self._orig_disk.get_source_path()
def get_new_disk_path(self):
if self._manual_path:
return self._manual_path
return self._generated_path
def set_clone_requested(self, val):
self._is_clone_requested = bool(val)
def set_manual_path(self, val):
self._manual_path = val
def set_new_vm_name(self, val):
if not val or val == self._new_vm_name:
return
self._new_vm_name = val
self._reset_generated_path()
def _reset_generated_path(self):
self._generated_path = Cloner.generate_clone_disk_path(
self._orig_disk.conn,
self._orig_vm_name,
self._new_vm_name,
self._orig_disk.get_source_path())
def set_values_on_diskinfo(self, diskinfo):
if not self._is_clone_requested:
diskinfo.set_share_requested()
diskinfo.new_disk = None
return
diskinfo.set_clone_requested()
sparse = True
newpath = self.get_new_disk_path()
diskinfo.set_new_path(newpath, sparse)
###################
# UI info helpers #
###################
def get_tooltip(self):
lines = []
lines.append(_("Disk target: %s") % self.get_target())
lines.append(_("Original path: %s") % self.get_orig_disk_path())
if self.get_new_disk_path():
lines.append(_("New path: %s") % self.get_new_disk_path())
lines.append("\n")
if self.share_msg:
lines.append(_("Storage is safe to share: %(reason)s") % {
"reason": self.share_msg})
else:
lines.append(
_("Sharing this storage is potentially dangerous."))
if self.cloneable_msg:
lines.append(_("Storage is not cloneable: %(reason)s") % {
"reason": self.cloneable_msg})
return "\n".join(lines)
def get_markup(self, vm):
lines = []
line = ""
path = self.get_orig_disk_path()
if path:
line += path
else:
line += _("No storage.")
lines.append(line)
line = ""
if self.is_share_requested():
line += _("Share disk with %s") % vm.get_name()
if self.is_clone_requested():
line += _("Clone this disk")
sizelabel = self.get_size_label()
if sizelabel:
line += " (%s)" % sizelabel
if line:
lines.append(line)
label = "\n".join(lines)
markup = "<small>%s</small>" % xmlutil.xml_escape(label)
return markup
def get_icon_name(self):
if self._orig_disk.is_floppy():
return "media-floppy"
if self._orig_disk.is_cdrom():
return "media-optical"
return "drive-harddisk"
def get_size_label(self):
size = self._orig_disk.get_size()
if not size:
return "" # pragma: no cover
return "%.1f GiB" % float(size)
class vmmCloneVM(vmmGObjectUI):
@classmethod
def show_instance(cls, parentobj, vm):
try:
# Maintain one dialog per connection
uri = vm.conn.get_uri()
if cls._instances is None:
cls._instances = {}
if uri not in cls._instances:
cls._instances[uri] = vmmCloneVM()
cls._instances[uri].show(parentobj.topwin, vm)
except Exception as e: # pragma: no cover
parentobj.err.show_err(
_("Error launching clone dialog: %s") % str(e))
def __init__(self):
vmmGObjectUI.__init__(self, "clone.ui", "vmm-clone")
self.vm = None
self._storage_list = None
self._storage_browser = None
self._storage_dialog = self.widget("vmm-change-storage")
self._storage_dialog.set_transient_for(self.topwin)
self.builder.connect_signals({
"on_clone_delete_event": self._close_cb,
"on_clone_cancel_clicked": self._close_cb,
"on_clone_ok_clicked": self._finish_clicked_cb,
"on_storage_selection_changed": self._storage_selection_changed_cb,
"on_storage_details_clicked": self._storage_details_clicked_cb,
# Storage subdialog signals
"on_vmm_change_storage_delete_event": self._storage_dialog_close_cb,
"on_change_storage_cancel_clicked": self._storage_dialog_close_cb,
"on_change_storage_ok_clicked": self._storage_dialog_finish_cb,
"on_change_storage_doclone_toggled": self._storage_dialog_doclone_toggled_cb,
"on_change_storage_browse_clicked": self._storage_dialog_browse_cb,
})
self.bind_escape_key_close()
self._cleanup_on_app_close()
self._init_ui()
#######################
# Standard UI methods #
#######################
@property
def conn(self):
return self.vm and self.vm.conn or None
def show(self, parent, vm):
log.debug("Showing clone wizard")
self._set_vm(vm)
self._reset_state()
self.topwin.set_transient_for(parent)
self.topwin.resize(1, 1)
self.topwin.present()
def _storage_dialog_close(self):
self._storage_dialog.hide()
return 1
def close(self, ignore1=None, ignore2=None):
log.debug("Closing clone wizard")
self._storage_dialog_close()
self.topwin.hide()
self._set_vm(None)
self._storage_list = None
return 1
def _vm_removed_cb(self, _conn, vm):
if self.vm == vm:
self.close()
def _set_vm(self, newvm):
oldvm = self.vm
if oldvm:
oldvm.conn.disconnect_by_obj(self)
if newvm:
newvm.conn.connect("vm-removed", self._vm_removed_cb)
self.vm = newvm
def _cleanup(self):
self._storage_dialog.destroy()
self._storage_dialog = None
if self._storage_browser:
self._storage_browser.cleanup()
self._storage_browser = None
###########
# UI init #
###########
def _init_ui(self):
storage_list = self.widget("storage-list")
# [disk target, tooltip]
model = Gtk.ListStore(str, str)
storage_list.set_model(model)
storage_list.set_tooltip_column(1)
cloneCol = Gtk.TreeViewColumn(_("Clone"))
pathCol = Gtk.TreeViewColumn(_("Storage"))
pathCol.set_sizing(Gtk.TreeViewColumnSizing.AUTOSIZE)
pathCol.set_expand(True)
storage_list.append_column(cloneCol)
storage_list.append_column(pathCol)
def separator_cb(_model, _iter):
return _model[_iter][0] == "separator"
storage_list.set_row_separator_func(separator_cb)
chkbox = Gtk.CellRendererToggle()
chkbox.connect('toggled', self._storage_clone_toggled_cb)
chkimg = Gtk.CellRendererPixbuf()
chkimg.set_property('stock-size', Gtk.IconSize.MENU)
cloneCol.pack_start(chkimg, False)
cloneCol.pack_start(chkbox, False)
def chk_cell_data_cb(column, cell, model, _iter, data):
target = model[_iter][0]
if target == "separator":
return
sinfo = self._storage_list[target]
visible = sinfo.is_cloneable()
active = sinfo.is_clone_requested()
_chkimg = column.get_cells()[0]
_chkbox = column.get_cells()[1]
_chkbox.set_property('active', active)
_chkbox.set_property('visible', visible)
_chkimg.set_property('visible', not visible)
icon = "dialog-information"
if sinfo.warn_about_sharing():
icon = "dialog-warning"
_chkimg.set_property('icon-name', icon)
tooltip = sinfo.get_tooltip()
if tooltip != model[_iter][1]:
model[_iter][1] = tooltip
cloneCol.set_cell_data_func(chkbox, chk_cell_data_cb)
cloneCol.set_cell_data_func(chkimg, chk_cell_data_cb)
pathtxt = Gtk.CellRendererText()
pathCol.set_sort_column_id(0)
pathtxt.set_property("width-chars", 30)
pathtxt.set_property("ellipsize", Pango.EllipsizeMode.MIDDLE)
pathimg = Gtk.CellRendererPixbuf()
pathimg.set_property('stock-size', Gtk.IconSize.MENU)
pathimg.set_padding(3, 0)
pathCol.pack_start(pathimg, False)
pathCol.pack_start(pathtxt, True)
def path_cb(column, cell, model, _iter, data):
target = model[_iter][0]
if target == "separator":
return
sinfo = self._storage_list[target]
_pathimg = column.get_cells()[0]
_pathtxt = column.get_cells()[1]
markup = sinfo.get_markup(self.vm)
_pathtxt.set_property('markup', markup)
_pathimg.set_property('icon-name', sinfo.get_icon_name())
pathCol.set_cell_data_func(pathtxt, path_cb)
pathCol.set_cell_data_func(pathimg, path_cb)
def _reset_state(self):
self.widget("clone-cancel").grab_focus()
self.widget("clone-new-name").set_text("")
# Populate default clone values
cloner = self._build_cloner()
cloner.prepare()
self.widget("clone-orig-name").set_text(cloner.src_name)
self.widget("clone-new-name").set_text(cloner.new_guest.name)
uiutil.set_grid_row_visible(
self.widget("clone-dest-host"), self.conn.is_remote())
self.widget("clone-dest-host").set_text(self.conn.get_pretty_desc())
# Collect info about all the VMs disks
self._storage_list = collections.OrderedDict()
for diskinfo in cloner.get_diskinfos():
sinfo = _StorageInfo(self.vm, cloner, diskinfo)
self._storage_list[sinfo.get_target()] = sinfo
self._populate_storage_ui()
#######################
# Functional routines #
#######################
def _build_cloner(self):
conn = self.conn.get_backend()
orig_name = self.vm.get_name()
new_name = self.widget("clone-new-name").get_text()
cloner = Cloner(conn, src_name=orig_name)
if new_name:
cloner.set_clone_name(new_name)
return cloner
#######################
# Storage UI building #
#######################
def _set_paths_from_clone_name(self):
newname = self.widget("clone-new-name").get_text()
for sinfo in list(self._storage_list.values()):
sinfo.set_new_vm_name(newname)
def _populate_storage_ui(self):
"""
Fill in the storage UI from the collected StorageInfo classes
"""
model = self.widget("storage-list").get_model()
model.clear()
for sinfo in self._storage_list.values():
model.append([sinfo.get_target(), sinfo.get_tooltip()])
model.append(["separator", None])
def _storage_clone_toggled_cb(self, src, treepath):
model = self.widget("storage-list").get_model()
row = model[treepath]
target = row[0]
sinfo = self._storage_list[target]
active = not src.get_property("active")
sinfo.set_clone_requested(active)
model.emit("row-changed", row.path, row.iter)
###########################
# Storage window handling #
###########################
def _show_storage_window(self):
tgt = uiutil.get_list_selection(self.widget("storage-list"))
sinfo = self._storage_list[tgt]
# If storage paths are dependent on manually entered clone name,
# make sure they are up to date
self._set_paths_from_clone_name()
orig = sinfo.get_orig_disk_path()
new = sinfo.get_new_disk_path()
size = sinfo.get_size_label()
can_clone = sinfo.is_cloneable()
do_clone = sinfo.is_clone_requested()
self.widget("change-storage-doclone").set_active(True)
self.widget("change-storage-doclone").toggled()
self.widget("change-storage-orig").set_text(orig)
self.widget("change-storage-target").set_text(tgt)
self.widget("change-storage-size").set_text(size or "-")
self.widget("change-storage-doclone").set_active(do_clone)
self.widget("change-storage-new").set_text(new or "")
self.widget("change-storage-doclone").set_sensitive(can_clone)
self.widget("vmm-change-storage").show_all()
def _storage_dialog_finish(self):
target = self.widget("change-storage-target").get_text()
sinfo = self._storage_list[target]
# Sync 'do clone' checkbox, and main dialog combo
do_clone = self.widget("change-storage-doclone").get_active()
sinfo.set_clone_requested(do_clone)
if not do_clone:
self._storage_dialog_close()
return
new_path = self.widget("change-storage-new").get_text()
if virtinst.DeviceDisk.path_definitely_exists(
self.vm.conn.get_backend(), new_path):
text1 = _("Cloning will overwrite the existing file")
text2 = _("Using an existing image will overwrite "
"the path during the clone process. Are "
"you sure you want to use this path?")
res = self.err.yes_no(text1, text2)
if not res:
return
sinfo.set_manual_path(new_path)
self._storage_dialog_close()
###################
# Finish routines #
###################
def _validate(self, cloner):
new_paths = []
warn_str = ""
for sinfo in self._storage_list.values():
if sinfo.is_clone_requested():
new_paths.append(sinfo.get_new_disk_path())
continue
if not sinfo.warn_about_sharing():
continue
warn_str += "%s: %s\n" % (
sinfo.get_target(), sinfo.get_orig_disk_path())
if warn_str:
res = self.err.ok_cancel(
_("Sharing storage may cause data to be overwritten."),
_("The following disk devices will be shared with %(vmname)s:"
"\n\n%(pathlist)s\n"
"Running the new guest could overwrite data in these "
"disk images.") % {
"vmname": self.vm.get_name(), "pathlist": warn_str})
if not res:
return False
for diskinfo in cloner.get_diskinfos():
diskinfo.raise_error()
def _finish_cb(self, error, details, conn, cloner):
self.reset_finish_cursor()
if error is not None:
error = (_("Error creating virtual machine clone '%(vm)s': "
"%(error)s") % {
"vm": cloner.new_guest.name,
"error": error,
})
self.err.show_err(error, details=details)
return
conn.schedule_priority_tick(pollvm=True)
self.close()
def _async_clone(self, asyncjob, cloner):
meter = asyncjob.get_meter()
refresh_pools = []
for diskinfo in cloner.get_nonshare_diskinfos():
disk = diskinfo.new_disk
pool = disk.get_parent_pool()
if not pool:
continue
poolname = pool.name()
if poolname not in refresh_pools:
refresh_pools.append(poolname)
cloner.start_duplicate(meter)
for poolname in refresh_pools:
try:
pool = self.conn.get_pool_by_name(poolname)
self.idle_add(pool.refresh)
except Exception: # pragma: no cover
log.debug("Error looking up pool=%s for refresh after "
"VM clone.", poolname, exc_info=True)
def _build_final_cloner(self):
self._set_paths_from_clone_name()
cloner = self._build_cloner()
for diskinfo in cloner.get_diskinfos():
target = diskinfo.disk.target
sinfo = self._storage_list[target]
sinfo.set_values_on_diskinfo(diskinfo)
cloner.prepare()
for diskinfo in cloner.get_diskinfos():
diskinfo.raise_error()
if self._validate(cloner) is False:
return
return cloner
def _finish(self):
try:
cloner = self._build_final_cloner()
if not cloner:
return
except Exception as e:
msg = _("Error with clone settings: %s") % str(e)
return self.err.show_err(msg)
self.set_finish_cursor()
title = (_("Creating virtual machine clone '%s'") %
cloner.new_guest.name)
text = title
if cloner.get_nonshare_diskinfos():
text = (_("Creating virtual machine clone '%s' and selected "
"storage (this may take a while)") %
cloner.new_guest.name)
progWin = vmmAsyncJob(self._async_clone, [cloner],
self._finish_cb, [self.conn, cloner],
title, text, self.topwin)
progWin.run()
################
# UI listeners #
################
def _close_cb(self, src, event=None):
return self.close()
def _finish_clicked_cb(self, src):
return self._finish()
def _storage_selection_changed_cb(self, src):
row = uiutil.get_list_selected_row(self.widget("storage-list"))
self.widget("storage-details-button").set_sensitive(bool(row))
def _storage_details_clicked_cb(self, src):
self._show_storage_window()
def _storage_dialog_doclone_toggled_cb(self, src):
do_clone = src.get_active()
self.widget("change-storage-new").set_sensitive(do_clone)
self.widget("change-storage-browse").set_sensitive(do_clone)
def _storage_dialog_close_cb(self, src, event=None):
return self._storage_dialog_close()
def _storage_dialog_finish_cb(self, src):
self._storage_dialog_finish()
def _storage_dialog_browse_cb(self, ignore):
def callback(src_ignore, txt):
self.widget("change-storage-new").set_text(txt)
if self._storage_browser is None:
self._storage_browser = vmmStorageBrowser(self.conn)
self._storage_browser.set_finish_cb(callback)
self._storage_browser.show(self.topwin)