diff --git a/src/wireviz/Harness.py b/src/wireviz/Harness.py index af02399..4153193 100644 --- a/src/wireviz/Harness.py +++ b/src/wireviz/Harness.py @@ -93,8 +93,8 @@ class Harness: def create_graph(self) -> Graph: dot = Graph() - dot.body.append(f'// Graph generated by {APP_NAME} {__version__}') - dot.body.append(f'// {APP_URL}') + dot.body.append(f'// Graph generated by {APP_NAME} {__version__}\n') + dot.body.append(f'// {APP_URL}\n') dot.attr('graph', rankdir='LR', ranksep='2', bgcolor=wv_colors.translate_color(self.options.bgcolor, "HEX"), @@ -189,13 +189,9 @@ class Harness: else: raise Exception('No side for loops') for loop in connector.loops: - if type(loop[0]) == str: - i_loop_0 = connector.pins.index(loop[0]) - i_loop_1 = connector.pins.index(loop[1]) - else: - i_loop_0 = loop[0] - i_loop_1 = loop[1] - dot.edge(f'{connector.name}:p{i_loop_0}{loop_side}:{loop_dir}', + i_loop_0 = connector.pins.index(loop[0])+1 + i_loop_1 = connector.pins.index(loop[1])+1 + dot.edge(f'{connector.name}:p{i_loop_0}{loop_side}:{loop_dir}', f'{connector.name}:p{i_loop_1}{loop_side}:{loop_dir}')