ThomasV
12 years ago
3 changed files with 189 additions and 10 deletions
@ -0,0 +1,179 @@ |
|||
# source: http://stackoverflow.com/questions/2758159/how-to-embed-a-python-interpreter-in-a-pyqt-widget |
|||
|
|||
import sys, os |
|||
import traceback |
|||
from PyQt4 import QtCore |
|||
from PyQt4 import QtGui |
|||
|
|||
class Console(QtGui.QPlainTextEdit): |
|||
def __init__(self, prompt='>> ', startup_message='', parent=None): |
|||
QtGui.QPlainTextEdit.__init__(self, parent) |
|||
self.prompt = prompt |
|||
self.history = [] |
|||
self.namespace = {} |
|||
self.construct = [] |
|||
|
|||
self.setGeometry(50, 75, 600, 400) |
|||
self.setWordWrapMode(QtGui.QTextOption.WrapAnywhere) |
|||
self.setUndoRedoEnabled(False) |
|||
self.document().setDefaultFont(QtGui.QFont("monospace", 10, QtGui.QFont.Normal)) |
|||
self.showMessage(startup_message) |
|||
|
|||
def updateNamespace(self, namespace): |
|||
self.namespace.update(namespace) |
|||
|
|||
def showMessage(self, message): |
|||
self.appendPlainText(message) |
|||
self.newPrompt() |
|||
|
|||
def newPrompt(self): |
|||
if self.construct: |
|||
prompt = '.' * len(self.prompt) |
|||
else: |
|||
prompt = self.prompt |
|||
self.appendPlainText(prompt) |
|||
self.moveCursor(QtGui.QTextCursor.End) |
|||
|
|||
def getCommand(self): |
|||
doc = self.document() |
|||
curr_line = unicode(doc.findBlockByLineNumber(doc.lineCount() - 1).text()) |
|||
curr_line = curr_line.rstrip() |
|||
curr_line = curr_line[len(self.prompt):] |
|||
return curr_line |
|||
|
|||
def setCommand(self, command): |
|||
if self.getCommand() == command: |
|||
return |
|||
self.moveCursor(QtGui.QTextCursor.End) |
|||
self.moveCursor(QtGui.QTextCursor.StartOfLine, QtGui.QTextCursor.KeepAnchor) |
|||
for i in range(len(self.prompt)): |
|||
self.moveCursor(QtGui.QTextCursor.Right, QtGui.QTextCursor.KeepAnchor) |
|||
self.textCursor().removeSelectedText() |
|||
self.textCursor().insertText(command) |
|||
self.moveCursor(QtGui.QTextCursor.End) |
|||
|
|||
def getConstruct(self, command): |
|||
if self.construct: |
|||
prev_command = self.construct[-1] |
|||
self.construct.append(command) |
|||
if not prev_command and not command: |
|||
ret_val = '\n'.join(self.construct) |
|||
self.construct = [] |
|||
return ret_val |
|||
else: |
|||
return '' |
|||
else: |
|||
if command and command[-1] == (':'): |
|||
self.construct.append(command) |
|||
return '' |
|||
else: |
|||
return command |
|||
|
|||
def getHistory(self): |
|||
return self.history |
|||
|
|||
def setHisory(self, history): |
|||
self.history = history |
|||
|
|||
def addToHistory(self, command): |
|||
if command and (not self.history or self.history[-1] != command): |
|||
self.history.append(command) |
|||
self.history_index = len(self.history) |
|||
|
|||
def getPrevHistoryEntry(self): |
|||
if self.history: |
|||
self.history_index = max(0, self.history_index - 1) |
|||
return self.history[self.history_index] |
|||
return '' |
|||
|
|||
def getNextHistoryEntry(self): |
|||
if self.history: |
|||
hist_len = len(self.history) |
|||
self.history_index = min(hist_len, self.history_index + 1) |
|||
if self.history_index < hist_len: |
|||
return self.history[self.history_index] |
|||
return '' |
|||
|
|||
def getCursorPosition(self): |
|||
return self.textCursor().columnNumber() - len(self.prompt) |
|||
|
|||
def setCursorPosition(self, position): |
|||
self.moveCursor(QtGui.QTextCursor.StartOfLine) |
|||
for i in range(len(self.prompt) + position): |
|||
self.moveCursor(QtGui.QTextCursor.Right) |
|||
|
|||
def runCommand(self): |
|||
command = self.getCommand() |
|||
self.addToHistory(command) |
|||
|
|||
command = self.getConstruct(command) |
|||
|
|||
if command: |
|||
tmp_stdout = sys.stdout |
|||
|
|||
class stdoutProxy(): |
|||
def __init__(self, write_func): |
|||
self.write_func = write_func |
|||
self.skip = False |
|||
|
|||
def write(self, text): |
|||
if not self.skip: |
|||
stripped_text = text.rstrip('\n') |
|||
self.write_func(stripped_text) |
|||
QtCore.QCoreApplication.processEvents() |
|||
self.skip = not self.skip |
|||
|
|||
sys.stdout = stdoutProxy(self.appendPlainText) |
|||
try: |
|||
try: |
|||
result = eval(command, self.namespace, self.namespace) |
|||
if result != None: |
|||
self.appendPlainText(repr(result)) |
|||
except SyntaxError: |
|||
exec command in self.namespace |
|||
except SystemExit: |
|||
self.close() |
|||
except: |
|||
traceback_lines = traceback.format_exc().split('\n') |
|||
# Remove traceback mentioning this file, and a linebreak |
|||
for i in (3,2,1,-1): |
|||
traceback_lines.pop(i) |
|||
self.appendPlainText('\n'.join(traceback_lines)) |
|||
sys.stdout = tmp_stdout |
|||
self.newPrompt() |
|||
|
|||
def keyPressEvent(self, event): |
|||
if event.key() in (QtCore.Qt.Key_Enter, QtCore.Qt.Key_Return): |
|||
self.runCommand() |
|||
return |
|||
if event.key() == QtCore.Qt.Key_Home: |
|||
self.setCursorPosition(0) |
|||
return |
|||
if event.key() == QtCore.Qt.Key_PageUp: |
|||
return |
|||
elif event.key() in (QtCore.Qt.Key_Left, QtCore.Qt.Key_Backspace): |
|||
if self.getCursorPosition() == 0: |
|||
return |
|||
elif event.key() == QtCore.Qt.Key_Up: |
|||
self.setCommand(self.getPrevHistoryEntry()) |
|||
return |
|||
elif event.key() == QtCore.Qt.Key_Down: |
|||
self.setCommand(self.getNextHistoryEntry()) |
|||
return |
|||
#elif event.key() == QtCore.Qt.Key_D and event.modifiers() == QtCore.Qt.ControlModifier: |
|||
# self.close() |
|||
super(Console, self).keyPressEvent(event) |
|||
|
|||
|
|||
welcome_message = ''' |
|||
--------------------------------------------------------------- |
|||
Welcome to a primitive Python interpreter. |
|||
--------------------------------------------------------------- |
|||
''' |
|||
|
|||
if __name__ == '__main__': |
|||
app = QtGui.QApplication(sys.argv) |
|||
console = Console(startup_message=welcome_message) |
|||
console.updateNamespace({'myVar1' : app, 'myVar2' : 1234}) |
|||
console.show(); |
|||
sys.exit(app.exec_()) |
|||
Loading…
Reference in new issue