Skip to content

Commit

Permalink
Additional styles by attribute.
Browse files Browse the repository at this point in the history
  • Loading branch information
thvitt committed Sep 7, 2018
1 parent 7ac95d7 commit cb35d55
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions visualize.py
Expand Up @@ -98,14 +98,18 @@ def write_dot(graph: nx.MultiDiGraph, target='base_graph.dot', style=_load_style
kind = attr.get('kind', None)
if kind in style['edge']:
simplified.edges[u, v, k].update(style['edge'][kind])
if attr.get('delete', False) and 'delete' in style['edge']:
simplified.edges[u, v, k].update(style['edge']['delete'])
for styled_attr in attr.keys() & style['edge']:
if attr[styled_attr]:
simplified.edges[u, v, k].update(style['edge'][styled_attr])

if 'node' in style:
for node, attr in simplified.nodes(data=True):
kind = attr.get('kind', None)
if kind in style['node']:
simplified.nodes[node].update(style['node'][kind])
simplified.nodes[node].update(style['node'][kind])
for styled_attr in attr.keys() & style['node']:
if attr[styled_attr]:
attr.update(style['node'][styled_attr])

if not edge_labels:
for u, v, k, attr in simplified.edges(data=True, keys=True):
Expand Down

0 comments on commit cb35d55

Please sign in to comment.