From 006fdba67748ae8dd792fa29b08ec0c9c4db27b4 Mon Sep 17 00:00:00 2001
From: Daniel Rojas <github@danielrojas.net>
Date: Fri, 22 May 2020 13:08:54 +0200
Subject: [PATCH] Make cable name hidden by default

---
 src/test.py    |  2 +-
 src/wireviz.py | 10 ++++++++--
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/test.py b/src/test.py
index d4af6a6..c44965e 100644
--- a/src/test.py
+++ b/src/test.py
@@ -6,7 +6,7 @@ COLORS_WEIRD = ("infrared","ultraviolet","transparent","invisible")
 # example 1
 X1 = wireviz.Node("X1", pinout=PINOUT_SERIAL, ports_right=True)
 X2 = wireviz.Node("X2", num_pins=6, ports_left=True)
-W1 = wireviz.Cable("W1", num_wires=3, color_code="DIN", shield=True)
+W1 = wireviz.Cable("W1", show_name=True, num_wires=3, color_code="DIN", shield=True)
 W1.connect(X1,(2,3,5),(1,2,3),X2,(1,3,2))
 X2.loop(5,6)
 objects = [X1, X2, W1]
diff --git a/src/wireviz.py b/src/wireviz.py
index 852e611..a93252f 100644
--- a/src/wireviz.py
+++ b/src/wireviz.py
@@ -90,8 +90,9 @@ class Node:
 
 class Cable:
 
-    def __init__(self, name, num_wires=None, colors=None, color_code=None, shield=False):
+    def __init__(self, name, show_name=False, num_wires=None, colors=None, color_code=None, shield=False):
         self.name = name
+        self.show_name = show_name
         self.connections = []
         if color_code is None and colors is None:
             self.colors = ("",) * num_wires
@@ -139,7 +140,12 @@ class Cable:
     def graphviz(self):
         s = ''
         # print header
-        s = s + '{name}[label="{name} | {{'.format(name=self.name)
+        s = s + '{name}[label="'.format(name=self.name)
+
+        if self.show_name == True:
+            s = s + '{name} | '.format(name=self.name)
+
+        s = s + '{'
         # print pinout
         s = s + '{'
         l = []