contrib/llvm/tools/lldb/source/Interpreter/embedded_interpreter.py

   1 import readline
   2 import code
   3 import sys
   4 import traceback
   5
   6 class SimpleREPL(code.InteractiveConsole):
   7    def __init__(self, prompt, dict):
   8        code.InteractiveConsole.__init__(self,dict)
   9        self.prompt = prompt
  10        self.loop_exit = False
  11        self.dict = dict
  12
  13    def interact(self):
  14        try:
  15            sys.ps1
  16        except AttributeError:
  17            sys.ps1 = ">>> "
  18        try:
  19            sys.ps2
  20        except AttributeError:
  21            sys.ps2 = "... "
  22
  23        while not self.loop_exit:
  24            try:
  25                self.read_py_command()
  26            except (SystemExit, EOFError):
  27                # EOF while in Python just breaks out to top level.
  28                self.write('\n')
  29                self.loop_exit = True
  30                break
  31            except KeyboardInterrupt:
  32                self.write("\nKeyboardInterrupt\n")
  33                self.resetbuffer()
  34                more = 0
  35            except:
  36                traceback.print_exc()
  37
  38    def process_input (self, in_str):
  39       # Canonicalize the format of the input string
  40       temp_str = in_str
  41       temp_str.strip(' \t')
  42       words = temp_str.split()
  43       temp_str = ('').join(words)
  44
  45       # Check the input string to see if it was the quit
  46       # command.  If so, intercept it, so that it doesn't
  47       # close stdin on us!
  48       if (temp_str.lower() == "quit()" or temp_str.lower() == "exit()"):
  49          self.loop_exit = True
  50          in_str = "raise SystemExit "
  51       return in_str
  52
  53    def my_raw_input (self, prompt):
  54       stream = sys.stdout
  55       stream.write (prompt)
  56       stream.flush ()
  57       try:
  58          line = sys.stdin.readline()
  59       except KeyboardInterrupt:
  60          line = " \n"
  61       except (SystemExit, EOFError):
  62          line = "quit()\n"
  63       if not line:
  64          raise EOFError
  65       if line[-1] == '\n':
  66          line = line[:-1]
  67       return line
  68
  69    def read_py_command(self):
  70        # Read off a complete Python command.
  71        more = 0
  72        while 1:
  73            if more:
  74                prompt = sys.ps2
  75            else:
  76                prompt = sys.ps1
  77            line = self.my_raw_input(prompt)
  78            # Can be None if sys.stdin was redefined
  79            encoding = getattr(sys.stdin, "encoding", None)
  80            if encoding and not isinstance(line, unicode):
  81                line = line.decode(encoding)
  82            line = self.process_input (line)
  83            more = self.push(line)
  84            if not more:
  85                break
  86
  87    def one_line (self, input):
  88       line = self.process_input (input)
  89       more = self.push(line)
  90       if more:
  91          self.write ("Input not a complete line.")
  92          self.resetbuffer()
  93          more = 0
  94
  95 def run_python_interpreter (dict):
  96    # Pass in the dictionary, for continuity from one session to the next.
  97    repl = SimpleREPL('>>> ', dict)
  98    repl.interact()
  99
 100 def run_one_line (dict, input_string):
 101    repl = SimpleREPL ('', dict)
 102    repl.one_line (input_string)
 103