Enhanced and really multi-platform process logging. Use wxProcess and wxExecute instead of python popen3 (unix only).
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#This file is part of Beremiz, a Integrated Development Environment for
#programming IEC 61131-3 automates supporting plcopen standard and CanFestival.
#
#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
#
#See COPYING file for copyrights details.
#
#This library is free software; you can redistribute it and/or
#modify it under the terms of the GNU General Public
#License as published by the Free Software Foundation; either
#version 2.1 of the License, or (at your option) any later version.
#
#This library is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
#General Public License for more details.
#
#You should have received a copy of the GNU General Public
#License along with this library; if not, write to the Free Software
#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
# based on wxPopen.py from boa-constructor
#
import time
from StringIO import StringIO
from wxPython.wx import *
class ProcessRunnerMix:
def __init__(self, input, handler=None):
if handler is None:
handler = self
self.handler = handler
EVT_IDLE(handler, self.OnIdle)
EVT_END_PROCESS(handler, -1, self.OnProcessEnded)
input.reverse() # so we can pop
self.input = input
self.reset()
def reset(self):
self.process = None
self.pid = -1
self.output = []
self.errors = []
self.inputStream = None
self.errorStream = None
self.outputStream = None
self.outputFunc = None
self.errorsFunc = None
self.finishedFunc = None
self.finished = false
self.responded = false
def execute(self, cmd):
self.process = wxProcess(self.handler)
self.process.Redirect()
self.pid = wxExecute(cmd, wxEXEC_NOHIDE, self.process)
self.inputStream = self.process.GetOutputStream()
self.errorStream = self.process.GetErrorStream()
self.outputStream = self.process.GetInputStream()
#self.OnIdle()
wxWakeUpIdle()
def setCallbacks(self, output, errors, finished):
self.outputFunc = output
self.errorsFunc = errors
self.finishedFunc = finished
def detach(self):
if self.process is not None:
self.process.CloseOutput()
self.process.Detach()
self.process = None
def kill(self):
if self.process is not None:
self.process.CloseOutput()
if wxProcess_Kill(self.pid, wxSIGTERM) != wxKILL_OK:
wxProcess_Kill(self.pid, wxSIGKILL)
self.process = None
def updateStream(self, stream, data):
if stream and stream.CanRead():
if not self.responded:
self.responded = true
text = stream.read()
data.append(text)
return text
else:
return None
def updateInpStream(self, stream, input):
if stream and input:
line = input.pop()
stream.write(line)
def updateErrStream(self, stream, data):
return self.updateStream(stream, data)
def updateOutStream(self, stream, data):
return self.updateStream(stream, data)
def OnIdle(self, event=None):
if self.process is not None:
self.updateInpStream(self.inputStream, self.input)
e = self.updateErrStream(self.errorStream, self.errors)
if e is not None and self.errorsFunc is not None:
wxCallAfter(self.errorsFunc, e)
o = self.updateOutStream(self.outputStream, self.output)
if o is not None and self.outputFunc is not None:
wxCallAfter(self.outputFunc, o)
#wxWakeUpIdle()
#time.sleep(0.001)
def OnProcessEnded(self, event):
self.OnIdle()
pid,exitcode = event.GetPid(), event.GetExitCode()
if self.process:
self.process.Destroy()
self.process = None
self.finished = true
# XXX doesn't work ???
#self.handler.Disconnect(-1, wxEVT_IDLE)
if self.finishedFunc:
wxCallAfter(self.finishedFunc, pid, exitcode)
class ProcessRunner(wxEvtHandler, ProcessRunnerMix):
def __init__(self, input):
wxEvtHandler.__init__(self)
ProcessRunnerMix.__init__(self, input)
def wxPopen3(cmd, input, output, errors, finish, handler=None):
p = ProcessRunnerMix(input, handler)
p.setCallbacks(output, errors, finish)
p.execute(cmd)
return p