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

Fix some cruft.

Browse Source
This commit is contained in:
Anna Rose 2013-05-29 19:02:30 -04:00
parent bd35cc174e
commit d325b78830
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
polygraph.py
View File

@ -20,7 +20,6 @@ def parse_args():
parser.add_argument('--input', '-i', default='network.json', help='Input file in json format') parser.add_argument('--input', '-i', default='network.json', help='Input file in json format')
parser.add_argument('--output', '-o', default='network.png', help='Output image file. Format auto-detected from file extension') parser.add_argument('--output', '-o', default='network.png', help='Output image file. Format auto-detected from file extension')
parser.add_argument('--gender', '-g', action='store_true', help='Indicate gender in output graph') parser.add_argument('--gender', '-g', action='store_true', help='Indicate gender in output graph')
parser.add_argument('--relationships', '-r', action='store_true', help='Indicate relationship details (std risk, relationship type) in output graph')
return parser.parse_args() return parser.parse_args()
@ -50,11 +49,8 @@ def main():
# Add edges to graph # Add edges to graph
edge_style = 'solid' edge_style = 'solid'
for (member1,member2,rel,edge) in data['relationships']: for (member1,member2,edge) in data['relationships']:
if settings.relationships: edge_style = edge
edge_style = edge
relationship_color = relationship_colors[rel]
graph.add_edge(member1, member2, style=edge_style, color='black') graph.add_edge(member1, member2, style=edge_style, color='black')
graph.layout() graph.layout()