Cole Robinson 0d243983d0 Revive pep8 and clean up the code
autopep8 is pretty cool :)
2013-04-13 15:22:43 -04:00

99 lines
3.4 KiB
Python

#
# removepool.py - Copyright (C) 2009 Red Hat, Inc.
# Written by Darryl L. Pierce <dpierce@redhat.com>
#
# 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; version 2 of the License.
#
# 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. A copy of the GNU General Public License is
# also available at http://www.gnu.org/copyleft/gpl.html.
from snack import Checkbox
from snack import Label
from virtManagerTui.storagelistconfigscreen import StorageListConfigScreen
LIST_POOLS_PAGE = 1
CONFIRM_PAGE = 2
FINISHED_REMOVING = 3
class RemoveStoragePoolConfigScreen(StorageListConfigScreen):
def __init__(self):
StorageListConfigScreen.__init__(self, "Remove A Storage Pool")
self.__confirm = None
def get_elements_for_page(self, screen, page):
if page is LIST_POOLS_PAGE:
return self.get_storage_pool_list_page(screen)
elif page is CONFIRM_PAGE:
return self.get_confirm_page(screen)
elif page is FINISHED_REMOVING:
return self.get_finished_removing_page(screen)
def page_has_next(self, page):
if page is LIST_POOLS_PAGE and self.has_selectable_pools():
return True
elif page is CONFIRM_PAGE:
return True
return False
def page_has_finish(self, page):
return page is FINISHED_REMOVING
def page_has_back(self, page):
ignore = page
return False
def validate_input(self, page, errors):
if page is LIST_POOLS_PAGE:
if self.get_selected_pool() is not None:
return True
else:
errors.append("Please select a storage pool to be removed.")
elif page is CONFIRM_PAGE:
if self.__confirm.value():
return True
else:
errors.append("You must confirm removing a storage pool.")
return False
def process_input(self, page):
if page is CONFIRM_PAGE:
try:
self.get_libvirt().destroy_storage_pool(self.get_selected_pool())
self.get_libvirt().undefine_storage_pool(self.get_selected_pool())
except Exception:
pass
def get_confirm_page(self, screen):
ignore = screen
self.__confirm = Checkbox("Check here to confirm deleting pool: %s" % self.get_selected_pool())
fields = []
fields.append((None, self.__confirm))
return [Label("Remove Selected Storage Pool"),
self.create_grid_from_fields(fields)]
def get_finished_removing_page(self, page):
ignore = page
self.set_finished()
pool = self.get_selected_pool()
state = ""
if self.get_libvirt().storage_pool_exists(pool):
state = "was not "
return [Label("Storage pool '%s' %sdeleted." % (pool, state))]
def RemoveStoragePool():
screen = RemoveStoragePoolConfigScreen()
screen.start()