def build_widgets(self):
if self.charm_class.allow_multi_units:
machine_string = "machines"
plural_string = "s"
else:
machine_string = "a machine"
plural_string = ""
instructions = Text("Select {} to host {}".format(
machine_string, self.charm_class.display_name))
self.service_widget = ServiceWidget(self.charm_class,
self.controller,
show_constraints=True,
show_placements=True)
all_actions = [(AssignmentType.BareMetal,
'Add as Bare Metal',
self.do_select_baremetal),
(AssignmentType.LXC,
'Add as LXC', self.do_select_lxc),
(AssignmentType.KVM,
'Add as KVM', self.do_select_kvm)]
actions = [(label, func) for atype, label, func in all_actions
if atype in self.charm_class.allowed_assignment_types]
constraints = self.charm_class.constraints
# NOTE: show_assignments=False is a WORKAROUND for #194
self.machines_list = MachinesList(self.controller,
actions,
constraints=constraints,
show_hardware=True,
show_assignments=False)
self.machines_list.update()
close_button = AttrMap(Button('X',
on_press=self.close_pressed),
'button_secondary', 'button_secondary focus')
p = Pile([GridFlow([close_button], 5, 1, 0, 'right'),
instructions, Divider(), self.service_widget,
Divider(), self.machines_list])
return LineBox(p, title="Select Machine{}".format(plural_string))
class MachineChooser(WidgetWrap):
"""Presents list of machines to assign a service to.
Supports multiple selection if the service does.
"""
def __init__(self, controller, charm_class, parent_widget):
self.controller = controller
self.charm_class = charm_class
self.parent_widget = parent_widget
w = self.build_widgets()
super().__init__(w)
def build_widgets(self):
if self.charm_class.allow_multi_units:
machine_string = "machines"
plural_string = "s"
else:
machine_string = "a machine"
plural_string = ""
instructions = Text("Select {} to host {}".format(
machine_string, self.charm_class.display_name))
self.service_widget = ServiceWidget(self.charm_class,
self.controller,
show_constraints=True,
show_placements=True)
all_actions = [(AssignmentType.BareMetal,
'Add as Bare Metal',
self.do_select_baremetal),
(AssignmentType.LXC,
'Add as LXC', self.do_select_lxc),
(AssignmentType.KVM,
'Add as KVM', self.do_select_kvm)]
actions = [(label, func) for atype, label, func in all_actions
if atype in self.charm_class.allowed_assignment_types]
constraints = self.charm_class.constraints
# NOTE: show_assignments=False is a WORKAROUND for #194
self.machines_list = MachinesList(self.controller,
actions,
constraints=constraints,
show_hardware=True,
show_assignments=False)
self.machines_list.update()
close_button = AttrMap(Button('X',
on_press=self.close_pressed),
'button_secondary', 'button_secondary focus')
p = Pile([GridFlow([close_button], 5, 1, 0, 'right'),
instructions, Divider(), self.service_widget,
Divider(), self.machines_list])
return LineBox(p, title="Select Machine{}".format(plural_string))
def do_select_baremetal(self, sender, machine):
self.do_select(sender, machine, AssignmentType.BareMetal)
def do_select_lxc(self, sender, machine):
self.do_select(sender, machine, AssignmentType.LXC)
def do_select_kvm(self, sender, machine):
self.do_select(sender, machine, AssignmentType.KVM)
def do_select(self, sender, machine, atype):
self.controller.assign(machine, self.charm_class, atype)
self.machines_list.update()
self.service_widget.update()
self.parent_widget.remove_overlay(self)
def close_pressed(self, sender):
self.parent_widget.remove_overlay(self)
