test/test_readers/test_python/showparse

   1 #! /usr/bin/env python
   2
   3 """
   4 This is a tool for exploring abstract syntax trees generated by
   5 ``parser.suite()`` from test data in
   6 docutils/test/test_readers/test_python/test_parser or stdin.
   7
   8 Usage::
   9
  10     showparse <key> <index>
  11
  12     showparse < <module.py>
  13
  14 Where ``<key>`` is the key to the ``totest`` dictionary, and ``<index>`` is
  15 the index of the list ``totest[key]``.  If no arguments are given, stdin is
  16 used for input.
  17 """
  18
  19 import sys
  20 import parser
  21 import token
  22 import symbol
  23 import pprint
  24 import test_parser
  25
  26 names = token.tok_name.copy()
  27 names.update(symbol.sym_name)
  28
  29 def name_elements(ast):
  30     if ast:
  31         name = names[ast[0]]
  32         ast[0] = '%s (%s)' % (name, ast[0])
  33     for node in ast[1:]:
  34         if type(node) is list:
  35             name_elements(node)
  36
  37 if len(sys.argv) > 1:
  38     key, caseno = sys.argv[1:]
  39     print 'totest["%s"][%s][0]:\n' % (key, caseno)
  40     input_text = test_parser.totest[key][int(caseno)][0]
  41 else:
  42     input_text = sys.stdin.read()
  43 print input_text
  44 module = parser.suite(input_text)
  45 ast = parser.ast2list(module, line_info=1)
  46 name_elements(ast)
  47 pprint.pprint(ast)