Files
gns3-gui/gns3/modules/virtualbox/dialogs/virtualbox_vm_wizard.py
2015-11-09 12:52:57 +01:00

115 lines
3.7 KiB
Python

# -*- coding: utf-8 -*-
#
# Copyright (C) 2014 GNS3 Technologies Inc.
#
# 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 3 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, see <http://www.gnu.org/licenses/>.
"""
Wizard for VirtualBox VMs.
"""
from gns3.qt import QtGui, QtWidgets
from gns3.servers import Servers
from gns3.dialogs.vm_wizard import VMWizard
from ..ui.virtualbox_vm_wizard_ui import Ui_VirtualBoxVMWizard
from .. import VirtualBox
class VirtualBoxVMWizard(VMWizard, Ui_VirtualBoxVMWizard):
"""
Wizard to create a VirtualBox VM.
:param virtualbox_vms: existing VirtualBox VMs
:param parent: parent widget
"""
def __init__(self, virtualbox_vms, parent):
super().__init__(virtualbox_vms, VirtualBox.instance().settings()["use_local_server"], parent)
self._virtualbox_vms = virtualbox_vms
self.setPixmap(QtWidgets.QWizard.LogoPixmap, QtGui.QPixmap(":/icons/virtualbox.png"))
if not Servers.instance().remoteServers():
# skip the server page if we use the local server
self.setStartId(1)
def validateCurrentPage(self):
"""
Validates the server.
"""
if super().validateCurrentPage() is False:
return False
if self.currentPage() == self.uiVirtualBoxWizardPage:
if not self.uiVMListComboBox.count():
QtWidgets.QMessageBox.critical(self, "VirtualBox VMs", "There is no VirtualBox VM available!")
return False
return True
def initializePage(self, page_id):
super().initializePage(page_id)
if self.page(page_id) == self.uiVirtualBoxWizardPage:
self.uiVMListComboBox.clear()
self._server.get("/virtualbox/vms", self._getVirtualBoxVMsFromServerCallback, progressText="Listing VirtualBox VMs...")
def _getVirtualBoxVMsFromServerCallback(self, result, error=False, **kwargs):
"""
Callback for getVirtualBoxVMsFromServer.
:param progress_dialog: QProgressDialog instance
:param result: server response
:param error: indicates an error (boolean)
"""
if error:
QtWidgets.QMessageBox.critical(self, "VirtualBox VMs", "{}".format(result["message"]))
else:
self.uiVMListComboBox.clear()
existing_vms = []
for existing_vm in self._virtualbox_vms.values():
existing_vms.append(existing_vm["vmname"])
for vm in result:
if vm["vmname"] not in existing_vms:
self.uiVMListComboBox.addItem(vm["vmname"], vm)
def getSettings(self):
"""
Returns the settings set in this Wizard.
:return: settings dict
"""
if self.uiLocalRadioButton.isChecked():
server = "local"
else:
server = self.uiRemoteServersComboBox.currentText()
index = self.uiVMListComboBox.currentIndex()
vmname = self.uiVMListComboBox.itemText(index)
vminfo = self.uiVMListComboBox.itemData(index)
settings = {
"vmname": vmname,
"server": server,
"ram": vminfo["ram"],
"linked_base": self.uiBaseVMCheckBox.isChecked()
}
return settings