class ProcessDialog(QtGui.QDialog):
def __init__(self, parent, callable, args):
QtGui.QDialog.__init__(self, parent)
# Set up the UI from designer
self.ui=UI_ProcessDialog()
self.ui.setupUi(self)
self.output=multiprocessing.Queue()
self.callable=callable
self.args=args+[self.output]
self.timer=QtCore.QTimer(self)
QtCore.QObject.connect(self.timer, QtCore.SIGNAL("timeout()"), self.showOutput)
self.proc=None
def exec_(self):
self.show()
self.start()
return QtGui.QDialog.exec_(self)
def start(self):
self.proc=multiprocessing.Process(target=self.callable, args=self.args)
self.proc.start()
self.showOutput()
def reject(self):
if self.proc and self.proc.is_alive():
self.proc.terminate()
return QtGui.QDialog.reject(self)
def showOutput(self):
while not self.output.empty():
[code, data]=self.output.get()
if code==0: # Regular output
self.ui.output.append(data+'<br>')
elif code==1:
self.ui.output.append('<b>'+data+'<b><br>')
elif code==2: # Really bad
QtGui.QMessageBox.critical(self, 'Error - uRSSus', data )
self.reject()
elif code==100: # The result data
self.result=data
self.accept()
if self.proc.is_alive():
self.timer.setInterval(500)
self.timer.start()
def __init__(self, parent, callable, args):
QtGui.QDialog.__init__(self, parent)
# Set up the UI from designer
self.ui=UI_ProcessDialog()
self.ui.setupUi(self)
self.output=multiprocessing.Queue()
self.callable=callable
self.args=args+[self.output]
self.timer=QtCore.QTimer(self)
QtCore.QObject.connect(self.timer, QtCore.SIGNAL("timeout()"), self.showOutput)
self.proc=None
