blob: 9c6702a6b095baffb0f294175c37a37f21c16886 [file] [log] [blame]
# Copyright (C) 2003-2007 John Goerzen
# <jgoerzen@complete.org>
#
# This program 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 of the License, or
# (at your option) any later version.
#
# This program 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 program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
import os, sys, re
nulldev = None
verbose = 0
class ExecProblem(Exception):
pass
def mainexec(program, args = [], child_stdout = None,
child_stdin = None, child_stderr = None, wait = 1, closefds = []):
"""Runs the program as a sub-process, passing to it args if specified.
The sub-process has its file descriptors adjusted as per the arguments.
If wait is 1, wait until the child exits, then return the result code from
os.waitpid().
If wait is 0, return the PID immediately."""
global verbose
def setfds(source, dest):
if source != None:
if hasattr(source, 'fileno'):
source = source.fileno()
os.dup2(source, dest)
pid = os.fork()
if not pid:
if verbose:
print "Running: ", program, args
setfds(child_stdin, 0)
setfds(child_stdout, 1)
setfds(child_stderr, 2)
for fd in closefds:
os.close(fd)
os.execvp(program, (program,) + tuple(args))
sys.exit(255)
else:
if wait:
return os.waitpid(pid, 0)[1]
else:
return pid
def safeexec(program, args = [], child_stdout = None,
child_stdin = None, child_stderr = None,
expected = 0):
"""Calls mainexec() with the appropriate arguments, and raises
an exception if the program died with a signal or returned an
error code other than expected. This function will always wait."""
result = mainexec(program, args, child_stdout, child_stdin, child_stderr)
return checkresult(result, expected)
def getstdoutsafeexec(program, args, expected = 0):
pipes = os.pipe()
pid = mainexec(program, args, child_stdout = pipes[1], wait = 0)
os.close(pipes[1])
fd = os.fdopen(pipes[0], 'r')
retval = fd.readlines()
checkpid(pid, expected)
os.close(pipes[0])
return retval
def silentsafeexec(program, args, expected = 0):
"""Silently runs the specified program."""
null = getnull()
result = mainexec(program, args, null, null, null)
return checkresult(result, expected)
def checkresult(result, expected):
info = []
if os.WIFSIGNALED(result):
info.append("got signal %d" % os.WTERMSIG(result))
if os.WIFEXITED(result):
info.append("exited with code %d" % os.WEXITSTATUS(result))
info = ",".join(info)
if not os.WIFEXITED(result):
raise ExecProblem, info
if os.WEXITSTATUS(result) != expected:
raise ExecProblem, info + " (expected exit code %d)" % expected
return result
def checkpid(pid, expected):
return checkresult(os.waitpid(pid, 0)[1], expected)
def getnull():
global nulldev
if not nulldev:
nulldev = open("/dev/null", "w+")
return nulldev
def chdircmd(newdir, func, *args, **kwargs):
cwd = os.getcwd()
os.chdir(newdir)
try:
return apply(func, args, kwargs)
finally:
os.chdir(cwd)
def maketree(path, addpath = None, ignore = [], res = None):
thisdir = os.listdir(path)
retval = []
others = []
if res == None:
res = [re.compile(x) for x in ignore]
for item in thisdir:
skip = 0
for retest in res:
if retest.search(item):
skip = 1
break
if skip:
continue
dirname = os.path.join(path, item)
if os.path.isdir(dirname) and not os.path.islink(dirname):
if addpath:
retval.append(os.path.join(addpath, item) + '/')
else:
retval.append(item + '/')
if addpath:
newaddpath = os.path.join(addpath, item)
else:
newaddpath = item
others.extend(maketree(dirname, newaddpath, res = res))
else:
if addpath:
retval.append(os.path.join(addpath, item))
else:
retval.append(item)
return sorttree(retval + others)
def sorttree(srctree, filesfirst = False):
retval = []
dirs = [x for x in srctree if x.endswith('/')]
files = [x for x in srctree if not x.endswith('/')]
dirs.sort()
files.sort()
if filesfirst:
return files + dirs
else:
return dirs + files
def copyfrom(srcdir, destdir):
pipes = os.pipe()
verbargs = []
#if verbose:
# verbargs.append('-v')
readerpid = chdircmd(srcdir, mainexec, "tar", ["-cSpf", "-", "."],
child_stdout = pipes[1], wait = 0,
closefds = [pipes[0]])
writerpid = chdircmd(destdir, mainexec, "tar", ["-xSpf", "-"] + verbargs,
child_stdin = pipes[0], wait = 0, closefds = [pipes[1]])
os.close(pipes[0])
os.close(pipes[1])
checkpid(readerpid, 0)
checkpid(writerpid, 0)