Added diagram background color config file option

beveradb · Mar 7, 2024 · 61b3ea3 · 61b3ea3
1 parent b3c8c2e
commit 61b3ea3
Show file tree

Hide file tree

Showing 8 changed files with 21 additions and 8 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -10,10 +10,11 @@ RUN fc-cache -fv
 # Set the working directory
 WORKDIR /app
 
-# Install the logo-diagram-generator package
-RUN pip install logo-diagram-generator
+# Install package globally with pip
+COPY . /app
+RUN pip install .
 
-# Configure docker to run the logo-diagram-generator
+# Configure docker to run the logo-diagram-generator command
 ENTRYPOINT ["logo-diagram-generator"]
 
 # Set the default args to pass if none are specified

diff --git a/examples/full.example.yml b/examples/full.example.yml
@@ -1,5 +1,6 @@
 ecosystem:
   style:
+    diagramBackgroundColor: "#ffffff"
     diagramEngine: neato
     diagramOverlap: prism
     diagramOverlapScaling: 4

diff --git a/examples/full.example_logos.svg b/examples/full.example_logos.svg
diff --git a/examples/full.example_text.svg b/examples/full.example_text.svg
diff --git a/examples/minimal.example_logos.svg b/examples/minimal.example_logos.svg
diff --git a/examples/minimal.example_text.svg b/examples/minimal.example_text.svg
diff --git a/logo_diagram_generator/generate_diagram.py b/logo_diagram_generator/generate_diagram.py
@@ -20,6 +20,7 @@ def generate_text_only_svg_diagram_from_config(config, diagram_name, output_svg_
     diagram_overlap_shrink = str(ecosystem_style.get("diagramOverlapShrink", "true"))
     diagram_padding = str(ecosystem_style.get("diagramPadding", "0.5"))
     diagram_rankdir = ecosystem_style.get("diagramRankdir", "TB")
+    diagram_background_color = ecosystem_style.get("diagramBackgroundColor", "#ffffff")
 
     group_label_shape = ecosystem_style.get("groupLabelShape", "box")
     group_label_style = ecosystem_style.get("groupLabelStyle", "rounded")
@@ -35,6 +36,16 @@ def generate_text_only_svg_diagram_from_config(config, diagram_name, output_svg_
     dot.attr(rankdir=diagram_rankdir)
     dot.attr(pad=diagram_padding)
     dot.attr(id=diagram_name)
+    dot.attr(bgcolor=diagram_background_color)
+
+    logging.info(
+        f"Diagram attributes set: engine={diagram_engine}, overlap={diagram_overlap}, "
+        f"overlap_scaling={diagram_overlap_scaling}, overlap_shrink={diagram_overlap_shrink}, "
+        f"rankdir={diagram_rankdir}, padding={diagram_padding}, background_color={diagram_background_color}, "
+        f"group_label_shape={group_label_shape}, group_label_style={group_label_style}, "
+        f"group_label_fontname={group_label_fontname}, group_label_fontcolor={group_label_fontcolor}, "
+        f"group_label_fontsize={group_label_fontsize}, group_label_margin={group_label_margin}"
+    )
 
     central_tool = config["ecosystem"]["centralTool"]
     central_tool_name = central_tool["name"]

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "logo-diagram-generator"
-version = "0.2.2"
+version = "0.2.3"
 description = "Generate SVG diagrams of a (tech) ecosystem, using logos from each tool organised into groups around a central logo"
 authors = ["Andrew Beveridge <[email protected]>"]
 license = "MIT"