mirror of
https://github.com/stevenrobertson/cuburn.git
synced 2025-02-05 03:30:05 -05:00
Improve graph.py
This commit is contained in:
parent
d51a582dd8
commit
f3737988a9
@ -1,8 +1,13 @@
|
|||||||
import os, sys
|
import os, sys
|
||||||
|
from os.path import isdir, join
|
||||||
|
|
||||||
|
dir = sys.argv[1] if len(sys.argv) > 1 else '.'
|
||||||
|
|
||||||
print 'digraph {'
|
print 'digraph {'
|
||||||
for i in os.listdir(sys.argv[1] if len(sys.argv) > 1 else '.'):
|
for i in os.listdir(dir):
|
||||||
parts = i.rsplit('.', 1)[0].split('=')
|
if not isdir(join(dir, i)):
|
||||||
|
i = i.rsplit('.', 1)[0]
|
||||||
|
parts = i.split('=')
|
||||||
print ' -> '.join(parts[:2])
|
print ' -> '.join(parts[:2])
|
||||||
# TODO: add label (optional section 3)
|
# TODO: add label (optional section 3)
|
||||||
print '}'
|
print '}'
|
||||||
|
Loading…
Reference in New Issue
Block a user