anna/polygraph
1
0
Fork 0
Code Pull requests Releases Activity Actions

Add documentation and port to python3.

Browse source
This commit is contained in:
Anna Rose Wiggins 2023-10-31 18:57:21 +00:00
parent 73ec40414e
commit 8b14daf190
3 changed files with 36 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

20
polygraph.py
View file

@ -1,4 +1,4 @@
#!/usr/bin/python
#!/usr/bin/python3
import pygraphviz as pgv
import json
@ -18,7 +18,7 @@ def main():
try:
data = json.load(open(settings.input, 'r'))
except Exception as e:
print "Error parsing data file: " + str(e)
print("Error parsing data file: " + str(e))
return
graph = pgv.AGraph()
@ -29,19 +29,23 @@ def main():
# Add nodes to graph
for user in data['members']:
node_color = 'white'
if settings.color and user.has_key('color'):
node_color = user['color']
if settings.color and 'color' in user:
node_color = user['color']
graph.add_node(user['name'], fillcolor=node_color)
# if user.has_key('cloud') and user['cloud'] = true:
# graph.add_node()
# graph.add_edge()
# Add edges to graph
for relationship in data['relationships']:
edge_style = 'solid'
if relationship.has_key('style'): edge_style = relationship['style']
if 'style' in relationship: edge_style = relationship['style']
edge_color = 'black'
if relationship.has_key('color'): edge_color = relationship['color']
if 'color' in relationship: edge_color = relationship['color']
edge_label = None
if relationship.has_key('label'): edge_label = relationship['label']
if 'label' in relationship: edge_label = relationship['label']
member0 = relationship['members'][0]
member1 = relationship['members'][1]