summaryrefslogtreecommitdiff
path: root/bin/addons/module_graph.py
diff options
context:
space:
mode:
Diffstat (limited to 'bin/addons/module_graph.py')
-rw-r--r--bin/addons/module_graph.py9
1 files changed, 7 insertions, 2 deletions
diff --git a/bin/addons/module_graph.py b/bin/addons/module_graph.py
index 865802526ba..01e6db67da8 100644
--- a/bin/addons/module_graph.py
+++ b/bin/addons/module_graph.py
@@ -2,11 +2,16 @@
import os
import sys
+import glob
-if not len(sys.argv)>1:
- raise Exception, 'Usage: gen_graph.sh module1 module2 module3'
+if len(sys.argv) == 2 and (sys.argv[1] in ['-h', '--help']):
+ print >>sys.stderr, 'Usage: module_graph.py [module1 module2 module3]\n\tWhen no module is specified, all modules in current directory are used'
+ sys.exit(1)
modules = sys.argv[1:]
+if not len(modules):
+ modules = map(os.path.dirname, glob.glob(os.path.join('*', '__terp__.py')))
+
done = []
print 'digraph G {'