added path for creating non core container based nodes

This commit is contained in:
bharnden 2019-11-15 13:09:53 -08:00
parent f1b7dd3824
commit 7981340b13
4 changed files with 18 additions and 12 deletions

Binary file not shown.

After

Width:  |  Height:  |  Size: 719 B

BIN
coretk/coretk/icons/lxc.gif Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 724 B

View file

@ -107,3 +107,5 @@ class ImageEnum(Enum):
FILEOPEN = "fileopen" FILEOPEN = "fileopen"
EDITDELETE = "edit-delete" EDITDELETE = "edit-delete"
ANTENNA = "antenna" ANTENNA = "antenna"
DOCKER = "docker"
LXC = "lxc"

View file

@ -130,28 +130,32 @@ class Toolbar(ttk.Frame):
self.hide_pickers() self.hide_pickers()
self.node_picker = ttk.Frame(self.master) self.node_picker = ttk.Frame(self.master)
nodes = [ nodes = [
(ImageEnum.ROUTER, "router"), (ImageEnum.ROUTER, core_pb2.NodeType.DEFAULT, "router"),
(ImageEnum.HOST, "host"), (ImageEnum.HOST, core_pb2.NodeType.DEFAULT, "host"),
(ImageEnum.PC, "PC"), (ImageEnum.PC, core_pb2.NodeType.DEFAULT, "PC"),
(ImageEnum.MDR, "mdr"), (ImageEnum.MDR, core_pb2.NodeType.DEFAULT, "mdr"),
(ImageEnum.PROUTER, "prouter"), (ImageEnum.PROUTER, core_pb2.NodeType.DEFAULT, "prouter"),
(ImageEnum.DOCKER, core_pb2.NodeType.DOCKER, "Docker"),
(ImageEnum.LXC, core_pb2.NodeType.LXC, "LXC"),
] ]
# draw default nodes # draw default nodes
for image_enum, model in nodes: for image_enum, node_type, model in nodes:
image = icon(image_enum) image = icon(image_enum)
func = partial( func = partial(
self.update_button, self.update_button, self.node_button, image, node_type, model
self.node_button,
image,
core_pb2.NodeType.DEFAULT,
model,
) )
self.create_picker_button(image, func, self.node_picker, model) self.create_picker_button(image, func, self.node_picker, model)
# draw custom nodes # draw custom nodes
for name in sorted(self.app.core.custom_nodes): for name in sorted(self.app.core.custom_nodes):
custom_node = self.app.core.custom_nodes[name] custom_node = self.app.core.custom_nodes[name]
image = custom_node.image image = custom_node.image
func = partial(self.update_button, self.node_button, image, name) func = partial(
self.update_button,
self.node_button,
image,
core_pb2.NodeType.DEFAULT,
name,
)
self.create_picker_button(image, func, self.node_picker, name) self.create_picker_button(image, func, self.node_picker, name)
# draw edit node # draw edit node
image = icon(ImageEnum.EDITNODE) image = icon(ImageEnum.EDITNODE)