126 lines
3.6 KiB
Python
126 lines
3.6 KiB
Python
|
import requests
|
||
|
|
import sys
|
||
|
|
|
||
|
|
url = 'http://localhost:8000/notes/graph.json'
|
||
|
|
reference_node = sys.argv[1]
|
||
|
|
out = sys.argv[2]
|
||
|
|
|
||
|
|
g = requests.get(url).json()
|
||
|
|
centered_graph = { reference_node: g[reference_node] }
|
||
|
|
del g[reference_node]
|
||
|
|
new_nodes = True
|
||
|
|
|
||
|
|
in_emacs_tree = {
|
||
|
|
reference_node: set(),
|
||
|
|
}
|
||
|
|
|
||
|
|
while new_nodes:
|
||
|
|
new_nodes = False
|
||
|
|
removed = set()
|
||
|
|
for k, v in g.items():
|
||
|
|
for link in v["links"]:
|
||
|
|
if link["target"].startswith("id:"):
|
||
|
|
link["target"] = link["target"][3:]
|
||
|
|
if link['target'] in centered_graph and link.get('relation') == 'in':
|
||
|
|
centered_graph[k] = v
|
||
|
|
|
||
|
|
for l in v["links"]:
|
||
|
|
if l.get('relation') == 'in':
|
||
|
|
t = l['target']
|
||
|
|
if t.startswith("id:"):
|
||
|
|
t = t[3:]
|
||
|
|
|
||
|
|
if '[' in t:
|
||
|
|
# Special case, to be handled on org_rw
|
||
|
|
continue
|
||
|
|
|
||
|
|
if t not in in_emacs_tree:
|
||
|
|
in_emacs_tree[t] = set()
|
||
|
|
in_emacs_tree[t].add(k)
|
||
|
|
|
||
|
|
v['links'] = [
|
||
|
|
l for l in v["links"]
|
||
|
|
if l.get('relation') != 'in'
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
removed.add(k)
|
||
|
|
new_nodes = True
|
||
|
|
break
|
||
|
|
for k in removed:
|
||
|
|
del g[k]
|
||
|
|
|
||
|
|
in_emacs = set(centered_graph.keys())
|
||
|
|
|
||
|
|
|
||
|
|
# One more round for the rest, not requiring "in"
|
||
|
|
for k, v in g.items():
|
||
|
|
for link in v["links"]:
|
||
|
|
if link["target"].startswith("id:"):
|
||
|
|
link["target"] = link["target"][3:]
|
||
|
|
if link['target'] in in_emacs:
|
||
|
|
centered_graph[k] = v
|
||
|
|
removed.add(k)
|
||
|
|
|
||
|
|
g = centered_graph
|
||
|
|
|
||
|
|
f = open('graph.dot', 'wt')
|
||
|
|
f.write('digraph {\n')
|
||
|
|
# f.write('bgcolor="#222222"\n')
|
||
|
|
# f.write('fontcolor="#ffffff"\n')
|
||
|
|
f.write('maxiter=1000\n')
|
||
|
|
f.write('splines=curved\n')
|
||
|
|
# f.write('splines=spline\n') # Not supported with edges to cluster
|
||
|
|
f.write('node[shape=rect]\n')
|
||
|
|
# f.write('edge[color="#ffffff"]\n')
|
||
|
|
|
||
|
|
def draw_subgraph(node_id):
|
||
|
|
f.write("subgraph cluster_{} {{\n".format(node_id.replace("-", "_")))
|
||
|
|
f.write('URL="./{}.node.html"\n'.format(node_id))
|
||
|
|
# f.write('color="#ffffff"\n')
|
||
|
|
|
||
|
|
f.write("label=\"{}\"\n".format(g[node_id]['title'].replace("\"", "'")))
|
||
|
|
f.write("\n")
|
||
|
|
|
||
|
|
# print("T: {}".format(in_emacs_tree), file=sys.stderr)
|
||
|
|
for k in in_emacs_tree[node_id]:
|
||
|
|
v = g[k]
|
||
|
|
print("_" + k.replace("-", "_") + "[label=\"" + v["title"].replace("\"", "'") + "\", URL=\"" + k + ".node.html\"];", file=f)
|
||
|
|
|
||
|
|
if k in in_emacs_tree:
|
||
|
|
draw_subgraph(k)
|
||
|
|
|
||
|
|
f.write("\n}")
|
||
|
|
|
||
|
|
draw_subgraph(reference_node)
|
||
|
|
|
||
|
|
for k, v in g.items():
|
||
|
|
if k not in in_emacs:
|
||
|
|
print("_" + k.replace("-", "_") + "[label=\"" + v["title"].replace("\"", "'") + "\", URL=\"" + k + ".node.html\"];", file=f)
|
||
|
|
|
||
|
|
for k, v in g.items():
|
||
|
|
for link in v["links"]:
|
||
|
|
if link["target"].startswith("id:"):
|
||
|
|
link["target"] = link["target"][3:]
|
||
|
|
|
||
|
|
if '[' in link['target']:
|
||
|
|
# Special case, to be handled on org_rw
|
||
|
|
continue
|
||
|
|
if link['target'] not in g:
|
||
|
|
# Irrelevant
|
||
|
|
continue
|
||
|
|
if link['target'] in in_emacs_tree:
|
||
|
|
t = 'cluster_{}'.format(link['target'].replace("-", "_"))
|
||
|
|
else:
|
||
|
|
t = "_" + link["target"].replace("-", "_")
|
||
|
|
print("_" + k.replace("-", "_") + "->" + t, file=f)
|
||
|
|
|
||
|
|
f.write('}\n')
|
||
|
|
# dot graph.dot -Tsvg graph.svg
|
||
|
|
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
subprocess.call("fdp graph.dot -Tsvg -o '{}'".format(out), shell=True)
|
||
|
|
# return "graph.svg"
|