class QPyShell(QWidget):
"""QPyShell - a Qt based python command shell based on code.InteractiveInterpreter.
Because it catches stdout and stderr there can be only one active instance of this
class. Make sure initInterpreter() and exitInterpreter() is called!
"""
activeInstance = None
def __init__(self, parent=None):
QWidget.__init__(self, parent)
if parent:
self.standalone = False
else:
self.standalone = True
self.setWindowTitle(QCoreApplication.translate("QPyShell", "QPyShell - a simple python shell widget for Qt"))
self.mainLayout = QVBoxLayout(self)
self.outputBrowser = QTextEdit(self)
self.outputBrowser.setMinimumSize(QSize(100,100))
self.outputBrowser.setReadOnly(True)
self.mainLayout.addWidget(self.outputBrowser)
self.clLayout = QHBoxLayout()
self.mainLayout.addLayout(self.clLayout)
self.promptLabel = QLabel()
self.promptLabel.setText(">>>")
self.clLayout.addWidget(self.promptLabel)
self.lineInput = QLineEdit()
self.lineInput.setToolTip(QCoreApplication.translate('QPyShell', 'The python commandline: enter code her'))
self.clLayout.addWidget(self.lineInput)
self.enterButton = QToolButton(self)
self.enterButton.setText(QCoreApplication.translate("QPyShell", "Enter"))
self.enterButton.setToolTip(QCoreApplication.translate('QPyShell', 'This button or [Enter] executes the command'))
self.enterButton.setIcon(QIcon(QPixmap(enterXPM)))
self.clLayout.addWidget(self.enterButton)
self.clLayout.addSpacing(8)
self.clearButton = QToolButton(self)
self.clearButton.setText(QCoreApplication.translate("QPyShell", "Clear"))
self.clearButton.setToolTip(QCoreApplication.translate('QPyShell', 'Clear the output window'))
self.clearButton.setIcon(QIcon(QPixmap(clearXPM)))
self.clLayout.addWidget(self.clearButton)
self.saveButton = QToolButton(self)
self.saveButton.setText(QCoreApplication.translate("QPyShell", "Save ..."))
self.saveButton.setToolTip(QCoreApplication.translate('QPyShell', 'Save the contents of the output window'))
self.saveButton.setIcon(QIcon(QPixmap(saveXPM)))
self.clLayout.addWidget(self.saveButton)
self.printButton = QToolButton(self)
self.printButton.setText(QCoreApplication.translate("QPyShell", "Print"))
self.printButton.setToolTip(QCoreApplication.translate('QPyShell', 'Print the contents of the output window'))
self.printButton.setIcon(QIcon(QPixmap(printXPM)))
self.clLayout.addWidget(self.printButton)
self.history = []
self.historyFile = None
self.history_i = -1
self.cmdBuffer = []
self.showCompletionLimit = 100
self.interpreter = None
self.old_displayhook = None
self.cursor = QTextCursor(self.outputBrowser.document())
self.outputBrowser.setTextCursor(self.cursor)
self.outputBrowser.append(greeting)
self.setTabOrder(self.lineInput, self.outputBrowser)
self.setTabOrder(self.outputBrowser, self.enterButton)
self.setTabOrder(self.enterButton, self.saveButton)
self.setTabOrder(self.saveButton, self.clearButton)
self.connect(self.enterButton, SIGNAL("pressed()"), self.run)
self.connect(self.saveButton, SIGNAL("pressed()"), self.saveContents)
self.connect(self.printButton, SIGNAL("pressed()"), self.printContents)
self.connect(self.clearButton, SIGNAL("pressed()"), self.outputBrowser.clear)
def initInterpreter(self, loc=None, greet=greeting, historyFile=None):
if QPyShell.activeInstance:
raise Exception(QCoreApplication.translate("QPyShell", "QPyShell: There can be only one highlander... sorry, I mean one active QPyShell widget!"))
QPyShell.activeInstance = self
self.loadHistoryFile(historyFile)
self.interpreter = InteractiveInterpreter(loc)
self.completer = Completer(loc)
self.old_stdout = sys.stdout
self.old_stderr = sys.stderr
sys.stdout = DummyFileW(self.write)
sys.stderr = sys.stdout
# there's a strange problem with gettext and interactive interpreters
# gettext's "_"" will be overwritten by the standard sys.displayhook...
self.old_displayhook = sys.displayhook
sys.displayhook = mydisplayhook
def loadHistoryFile(self, historyFile):
self.historyFile = historyFile
if historyFile and os.path.exists(historyFile):
lines = open(historyFile, 'r').read().split(os.linesep)
self.history = [l for l in lines if not l.startswith('#')]
def saveHistoryFile(self):
if self.historyFile:
h = self.history
if len(h) > 100:
h = h[:100]
h.insert(0, unicode(QCoreApplication.translate('QPyShell', '# this is the command history of the QPyShell')))
open(self.historyFile, 'w').write(os.linesep.join(h))
def exitInterpreter(self, loc=None):
sys.stdout = self.old_stdout
sys.stderr = self.old_stderr
self.saveHistoryFile()
del self.interpreter
self.interpreter = None
sys.displayhook = self.old_displayhook
QPyShell.ativeInstance = None
def run(self):
if not self.interpreter:
raise Exception(QCoreApplication.translate("QPyShell", "No interpreter found! You need to call QPyShell.initInterpreter() first!"))
line = unicode(self.lineInput.text())
self.lineInput.clear()
if line:
if (not self.history) or line != self.history[0]:
# no consecutive identical entries
self.history.insert(0, line)
self.history_i = -1
self.cursor.movePosition(QTextCursor.End)
if not self.cmdBuffer:
self.cursor.insertHtml('<br><b>|>>> %s</b><br>' % formatPyLine(line))
else:
self.cursor.insertHtml('<br><b>|. . . %s</b><br>' % formatPyLine(line))
self.cursor.movePosition(QTextCursor.End)
self.outputBrowser.ensureCursorVisible()
self.cmdBuffer.append(line)
more = self.interpreter.runsource('\n'.join(self.cmdBuffer))
if more:
self.promptLabel.setText('...')
else:
self.cmdBuffer = []
self.promptLabel.setText('>>>')
def appendHtml(self, txt):
self.cursor.movePosition(QTextCursor.End)
self.cursor.insertHtml(txt)
self.cursor.movePosition(QTextCursor.End)
self.outputBrowser.ensureCursorVisible()
def appendText(self, txt):
self.cursor.movePosition(QTextCursor.End)
self.outputBrowser.insertPlainText(txt)
self.cursor.movePosition(QTextCursor.End)
self.outputBrowser.ensureCursorVisible()
write = appendText
def keyPressEvent(self, event):
key = event.key()
mod = event.modifiers()
if key == Qt.Key_Up and self.history:
self.history_i = min(self.history_i+1, len(self.history)-1)
self.lineInput.setText(self.history[self.history_i])
elif key == Qt.Key_Down and self.history:
self.history_i = max(self.history_i-1, -1)
if self.history_i >= 0:
self.lineInput.setText(self.history[self.history_i])
else:
self.lineInput.setText('')
elif key == Qt.Key_Enter or key == Qt.Key_Return:
if mod & Qt.ShiftModifier:
self.complete()
else:
self.run()
elif key == Qt.Key_Escape:
txt, r = QInputDialog.getItem(self, QCoreApplication.translate("QPyShell", "Command line history"),
QCoreApplication.translate("QPyShell", "Please select a history item:"),
self.history, 0, False)
if r and txt:
self.lineInput.setText(txt)
elif self.standalone and key == Qt.Key_Print:
self.printContents()
elif self.standalone and key == Qt.Key_Q and (mod & Qt.ControlModifier):
self.close()
else:
QWidget.keyPressEvent(self, event)
def closeEvent(self, event):
if self.standalone:
self.exitInterpreter()
QWidget.closeEvent(self, event)
def printContents(self, printer=None):
if not printer:
printer = QPrinter()
dialog = QPrintDialog(printer, self)
dialog.setWindowTitle(QCoreApplication.translate("QPyShell", "Print Document"))
if dialog.exec_() != QDialog.Accepted:
return
self.outputBrowser.document().print_(printer)
def saveContents(self, fileName=None):
if not fileName:
fileTypes = {'Text':('txt',), 'HTML':('htm', 'html')}
filters = ';;'.join(['%s (%s)' % (k, ' '.join(['*.'+e for e in v])) for k, v in fileTypes.items()])
dlg = QFileDialog(self,
QCoreApplication.translate('QPyShell', 'Select name of file to save'),
os.getcwd(), filters)
dlg.setFileMode(QFileDialog.AnyFile)
dlg.setAcceptMode(QFileDialog.AcceptSave)
if dlg.exec_() != QDialog.Accepted:
return
tmp = unicode(dlg.selectedFilter())
fileType = tmp[:tmp.find('(')-1]
dlg.setDefaultSuffix(fileTypes[fileType][0])
files = dlg.selectedFiles()
if not files:
return
fileName = unicode(files[0])
if fileType == 'Text':
txt = self.outputBrowser.toPlainText()
elif fileType == 'HTML':
txt = self.outputBrowser.toHtml()
else:
raise IOError('Unknown FileType: %s' % fileType)
try:
open(fileName, 'w').write(txt)
except IOError:
QMessageBox.critical(self,
QCoreApplication.translate('QPyShell', 'Could not save file!'),
QCoreApplication.translate('QPyShell', 'Writing failed! Make sure you have write permissions!'))
def complete(self):
""" a very simple, quick and dirty completion of identifiers in the namespace """
# FIXME: fertig machen!
txt = unicode(self.lineInput.text())
cur = self.lineInput.cursorPosition()
s = cur
while s>0 and txt[s-1] in identChars:
s -= 1
try:
completions = self.completer.matches(txt[s:cur])
except:
return
if not completions:
return
n_comp = len(completions)
if n_comp == 1:
comp = completions.pop()
self.lineInput.insert(comp[cur-s:])
elif n_comp < self.showCompletionLimit:
tmp = list(completions)
tmp.sort()
self.appendHtml('<font color="#0000ff">[%s]</font>' % ', '.join(tmp))
# get common prefix ... stolen from the python cookbook
pref = tmp[0][:([min([x[0]==elem for elem in x]) for x in zip(*tmp)]+[0]).index(0)]
if len(pref) > (cur-s):
self.lineInput.insert(pref[cur-s:])
else:
self.appendHtml(unicode(QCoreApplication.translate("QPyShell",
'<font color="#0000ff">[Too many completions: %d]</font>')) % n_comp)
