annabunches/polygraph
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Modify input file format to be easier to append/maintain

Browse Source
This commit is contained in:
Anna Rose 2012-09-10 20:32:33 -04:00
parent 6ba1b97f97
commit 18104d8317
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
polygraph.py
View File

@ -22,8 +22,8 @@ gender_colors = {
# Relationship types... feel free to edit to suit your poly family's needs # Relationship types... feel free to edit to suit your poly family's needs
relationship_colors = { relationship_colors = {
's': 'blue', 'stable': 'blue',
'd': 'red' 'developing': 'red'
} }
@ -40,16 +40,15 @@ def main():
graph.edge_attr['arrowsize'] = 0.5 graph.edge_attr['arrowsize'] = 0.5
# Add nodes to graph # Add nodes to graph
for user in data: for user in data['members']:
graph.add_node(user['name'], fillcolor=gender_colors[user['gender']]) graph.add_node(user['name'], fillcolor=gender_colors[user['gender']])
# Add edges to graph # Add edges to graph
for user1 in data: for (member1,member2,rel,fluid) in data['relationships']:
for (user2, rel, fluid) in user1['connections']: graph.add_edge(member1, member2, style=fluid_colors[fluid], color=relationship_colors[rel])
graph.add_edge(user1['name'], user2, style=fluid_colors[fluid], color=relationship_colors[rel])
graph.layout() graph.layout()
graph.draw('/tmp/graph.png') graph.draw('poly_network.png')
if __name__ == '__main__': main() if __name__ == '__main__': main()