Improve graph.py

This commit is contained in:
Steven Robertson 2013-01-06 22:02:21 -08:00
parent d51a582dd8
commit f3737988a9

View File

@ -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 '}'