import code class Console(code.InteractiveConsole): EOF_key_sequence = '-' def __init__(self, stdin, stdout, locals=None, filename="<console>"): self.stdin = stdin self.stdout = stdout code.InteractiveConsole.__init__(self, locals, filename) def raw_input(self, prompt=""): """Write a prompt and read a line. The returned line does not include the trailing newline. When the user enters the EOF key sequence, EOFError is raised. """ self.write(prompt) line = self.stdin.readline() if not line: raise EOFError line = line.rstrip() if line == self.EOF_key_sequence: raise EOFError else: return line def write(self, data): self.stdout.write(data) self.stdout.flush()