def __init__(self):
     StorageListConfigScreen.__init__(self, "Add A New Storage Volume")
     self.__config = StorageVolumeConfig()
     self.__name = None
     self.__capacity = None
     self.__allocation = None
     self.__formats = None
 def __init__(self):
     StorageListConfigScreen.__init__(self, "Add A New Storage Volume")
     self.__config = StorageVolumeConfig()
     self.__name = None
     self.__capacity = None
     self.__allocation = None
     self.__formats = None
class AddVolumeConfigScreen(StorageListConfigScreen):
    def __init__(self):
        StorageListConfigScreen.__init__(self, "Add A New Storage Volume")
        self.__config = StorageVolumeConfig()
        self.__name = None
        self.__capacity = None
        self.__allocation = None
        self.__formats = None

    def get_elements_for_page(self, screen, page):
        if   page is SELECT_POOL_PAGE:
            return self.get_storage_pool_list_page(screen)
        elif page is VOLUME_NAME_PAGE:
            return self.get_volume_name_page(screen)
        elif page is VOLUME_FORMAT_PAGE:
            return self.get_volume_format_page(screen)
        elif page is MAX_CAPACITY_PAGE:
            return self.get_max_capacity_page(screen)
        elif page is CONFIRM_PAGE:
            return self.get_confirm_page(screen)

    def page_has_next(self, page):
        if page is SELECT_POOL_PAGE:
            return self.has_selectable_pools()
        else:
            if page < CONFIRM_PAGE:
                return True
        return False

    def page_has_back(self, page):
        if page > SELECT_POOL_PAGE:
            return True
        return False

    def page_has_finish(self, page):
        return page is CONFIRM_PAGE

    def get_next_page(self, page):
        if page is VOLUME_NAME_PAGE:
            if self.__config.needs_format():
                return VOLUME_FORMAT_PAGE
            else:
                return MAX_CAPACITY_PAGE
        return StorageListConfigScreen.get_next_page(self, page)

    def get_back_page(self, page):
        if page is MAX_CAPACITY_PAGE:
            if self.__config.needs_format():
                return VOLUME_FORMAT_PAGE
            else:
                return VOLUME_NAME_PAGE
        return StorageListConfigScreen.get_back_page(self, page)

    def validate_input(self, page, errors):
        if page is SELECT_POOL_PAGE:
            if self.get_selected_pool() is not None:
                return True
            else:
                errors.append("You must select a storage pool.")
        elif page is VOLUME_NAME_PAGE:
            if utils.string_is_not_blank(self.__name.value()):
                return True
            else:
                errors.append("Storage object name can only contain alphanumeric, '_', '.', or '-' characters.")
        elif page is VOLUME_FORMAT_PAGE:
            if self.__formats.current() is not None:
                return True
            else:
                errors.append("You must select a volume format.")
        elif page is MAX_CAPACITY_PAGE:
            if utils.string_is_not_blank(self.__capacity.value()):
                if utils.string_is_not_blank(self.__allocation.value()):
                    capacity = int(self.__capacity.value())
                    allocation = int(self.__allocation.value())
                    if capacity > 0:
                        if capacity <= self.__config.get_pool().info()[3] / (1024 ** 2):
                            if allocation >= 0:
                                if allocation <= capacity:
                                    return True
                                else:
                                    errors.append("Allocation cannot exceed the maximum capacity.")
                            else:
                                errors.append("The allocation must be greater than or equal to 0.")
                        else:
                            errors.append("The maximum capacity cannot exceed the storage pool size.")
                    else:
                        errors.append("The capacity must be greater than zero.")
                else:
                    errors.append("An allocation value must be entered.")
            else:
                errors.append("A maximum volume capacity must be entered.")
        elif page is CONFIRM_PAGE:
            return True
        return False

    def process_input(self, page):
        if page is SELECT_POOL_PAGE:
            self.__config.set_pool(self.get_libvirt().get_storage_pool(self.get_selected_pool()))
        elif page is VOLUME_NAME_PAGE:
            self.__config.set_name(self.__name.value())
        elif page is VOLUME_FORMAT_PAGE:
            self.__config.set_format(self.__formats.current())
        elif page is MAX_CAPACITY_PAGE:
            self.__config.set_max_capacity(int(self.__capacity.value()))
            self.__config.set_allocation(int(self.__allocation.value()))
        elif page is CONFIRM_PAGE:
            self.get_libvirt().define_storage_volume(self.__config, CreateMeter())
            self.set_finished()

    def get_volume_name_page(self, screen):
        ignore = screen
        self.__name = snack.Entry(50, self.__config.get_name())
        grid = snack.Grid(2, 1)
        grid.setField(snack.Label("Name:"), 0, 0, anchorRight=1)
        grid.setField(self.__name, 1, 0, anchorLeft=1)
        return [snack.Label("New Storage Volume"),
                grid,
                snack.Label("Name of the volume to create. File extension may be appended.")]

    def get_volume_format_page(self, screen):
        ignore = screen
        self.__formats = snack.Listbox(0)
        for fmt in self.__config.get_formats_for_pool():
            self.__formats.append(fmt, fmt)
        grid = snack.Grid(1, 1)
        grid.setField(self.__formats, 0, 0)
        return [snack.Label("Select The Volume Format"),
                grid]

    def get_max_capacity_page(self, screen):
        ignore = screen
        self.__capacity = snack.Entry(6, str(self.__config.get_max_capacity()))
        self.__allocation = snack.Entry(6, str(self.__config.get_allocation()))
        grid = snack.Grid(2, 2)
        grid.setField(snack.Label("Max. Capacity (MB):"), 0, 0, anchorRight=1)
        grid.setField(self.__capacity, 1, 0, anchorLeft=1)
        grid.setField(snack.Label("Allocation (MB):"), 0, 1, anchorRight=1)
        grid.setField(self.__allocation, 1, 1, anchorLeft=1)
        return [snack.Label("Storage Volume Quota"),
                snack.Label("%s's available space: %s" % (self.__config.get_pool().name(),
                                                    utils.size_as_mb_or_gb(self.__config.get_pool().info()[3]))),
                grid]

    def get_confirm_page(self, screen):
        ignore = screen
        grid = snack.Grid(2, 5)
        grid.setField(snack.Label("Volume Name:"), 0, 0, anchorRight=1)
        grid.setField(snack.Label("%s (%s)" % (self.__config.get_name(), self.__config.get_pool().name())), 1, 0, anchorLeft=1)
        if self.__config.needs_format():
            grid.setField(snack.Label("Format:"), 0, 1, anchorRight=1)
            grid.setField(snack.Label(self.__config.get_format()), 1, 1, anchorLeft=1)
        # NOTE: here we multiply the sizes by 1024^2 since the size_as_mb_or_gb is expect bytes
        grid.setField(snack.Label("Max. Capacity:"), 0, 2, anchorRight=1)
        grid.setField(snack.Label("%s" % (utils.size_as_mb_or_gb(self.__config.get_max_capacity() * (1024 ** 2)))), 1, 2, anchorLeft=1)
        grid.setField(snack.Label("Allocation:"), 0, 3, anchorRight=1)
        grid.setField(snack.Label("%s" % (utils.size_as_mb_or_gb(self.__config.get_allocation() * (1024 ** 2)))), 1, 3, anchorLeft=1)
        return [snack.Label("Ready To Allocation New Storage Volume"),
                grid]
class AddVolumeConfigScreen(StorageListConfigScreen):
    def __init__(self):
        StorageListConfigScreen.__init__(self, "Add A New Storage Volume")
        self.__config = StorageVolumeConfig()
        self.__name = None
        self.__capacity = None
        self.__allocation = None
        self.__formats = None

    def get_elements_for_page(self, screen, page):
        if page is SELECT_POOL_PAGE:
            return self.get_storage_pool_list_page(screen)
        elif page is VOLUME_NAME_PAGE:
            return self.get_volume_name_page(screen)
        elif page is VOLUME_FORMAT_PAGE:
            return self.get_volume_format_page(screen)
        elif page is MAX_CAPACITY_PAGE:
            return self.get_max_capacity_page(screen)
        elif page is CONFIRM_PAGE:
            return self.get_confirm_page(screen)

    def page_has_next(self, page):
        if page is SELECT_POOL_PAGE:
            return self.has_selectable_pools()
        else:
            if page < CONFIRM_PAGE:
                return True
        return False

    def page_has_back(self, page):
        if page > SELECT_POOL_PAGE:
            return True
        return False

    def page_has_finish(self, page):
        return page is CONFIRM_PAGE

    def get_next_page(self, page):
        if page is VOLUME_NAME_PAGE:
            if self.__config.needs_format():
                return VOLUME_FORMAT_PAGE
            else:
                return MAX_CAPACITY_PAGE
        return StorageListConfigScreen.get_next_page(self, page)

    def get_back_page(self, page):
        if page is MAX_CAPACITY_PAGE:
            if self.__config.needs_format():
                return VOLUME_FORMAT_PAGE
            else:
                return VOLUME_NAME_PAGE
        return StorageListConfigScreen.get_back_page(self, page)

    def validate_input(self, page, errors):
        if page is SELECT_POOL_PAGE:
            if self.get_selected_pool() is not None:
                return True
            else:
                errors.append("You must select a storage pool.")
        elif page is VOLUME_NAME_PAGE:
            if utils.string_is_not_blank(self.__name.value()):
                return True
            else:
                errors.append(
                    "Storage object name can only contain alphanumeric, '_', '.', or '-' characters."
                )
        elif page is VOLUME_FORMAT_PAGE:
            if self.__formats.current() is not None:
                return True
            else:
                errors.append("You must select a volume format.")
        elif page is MAX_CAPACITY_PAGE:
            if utils.string_is_not_blank(self.__capacity.value()):
                if utils.string_is_not_blank(self.__allocation.value()):
                    capacity = int(self.__capacity.value())
                    allocation = int(self.__allocation.value())
                    if capacity > 0:
                        if capacity <= self.__config.get_pool().info()[3] / (
                                1024**2):
                            if allocation >= 0:
                                if allocation <= capacity:
                                    return True
                                else:
                                    errors.append(
                                        "Allocation cannot exceed the maximum capacity."
                                    )
                            else:
                                errors.append(
                                    "The allocation must be greater than or equal to 0."
                                )
                        else:
                            errors.append(
                                "The maximum capacity cannot exceed the storage pool size."
                            )
                    else:
                        errors.append(
                            "The capacity must be greater than zero.")
                else:
                    errors.append("An allocation value must be entered.")
            else:
                errors.append("A maximum volume capacity must be entered.")
        elif page is CONFIRM_PAGE:
            return True
        return False

    def process_input(self, page):
        if page is SELECT_POOL_PAGE:
            self.__config.set_pool(self.get_libvirt().get_storage_pool(
                self.get_selected_pool()))
        elif page is VOLUME_NAME_PAGE:
            self.__config.set_name(self.__name.value())
        elif page is VOLUME_FORMAT_PAGE:
            self.__config.set_format(self.__formats.current())
        elif page is MAX_CAPACITY_PAGE:
            self.__config.set_max_capacity(int(self.__capacity.value()))
            self.__config.set_allocation(int(self.__allocation.value()))
        elif page is CONFIRM_PAGE:
            self.get_libvirt().define_storage_volume(self.__config,
                                                     CreateMeter())
            self.set_finished()

    def get_volume_name_page(self, screen):
        ignore = screen
        self.__name = snack.Entry(50, self.__config.get_name())
        grid = snack.Grid(2, 1)
        grid.setField(snack.Label("Name:"), 0, 0, anchorRight=1)
        grid.setField(self.__name, 1, 0, anchorLeft=1)
        return [
            snack.Label("New Storage Volume"), grid,
            snack.Label(
                "Name of the volume to create. File extension may be appended."
            )
        ]

    def get_volume_format_page(self, screen):
        ignore = screen
        self.__formats = snack.Listbox(0)
        for fmt in self.__config.get_formats_for_pool():
            self.__formats.append(fmt, fmt)
        grid = snack.Grid(1, 1)
        grid.setField(self.__formats, 0, 0)
        return [snack.Label("Select The Volume Format"), grid]

    def get_max_capacity_page(self, screen):
        ignore = screen
        self.__capacity = snack.Entry(6, str(self.__config.get_max_capacity()))
        self.__allocation = snack.Entry(6, str(self.__config.get_allocation()))
        grid = snack.Grid(2, 2)
        grid.setField(snack.Label("Max. Capacity (MB):"), 0, 0, anchorRight=1)
        grid.setField(self.__capacity, 1, 0, anchorLeft=1)
        grid.setField(snack.Label("Allocation (MB):"), 0, 1, anchorRight=1)
        grid.setField(self.__allocation, 1, 1, anchorLeft=1)
        return [
            snack.Label("Storage Volume Quota"),
            snack.Label(
                "%s's available space: %s" %
                (self.__config.get_pool().name(),
                 utils.size_as_mb_or_gb(self.__config.get_pool().info()[3]))),
            grid
        ]

    def get_confirm_page(self, screen):
        ignore = screen
        grid = snack.Grid(2, 5)
        grid.setField(snack.Label("Volume Name:"), 0, 0, anchorRight=1)
        grid.setField(snack.Label(
            "%s (%s)" %
            (self.__config.get_name(), self.__config.get_pool().name())),
                      1,
                      0,
                      anchorLeft=1)
        if self.__config.needs_format():
            grid.setField(snack.Label("Format:"), 0, 1, anchorRight=1)
            grid.setField(snack.Label(self.__config.get_format()),
                          1,
                          1,
                          anchorLeft=1)
        # NOTE: here we multiply the sizes by 1024^2 since the size_as_mb_or_gb is expect bytes
        grid.setField(snack.Label("Max. Capacity:"), 0, 2, anchorRight=1)
        grid.setField(snack.Label(
            "%s" % (utils.size_as_mb_or_gb(self.__config.get_max_capacity() *
                                           (1024**2)))),
                      1,
                      2,
                      anchorLeft=1)
        grid.setField(snack.Label("Allocation:"), 0, 3, anchorRight=1)
        grid.setField(snack.Label(
            "%s" % (utils.size_as_mb_or_gb(self.__config.get_allocation() *
                                           (1024**2)))),
                      1,
                      3,
                      anchorLeft=1)
        return [snack.Label("Ready To Allocation New Storage Volume"), grid]
Esempio n. 5
0
 def __init__(self):
     StorageListConfigScreen.__init__(self, "Add A New Storage Volume")
     self.__config = StorageVolumeConfig()
Esempio n. 6
0
 def __init__(self):
     StorageListConfigScreen.__init__(self, "Add A New Storage Volume")
     self.__config = StorageVolumeConfig()