From bd35cc174e0ad4c773242d05f9f0035663278bec Mon Sep 17 00:00:00 2001
From: Anna Wiggins <annabunches@gmail.com>
Date: Wed, 29 May 2013 18:44:38 -0400
Subject: [PATCH] Allow arbitrary usage of edge line styles

---
 polygraph.py | 25 +++++--------------------
 1 file changed, 5 insertions(+), 20 deletions(-)

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)