ThomasV
13 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