radiochart 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

radiochart/cli.py

@@ -2,21 +2,49 @@ import argparse
 import json
 from graphviz import Digraph
 import os
+import itertools
+
+# Predefined colors for squads
+SQUAD_COLORS = [
+    "lightblue",
+    "lightgreen",
+    "lightyellow",
+    "lightpink",
+    "lightcyan",
+    "lightsalmon",
+    "lightgray"
+]
 
 def load_json(filepath):
     with open(filepath, 'r') as f:
         return json.load(f)
 
-def build_graph(tree, graph=None, parent=None):
+def is_squad(name):
+    return "Squad" in name
+
+def build_graph(tree, graph=None, parent=None, squad_color=None, color_cycle=None):
     if graph is None:
         graph = Digraph(format='png')
-        graph.attr('node', shape='box', style='filled', fillcolor='lightgrey', fontname='Arial')
+        # Only set default shape and font globally — no fillcolor here!
+        graph.attr('node', shape='box', fontname='Arial')
+
+    if color_cycle is None:
+        color_cycle = itertools.cycle(SQUAD_COLORS)
 
     for node, children in tree.items():
-        graph.node(node)
+        # Assign color for squads and their children
+        if is_squad(node):
+            squad_color = next(color_cycle)
+
+        fillcolor = squad_color if squad_color else "white"
+        # Set fillcolor and style per node (style='filled' enables fillcolor)
+        graph.node(node, style='filled', fillcolor=fillcolor)
+
         if parent:
             graph.edge(parent, node)
-        build_graph(children, graph, node)
+
+        # Pass the squad color down to children if applicable
+        build_graph(children, graph, node, squad_color, color_cycle)
 
     return graph
 
@@ -32,3 +60,6 @@ def main():
     output_path = os.path.splitext(args.output)[0]
     graph.render(output_path, cleanup=True)
     print(f"✅ Chart saved as {output_path}.png")
+
+if __name__ == "__main__":
+    main()

{radiochart-0.1.0.dist-info → radiochart-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: radiochart
-Version: 0.1.0
+Version: 0.2.0
 Summary: CLI tool to generate Arma 3 radio frequency hierarchy charts
 Author-email: KingCharlesVII <viikingcharles@gmail.com>
 License: MIT

radiochart-0.2.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+radiochart/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+radiochart/cli.py,sha256=Hw9fbVl17eUgsI-u9wB8kFrfiZoBfwnuM8NUEle6UnQ,1970
+radiochart-0.2.0.dist-info/METADATA,sha256=2XG1FVda0OyBr68-uZzq04CUjnpzxq7RKDeqGNSLjZQ,312
+radiochart-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+radiochart-0.2.0.dist-info/entry_points.txt,sha256=0YgYH_Sc1VwuKgcS1Vum6VAnn1e4HUyE9JLOHEYEB9w,51
+radiochart-0.2.0.dist-info/top_level.txt,sha256=Q1BACtGLsnWStOzjTlXfurlbIBu5Wx6sbP-zZtLnXj8,11
+radiochart-0.2.0.dist-info/RECORD,,

radiochart-0.1.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-radiochart/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-radiochart/cli.py,sha256=Z7-3E_D3cs2QhtY6wzlHCNJdyPF4IApsZoQ9ntG_OQ0,1119
-radiochart-0.1.0.dist-info/METADATA,sha256=wOnW7hqtaTZBSkF9OF_tlQMBxsjUPUsKhCrQUR2Jp4Q,312
-radiochart-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-radiochart-0.1.0.dist-info/entry_points.txt,sha256=0YgYH_Sc1VwuKgcS1Vum6VAnn1e4HUyE9JLOHEYEB9w,51
-radiochart-0.1.0.dist-info/top_level.txt,sha256=Q1BACtGLsnWStOzjTlXfurlbIBu5Wx6sbP-zZtLnXj8,11
-radiochart-0.1.0.dist-info/RECORD,,