cec7f71502
Failed and done can stay unimplemented. If we read stdout or stderr, reschedule immediately to process all output before considering the possibility the program stopped.
147 lines
3.5 KiB
Python
147 lines
3.5 KiB
Python
# -*- Mode: Python -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
|
|
import os
|
|
import signal
|
|
import subprocess
|
|
|
|
from morituri.extern import asyncsub
|
|
from morituri.extern.log import log
|
|
from morituri.extern.task import task, gstreamer
|
|
|
|
# log.Loggable first to get logging
|
|
|
|
|
|
class SyncRunner(log.Loggable, task.SyncRunner):
|
|
pass
|
|
|
|
|
|
class GstPipelineTask(log.Loggable, gstreamer.GstPipelineTask):
|
|
pass
|
|
|
|
|
|
class PopenTask(log.Loggable, task.Task):
|
|
"""
|
|
I am a task that runs a command using Popen.
|
|
"""
|
|
|
|
logCategory = 'PopenTask'
|
|
bufsize = 1024
|
|
command = None
|
|
cwd = None
|
|
|
|
def start(self, runner):
|
|
task.Task.start(self, runner)
|
|
|
|
try:
|
|
self._popen = asyncsub.Popen(self.command,
|
|
bufsize=self.bufsize,
|
|
stdin=subprocess.PIPE, stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE, close_fds=True, cwd=self.cwd)
|
|
except OSError, e:
|
|
import errno
|
|
if e.errno == errno.ENOENT:
|
|
self.commandMissing()
|
|
|
|
raise
|
|
|
|
self.debug('Started %r with pid %d', self.command,
|
|
self._popen.pid)
|
|
|
|
self.schedule(1.0, self._read, runner)
|
|
|
|
def _read(self, runner):
|
|
try:
|
|
read = False
|
|
|
|
ret = self._popen.recv()
|
|
|
|
if ret:
|
|
self.log("read from stdout: %s", ret)
|
|
self.readbytesout(ret)
|
|
read = True
|
|
|
|
ret = self._popen.recv_err()
|
|
|
|
if ret:
|
|
self.log("read from stderr: %s", ret)
|
|
self.readbyteserr(ret)
|
|
read = True
|
|
|
|
# if we read anything, we might have more to read, so
|
|
# reschedule immediately
|
|
if read and self.runner:
|
|
self.schedule(0.0, self._read, runner)
|
|
return
|
|
|
|
# if we didn't read anything, give the command more time to
|
|
# produce output
|
|
if self._popen.poll() is None and self.runner:
|
|
# not finished yet
|
|
self.schedule(1.0, self._read, runner)
|
|
return
|
|
|
|
self._done()
|
|
except Exception, e:
|
|
self.debug('exception during _read()')
|
|
self.debug(log.getExceptionMessage(e))
|
|
self.setException(e)
|
|
self.stop()
|
|
|
|
def _done(self):
|
|
assert self._popen.returncode is not None, "No returncode"
|
|
|
|
if self._popen.returncode >= 0:
|
|
self.debug('Return code was %d', self._popen.returncode)
|
|
else:
|
|
self.debug('Terminated with signal %d',
|
|
-self._popen.returncode)
|
|
|
|
self.setProgress(1.0)
|
|
|
|
if self._popen.returncode != 0:
|
|
self.failed()
|
|
else:
|
|
self.done()
|
|
|
|
self.stop()
|
|
return
|
|
|
|
def abort(self):
|
|
self.debug('Aborting, sending SIGTERM to %d', self._popen.pid)
|
|
os.kill(self._popen.pid, signal.SIGTERM)
|
|
# self.stop()
|
|
|
|
def readbytesout(self, bytes):
|
|
"""
|
|
Called when bytes have been read from stdout.
|
|
"""
|
|
pass
|
|
|
|
def readbyteserr(self, bytes):
|
|
"""
|
|
Called when bytes have been read from stderr.
|
|
"""
|
|
pass
|
|
|
|
def done(self):
|
|
"""
|
|
Called when the command completed successfully.
|
|
"""
|
|
pass
|
|
|
|
def failed(self):
|
|
"""
|
|
Called when the command failed.
|
|
"""
|
|
pass
|
|
|
|
|
|
def commandMissing(self):
|
|
"""
|
|
Called when the command is missing.
|
|
"""
|
|
pass
|
|
|
|
|