diff --git a/polygraph.py b/polygraph.py index 8ee64c7..216b113 100755 --- a/polygraph.py +++ b/polygraph.py @@ -5,14 +5,6 @@ import json import argparse # These are the attributes to use for color-coding the graphs -# fixme: make these configurable - - -# Relationship types... feel free to edit to suit your poly family's needs -relationship_colors = { - 'stable': 'blue', - 'developing': 'red' - } # Three gender colours, so that we encapsulate masculine, feminine, and a catch-all for other identities # If you'd rather explicitly colour-code other identities, just add them :) @@ -22,12 +14,6 @@ gender_colors = { 'o': 'green', } -# This represents whether a particular relationship involves any -# std risk. The values are applied to the edge 'style' attribute. -sti_styles = { 'yes': 'solid', - 'no': 'dashed', - } - def parse_args(): parser = argparse.ArgumentParser(description='Generate graphs of polyamorous family networks from data files') @@ -62,15 +48,14 @@ def main(): graph.add_node(user['name'], fillcolor=gender_color) # Add edges to graph - sti_style = 'solid' - relationship_color = 'black' + edge_style = 'solid' - for (member1,member2,rel,sti) in data['relationships']: - if settings.relationship_types: - sti_style = sti_styles[sti] + for (member1,member2,rel,edge) in data['relationships']: + if settings.relationships: + edge_style = edge relationship_color = relationship_colors[rel] - graph.add_edge(member1, member2, style=sti_style, color=relationship_color) + graph.add_edge(member1, member2, style=edge_style, color='black') graph.layout() graph.draw(settings.output)