github/KubeDiagrams
1
0
Fork 0
mirror of https://github.com/philippemerle/KubeDiagrams.git synced 2026-08-22 14:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

extend functions to suport custom object icons

Browse Source 
- def create_custom_cluster:

  easy fix to allow icons on custom cluster definitions by using
  an additional "type" key/attribute that refers to a diagram class

  allowing custom icons seems a bit more tricky as it may require a
  refactoring of the icon function. maybe this can also be added to
  have a consistent behavior...?

- create_custom_node:

  custom nodes already support icons by specifying a diagram class
  via "type" key/attribute

  in addition, it is now possible to specify a custom icon using the
  new key/attribute "icon"
This commit is contained in:
Michael Lucka
2025-09-12 11:37:40 +02:00
parent f3b087334c
commit e4868de5c4
1 changed files with 33 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bin/kube-diagrams
+33 -15
View File
@@ -1352,28 +1352,46 @@ def create_custom_node(node_id, node_def):
"""
Create a custom node.
"""
diagram_node_class = ManagedServices # default diagram node class
# Get diagram node class name
diagram_node_classname = node_def.get("type")
if diagram_node_classname is not None: # classname defined
# Import Diagrams node class module
idx = diagram_node_classname.rfind('.')
if idx != -1:
module = importlib.import_module(diagram_node_classname[:idx])
# Get diagram node class
diagram_node_class = getattr(module, diagram_node_classname[idx+1:])
node_icon = node_def.get("icon")
node_label = split_node_label(node_def.get("name", ""))
diagram_nodes[node_id] = diagram_node_class(node_label, tooltip=node_label)
if node_icon is not None: # icon defined
diagram_nodes[node_id] = Custom(
node_label,
os.path.abspath(node_icon.replace("$KD", DIRNAME)),
tooltip=node_label
)
else:
diagram_node_class = ManagedServices # default diagram node class
# Get diagram node class name
diagram_node_classname = node_def.get("type")
if diagram_node_classname is not None: # classname defined
# Import Diagrams node class module
idx = diagram_node_classname.rfind('.')
if idx != -1:
module = importlib.import_module(diagram_node_classname[:idx])
# Get diagram node class
diagram_node_class = getattr(module, diagram_node_classname[idx+1:])
diagram_nodes[node_id] = diagram_node_class(node_label, tooltip=node_label)
def create_custom_cluster(cluster_id, cluster_def):
"""
Create a custom cluster.
"""
cluster_name = query_path(cluster_def, "name")
with Cluster(cluster_name, graph_attr={
"tooltip": cluster_name,
**cluster_def.get("graph_attr", {})
}):
cluster_type = query_path(cluster_def, "type")
# Build base graph_attr
graph_attr = {
"tooltip": cluster_name,
**cluster_def.get("graph_attr", {}),
}
if cluster_type:
# Dynamically import diagrams node class
idx = cluster_type.rfind(".")
if idx != -1:
module = importlib.import_module(cluster_type[:idx])
diagram_class = getattr(module, cluster_type[idx+1:])
graph_attr["label"] = icon(diagram_class, cluster_name)
with Cluster(cluster_name, graph_attr=graph_attr):
create_custom_clusters_nodes(cluster_id, cluster_def)
def create_custom_clusters_nodes(container_id, container_def):