simono41/compose-viz
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat: add device graph

Browse source
This commit is contained in:
Chuan Ou Yang 2022-06-08 23:25:52 +08:00
parent 198a8aada0
commit bdb227b1d8
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
compose_viz/graph.py
View file

@ -28,6 +28,9 @@ def apply_vertex_style(type) -> dict:
"cgroup": { "cgroup": {
"shape": "diamond", "shape": "diamond",
}, },
"device": {
"shape": "box3d",
},
} }
return style[type] return style[type]
@ -35,7 +38,7 @@ def apply_vertex_style(type) -> dict:
def apply_edge_style(type) -> dict: def apply_edge_style(type) -> dict:
style = { style = {
"expose": { "exposes": {
"style": "solid", "style": "solid",
"dir": "both", "dir": "both",
}, },
@ -103,7 +106,7 @@ class Graph:
self.add_edge(service.name, volume.source, "volumes", lable=volume.target) self.add_edge(service.name, volume.source, "volumes", lable=volume.target)
for expose in service.expose: for expose in service.expose:
self.add_vertex(expose, "port") self.add_vertex(expose, "port")
self.add_edge(expose, service.name, "expose") self.add_edge(expose, service.name, "exposes")
for port in service.ports: for port in service.ports:
self.add_vertex(port.host_port, "port", lable=port.host_port) self.add_vertex(port.host_port, "port", lable=port.host_port)
self.add_edge(port.host_port, service.name, "links", lable=port.container_port) self.add_edge(port.host_port, service.name, "links", lable=port.container_port)
@ -121,5 +124,10 @@ class Graph:
for porfile in service.profiles: for porfile in service.profiles:
self.add_vertex(porfile, "porfile") self.add_vertex(porfile, "porfile")
self.add_edge(porfile, service.name, "links") self.add_edge(porfile, service.name, "links")
for device in service.devices:
self.add_vertex(device.host_path, "device")
self.add_edge(
device.host_path, service.name, "exposes", f"{device.container_path}\n({device.cgroup_permissions})"
)
self.dot.render(outfile=f"{self.filename}.{format}", format=format, cleanup=cleanup) self.dot.render(outfile=f"{self.filename}.{format}", format=format, cleanup=cleanup)