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
|
||||
from os.path import isdir, join
|
||||
|
||||
dir = sys.argv[1] if len(sys.argv) > 1 else '.'
|
||||
|
||||
print 'digraph {'
|
||||
for i in os.listdir(sys.argv[1] if len(sys.argv) > 1 else '.'):
|
||||
parts = i.rsplit('.', 1)[0].split('=')
|
||||
for i in os.listdir(dir):
|
||||
if not isdir(join(dir, i)):
|
||||
i = i.rsplit('.', 1)[0]
|
||||
parts = i.split('=')
|
||||
print ' -> '.join(parts[:2])
|
||||
# TODO: add label (optional section 3)
|
||||
print '}'
|
||||
|
Loading…
Reference in New Issue
Block a user