From 7e1b23878d97a14c7643657bf372a922da2633e9 Mon Sep 17 00:00:00 2001 From: Daniel Rojas Date: Fri, 22 May 2020 14:48:43 +0200 Subject: [PATCH] Allow hiding connector name --- src/wireviz.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/wireviz.py b/src/wireviz.py index 1780e96..f8fa61b 100644 --- a/src/wireviz.py +++ b/src/wireviz.py @@ -18,8 +18,9 @@ color_dict = {'BK': '#000000', class Node: - def __init__(self, name, num_pins=None, pinout=None, ports_left=False, ports_right=False): + def __init__(self, name, show_name=True, num_pins=None, pinout=None, ports_left=False, ports_right=False): self.name = name + self.show_name = show_name self.ports_left = ports_left self.ports_right = ports_right self.loops = [] @@ -54,7 +55,14 @@ class Node: 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 if self.ports_left == True: s = s + '{'