X-Git-Url: http://git.tremily.us/?p=depgraph.git;a=blobdiff_plain;f=py2depgraph.py;fp=py2depgraph.py;h=b17c4ddbb3217c9d90fb9ffb9f2c24b4ad00896b;hp=cb2bc8635f68004d1bff3201e71551db8cad717b;hb=919a27a22788755c63a7460ec5a97b936e177755;hpb=6ace694c676cfc82860f826efb85cb784d407378

diff --git a/py2depgraph.py b/py2depgraph.py
index cb2bc86..b17c4dd 100755
--- a/py2depgraph.py
+++ b/py2depgraph.py
@@ -22,7 +22,9 @@
 # TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 # SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
-import sys, pprint
+"""Extract a tree of module imports from a Python script.
+"""
+
 import modulefinder
 
 class mymf(modulefinder.ModuleFinder):
@@ -55,13 +57,21 @@ class mymf(modulefinder.ModuleFinder):
         return r
         
         
-def main(argv):    
+if __name__=='__main__':
+    from optparse import OptionParser
+    from pprint import pprint
+    import sys
+
+    usage = '%prog [options] path/to/script.py'
+    p = OptionParser(usage=usage, description=__doc__)
+
+    options,args = p.parse_args()
+
+    script = args[0]
+
     path = sys.path[:]
     debug = 0
     exclude = []
     mf = mymf(path,debug,exclude)
-    mf.run_script(argv[0])
-    pprint.pprint({'depgraph':mf._depgraph,'types':mf._types,'paths':mf._paths})
-    
-if __name__=='__main__':
-    main(sys.argv[1:])
+    mf.run_script(script)
+    pprint({'depgraph':mf._depgraph,'types':mf._types,'paths':mf._paths})