port to argparse

Author: hongxuchen

py/dump.py

@@ -1,36 +1,33 @@
 #!/usr/bin/python3
 
+import sys
+
 
 def main():
     from clang.cindex import Index
     from pprint import pprint
 
-    from optparse import OptionParser
-
-    global opts
-
-    parser = OptionParser("usage: %prog [options] {filename} [clang-args*]")
-    parser.add_option("", "--show-ids", dest="showIDs",
-                      help="don't compute cursor IDs(very slow)",
-                      default=False)
-    parser.add_option("", "--max-depth", dest="maxDepth",
-                      help="limit cursor expansion to depth N",
-                      metavar="N", type=int, default=None)
-    parser.disable_interspersed_args()
-    (opts, args) = parser.parse_args()
+    from argparse import ArgumentParser
 
-    if len(args) == 0:
-        parser.error('invalid number arguments')
+    parser = ArgumentParser(
+        description="dump TU info",
+        usage="%(prog)s [options] {filename} [clang-args*]")
 
-    print("type(args):", type(args), " args: ", args)
+    parser.add_argument("--show-ids", dest='showIDs',
+                        help="don't compute cursor IDs(very slow)",
+                        default=False)
+    parser.add_argument("--max-depth", dest='maxDepth',
+                        help="limit cursor expansion to depth N",
+                        metavar="N", type=int, default=None)
+    opts, unknown = parser.parse_known_args(sys.argv)
 
     index = Index.create()
-    tu = index.parse(None, args)
+    tu = index.parse(None, args=sys.argv[1:])
     if not tu:
         parser.error("unable to load input")
 
-    # pprint(('diags', map(get_diag_info, tu.diagnostics)))
-    # pprint(('nodes', get_info(tu.cursor)))
+    pprint(('diags', list(map(get_diag_info, tu.diagnostics))))
+    pprint(('nodes', get_info(opts, tu.cursor)))
 
 
 def get_diag_info(diag):
@@ -43,7 +40,7 @@ def get_diag_info(diag):
     }
 
 
-def get_cursor_id(cursor, cursor_list=[]):
+def get_cursor_id(opts, cursor, cursor_list=[]):
     if not opts.showIDs:
         return None
 
@@ -57,21 +54,21 @@ def get_cursor_id(cursor, cursor_list=[]):
     return len(cursor_list) - 1
 
 
-def get_info(node, depth=0):
+def get_info(opts, node, depth=0):
     if opts.maxDepth is not None and depth >= opts.maxDepth:
         children = None
     else:
-        children = [get_info(c, depth+1) for c in node.get_children()]
+        children = [get_info(opts, c, depth + 1) for c in node.get_children()]
 
     return {
-        'id': get_cursor_id(node),
+        'id': get_cursor_id(opts, node),
         'kind': node.kind,
         'usr': node.get_usr(),
         'spelling': node.spelling,
         'extent.start': node.extent.start,
         'extent.end': node.extent.end,
         'is_definition': node.is_definition(),
-        'definition_id': get_cursor_id(node.get_definition()),
+        'definition_id': get_cursor_id(opts, node.get_definition()),
         'children': children
     }