2363d6de43
git-svn-id: http://google-refine.googlecode.com/svn/trunk@517 7d457c2a-affb-35e4-300a-418c747d4874
225 lines
7.6 KiB
Python
225 lines
7.6 KiB
Python
"""distutils.spawn
|
|
|
|
Provides the 'spawn()' function, a front-end to various platform-
|
|
specific functions for launching another program in a sub-process.
|
|
Also provides the 'find_executable()' to search the path for a given
|
|
executable name.
|
|
"""
|
|
|
|
# This module should be kept compatible with Python 2.1.
|
|
|
|
__revision__ = "$Id: spawn.py 37828 2004-11-10 22:23:15Z loewis $"
|
|
|
|
import sys, os, string
|
|
from distutils.errors import *
|
|
from distutils import log
|
|
|
|
def spawn (cmd,
|
|
search_path=1,
|
|
verbose=0,
|
|
dry_run=0):
|
|
|
|
"""Run another program, specified as a command list 'cmd', in a new
|
|
process. 'cmd' is just the argument list for the new process, ie.
|
|
cmd[0] is the program to run and cmd[1:] are the rest of its arguments.
|
|
There is no way to run a program with a name different from that of its
|
|
executable.
|
|
|
|
If 'search_path' is true (the default), the system's executable
|
|
search path will be used to find the program; otherwise, cmd[0]
|
|
must be the exact path to the executable. If 'dry_run' is true,
|
|
the command will not actually be run.
|
|
|
|
Raise DistutilsExecError if running the program fails in any way; just
|
|
return on success.
|
|
"""
|
|
if os.name == 'posix':
|
|
_spawn_posix(cmd, search_path, dry_run=dry_run)
|
|
elif os.name == 'nt':
|
|
_spawn_nt(cmd, search_path, dry_run=dry_run)
|
|
elif os.name == 'os2':
|
|
_spawn_os2(cmd, search_path, dry_run=dry_run)
|
|
elif os.name == 'java':
|
|
_spawn_java(cmd, search_path, dry_run=dry_run)
|
|
else:
|
|
raise DistutilsPlatformError, \
|
|
"don't know how to spawn programs on platform '%s'" % os.name
|
|
|
|
# spawn ()
|
|
|
|
|
|
def _nt_quote_args (args):
|
|
"""Quote command-line arguments for DOS/Windows conventions: just
|
|
wraps every argument which contains blanks in double quotes, and
|
|
returns a new argument list.
|
|
"""
|
|
|
|
# XXX this doesn't seem very robust to me -- but if the Windows guys
|
|
# say it'll work, I guess I'll have to accept it. (What if an arg
|
|
# contains quotes? What other magic characters, other than spaces,
|
|
# have to be escaped? Is there an escaping mechanism other than
|
|
# quoting?)
|
|
|
|
for i in range(len(args)):
|
|
if string.find(args[i], ' ') != -1:
|
|
args[i] = '"%s"' % args[i]
|
|
return args
|
|
|
|
def _spawn_nt (cmd,
|
|
search_path=1,
|
|
verbose=0,
|
|
dry_run=0):
|
|
|
|
executable = cmd[0]
|
|
cmd = _nt_quote_args(cmd)
|
|
if search_path:
|
|
# either we find one or it stays the same
|
|
executable = find_executable(executable) or executable
|
|
log.info(string.join([executable] + cmd[1:], ' '))
|
|
if not dry_run:
|
|
# spawn for NT requires a full path to the .exe
|
|
try:
|
|
rc = os.spawnv(os.P_WAIT, executable, cmd)
|
|
except OSError, exc:
|
|
# this seems to happen when the command isn't found
|
|
raise DistutilsExecError, \
|
|
"command '%s' failed: %s" % (cmd[0], exc[-1])
|
|
if rc != 0:
|
|
# and this reflects the command running but failing
|
|
raise DistutilsExecError, \
|
|
"command '%s' failed with exit status %d" % (cmd[0], rc)
|
|
|
|
|
|
def _spawn_os2 (cmd,
|
|
search_path=1,
|
|
verbose=0,
|
|
dry_run=0):
|
|
|
|
executable = cmd[0]
|
|
#cmd = _nt_quote_args(cmd)
|
|
if search_path:
|
|
# either we find one or it stays the same
|
|
executable = find_executable(executable) or executable
|
|
log.info(string.join([executable] + cmd[1:], ' '))
|
|
if not dry_run:
|
|
# spawnv for OS/2 EMX requires a full path to the .exe
|
|
try:
|
|
rc = os.spawnv(os.P_WAIT, executable, cmd)
|
|
except OSError, exc:
|
|
# this seems to happen when the command isn't found
|
|
raise DistutilsExecError, \
|
|
"command '%s' failed: %s" % (cmd[0], exc[-1])
|
|
if rc != 0:
|
|
# and this reflects the command running but failing
|
|
print "command '%s' failed with exit status %d" % (cmd[0], rc)
|
|
raise DistutilsExecError, \
|
|
"command '%s' failed with exit status %d" % (cmd[0], rc)
|
|
|
|
|
|
def _spawn_posix (cmd,
|
|
search_path=1,
|
|
verbose=0,
|
|
dry_run=0):
|
|
|
|
log.info(string.join(cmd, ' '))
|
|
if dry_run:
|
|
return
|
|
exec_fn = search_path and os.execvp or os.execv
|
|
|
|
pid = os.fork()
|
|
|
|
if pid == 0: # in the child
|
|
try:
|
|
#print "cmd[0] =", cmd[0]
|
|
#print "cmd =", cmd
|
|
exec_fn(cmd[0], cmd)
|
|
except OSError, e:
|
|
sys.stderr.write("unable to execute %s: %s\n" %
|
|
(cmd[0], e.strerror))
|
|
os._exit(1)
|
|
|
|
sys.stderr.write("unable to execute %s for unknown reasons" % cmd[0])
|
|
os._exit(1)
|
|
|
|
|
|
else: # in the parent
|
|
# Loop until the child either exits or is terminated by a signal
|
|
# (ie. keep waiting if it's merely stopped)
|
|
while 1:
|
|
try:
|
|
(pid, status) = os.waitpid(pid, 0)
|
|
except OSError, exc:
|
|
import errno
|
|
if exc.errno == errno.EINTR:
|
|
continue
|
|
raise DistutilsExecError, \
|
|
"command '%s' failed: %s" % (cmd[0], exc[-1])
|
|
if os.WIFSIGNALED(status):
|
|
raise DistutilsExecError, \
|
|
"command '%s' terminated by signal %d" % \
|
|
(cmd[0], os.WTERMSIG(status))
|
|
|
|
elif os.WIFEXITED(status):
|
|
exit_status = os.WEXITSTATUS(status)
|
|
if exit_status == 0:
|
|
return # hey, it succeeded!
|
|
else:
|
|
raise DistutilsExecError, \
|
|
"command '%s' failed with exit status %d" % \
|
|
(cmd[0], exit_status)
|
|
|
|
elif os.WIFSTOPPED(status):
|
|
continue
|
|
|
|
else:
|
|
raise DistutilsExecError, \
|
|
"unknown error executing '%s': termination status %d" % \
|
|
(cmd[0], status)
|
|
# _spawn_posix ()
|
|
|
|
|
|
def _spawn_java(cmd,
|
|
search_path=1,
|
|
verbose=0,
|
|
dry_run=0):
|
|
executable = cmd[0]
|
|
cmd = ' '.join(_nt_quote_args(cmd))
|
|
log.info(cmd)
|
|
if not dry_run:
|
|
try:
|
|
rc = os.system(cmd) >> 8
|
|
except OSError, exc:
|
|
# this seems to happen when the command isn't found
|
|
raise DistutilsExecError, \
|
|
"command '%s' failed: %s" % (executable, exc[-1])
|
|
if rc != 0:
|
|
# and this reflects the command running but failing
|
|
print "command '%s' failed with exit status %d" % (executable, rc)
|
|
raise DistutilsExecError, \
|
|
"command '%s' failed with exit status %d" % (executable, rc)
|
|
|
|
|
|
def find_executable(executable, path=None):
|
|
"""Try to find 'executable' in the directories listed in 'path' (a
|
|
string listing directories separated by 'os.pathsep'; defaults to
|
|
os.environ['PATH']). Returns the complete filename or None if not
|
|
found.
|
|
"""
|
|
if path is None:
|
|
path = os.environ['PATH']
|
|
paths = string.split(path, os.pathsep)
|
|
(base, ext) = os.path.splitext(executable)
|
|
if (sys.platform == 'win32' or os.name == 'os2') and (ext != '.exe'):
|
|
executable = executable + '.exe'
|
|
if not os.path.isfile(executable):
|
|
for p in paths:
|
|
f = os.path.join(p, executable)
|
|
if os.path.isfile(f):
|
|
# the file exists, we have a shot at spawn working
|
|
return f
|
|
return None
|
|
else:
|
|
return executable
|
|
|
|
# find_executable()
|