virt-manager/virtManager/storagebrowse.py

163 lines
5.0 KiB
Python

# Copyright (C) 2009, 2013, 2014 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.
from virtinst import log
from .lib import uiutil
from .baseclass import vmmGObjectUI
from .hoststorage import vmmHostStorage
class vmmStorageBrowser(vmmGObjectUI):
def __init__(self, conn):
vmmGObjectUI.__init__(self, "storagebrowse.ui", "vmm-storage-browse")
self.conn = conn
self._first_run = False
self._finish_cb = None
# Passed to browse_local
self._browse_reason = None
self.storagelist = vmmHostStorage(self.conn, self.builder, self.topwin,
self._vol_sensitive_cb)
self._init_ui()
self.builder.connect_signals({
"on_vmm_storage_browse_delete_event": self.close,
})
self.bind_escape_key_close()
def show(self, parent):
log.debug("Showing storage browser")
if not self._first_run:
self._first_run = True
pool = self.conn.get_default_pool()
uiutil.set_list_selection(
self.storagelist.widget("pool-list"), pool)
self.topwin.set_transient_for(parent)
self.topwin.present()
self.conn.schedule_priority_tick(pollpool=True)
def close(self, ignore1=None, ignore2=None):
if self.is_visible():
log.debug("Closing storage browser")
self.topwin.hide()
self.storagelist.close()
return 1
def _cleanup(self):
self.conn = None
self.storagelist.cleanup()
self.storagelist = None
###########
# UI init #
###########
def _init_ui(self):
self.storagelist.connect("browse-clicked", self._browse_clicked)
self.storagelist.connect("volume-chosen", self._volume_chosen)
self.storagelist.connect("cancel-clicked", self.close)
self.widget("storage-align").add(self.storagelist.top_box)
self.err.set_modal_default(True)
self.storagelist.err.set_modal_default(True)
tooltip = ""
is_remote = self.conn.is_remote()
self.storagelist.widget("browse-local").set_sensitive(not is_remote)
if is_remote:
tooltip = _("Cannot use local storage on remote connection.")
self.storagelist.widget("browse-local").set_tooltip_text(tooltip)
uiutil.set_grid_row_visible(
self.storagelist.widget("pool-autostart"), False)
uiutil.set_grid_row_visible(
self.storagelist.widget("pool-name-entry"), False)
uiutil.set_grid_row_visible(
self.storagelist.widget("pool-state-box"), False)
self.storagelist.widget("browse-local").set_visible(True)
self.storagelist.widget("browse-cancel").set_visible(True)
self.storagelist.widget("choose-volume").set_visible(True)
self.storagelist.widget("choose-volume").set_sensitive(False)
self.storagelist.widget("pool-apply").set_visible(False)
self.set_browse_reason(self._browse_reason)
##############
# Public API #
##############
def set_finish_cb(self, callback):
self._finish_cb = callback
def set_vm_name(self, name):
self.storagelist.set_name_hint(name)
def set_browse_reason(self, reason):
self._browse_reason = reason
data = self.config.browse_reason_data.get(self._browse_reason, {})
allow_create = True
title = _("Choose Storage Volume")
if data:
allow_create = data["enable_create"]
title = data["storage_title"]
self.topwin.set_title(title)
self.storagelist.widget("vol-add").set_sensitive(allow_create)
#############
# Listeners #
#############
def _browse_clicked(self, src):
ignore = src
return self._browse_local()
def _volume_chosen(self, src, volume):
ignore = src
log.debug("Chosen volume XML:\n%s", volume.xmlobj.get_xml())
self._finish(volume.get_target_path())
def _vol_sensitive_cb(self, fmt):
if ((self._browse_reason == self.config.CONFIG_DIR_FS) and
fmt != 'dir'):
return False
return True
####################
# Internal helpers #
####################
def _browse_local(self):
dialog_type = None
dialog_name = None
choose_button = None
data = self.config.browse_reason_data.get(self._browse_reason)
if data:
dialog_name = data["local_title"] or None
dialog_type = data.get("dialog_type")
choose_button = data.get("choose_button")
filename = self.err.browse_local(self.conn,
dialog_type=dialog_type, browse_reason=self._browse_reason,
dialog_name=dialog_name, choose_button=choose_button)
if filename:
log.debug("Browse local chose path=%s", filename)
self._finish(filename)
def _finish(self, path):
if self._finish_cb:
self._finish_cb(self, path)
self.close()