Use optparse.OptionParser to parse script arguments.

[depgraph.git] / py2depgraph.py

diff --git a/py2depgraph.py b/py2depgraph.py
index cb2bc8635f68004d1bff3201e71551db8cad717b..b17c4ddbb3217c9d90fb9ffb9f2c24b4ad00896b 100755 (executable)
--- 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.
 
 # 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):
 import modulefinder
 
 class mymf(modulefinder.ModuleFinder):


@@ -55,13 +57,21 @@ class mymf(modulefinder.ModuleFinder):
         return r
         
         
         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)
     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})