mirror of https://github.com/tLDP/python-tldp
141 lines
4.2 KiB
Python
141 lines
4.2 KiB
Python
#! /usr/bin/python
|
|
|
|
from __future__ import absolute_import, division, print_function
|
|
|
|
import os
|
|
import io
|
|
import sys
|
|
import errno
|
|
import operator
|
|
import subprocess
|
|
import functools
|
|
from tempfile import mkstemp
|
|
import logging
|
|
|
|
|
|
def getLogger(**opts):
|
|
level = opts.get('level', logging.INFO)
|
|
logging.basicConfig(stream=sys.stderr, level=level)
|
|
logger = logging.getLogger()
|
|
return logger
|
|
|
|
logger = getLogger()
|
|
|
|
|
|
def execute(cmd, stdin=None, stdout=None, stderr=None,
|
|
logdir=None, env=os.environ):
|
|
prefix = os.path.basename(cmd[0]) + '.' + str(os.getpid()) + '-'
|
|
|
|
if logdir is None:
|
|
raise Exception("Missing required parameter: logdir.")
|
|
|
|
if not os.path.isdir(logdir):
|
|
raise IOError(errno.ENOENT, os.strerror(errno.ENOENT), logdir)
|
|
|
|
# -- not remapping STDIN, because that doesn't make sense here
|
|
if stdout is None:
|
|
stdout, stdoutname = mkstemp(prefix=prefix, suffix='.stdout',
|
|
dir=logdir)
|
|
if stderr is None:
|
|
stderr, stderrname = mkstemp(prefix=prefix, suffix='.stderr',
|
|
dir=logdir)
|
|
|
|
logger.debug("About to execute: %r", cmd)
|
|
proc = subprocess.Popen(cmd, shell=False, close_fds=True,
|
|
stdin=stdin, stdout=stdout, stderr=stderr,
|
|
env=env, preexec_fn=os.setsid)
|
|
result = proc.wait()
|
|
if result != 0:
|
|
logger.warning("Return code (%s) for process: %r", result, cmd)
|
|
logger.warning("Find STDOUT/STDERR in %s/%s", logdir, prefix)
|
|
return result
|
|
|
|
|
|
def is_executable(fpath):
|
|
return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
|
|
|
|
|
|
def which(program):
|
|
'''return None or the full path to an executable (respecting $PATH)
|
|
http://stackoverflow.com/questions/377017/test-if-executable-exists-in-python/377028#377028
|
|
'''
|
|
fpath, fname = os.path.split(program)
|
|
if fpath and is_executable(program):
|
|
return program
|
|
else:
|
|
for path in os.environ["PATH"].split(os.pathsep):
|
|
path = path.strip('"')
|
|
sut = os.path.join(path, program)
|
|
if is_executable(sut):
|
|
return sut
|
|
return None
|
|
|
|
|
|
def makefh(thing):
|
|
if isinstance(thing, io.IOBase):
|
|
f = thing
|
|
elif isinstance(thing, str) and os.path.isfile(thing):
|
|
f = open(thing)
|
|
else:
|
|
raise TypeError("Cannot make file from %s of %r" %
|
|
(type(thing), thing,))
|
|
return f
|
|
|
|
|
|
def getfileset(dirname):
|
|
statinfo = statfiles(dirname)
|
|
return set(statinfo.keys())
|
|
|
|
|
|
def statfiles(dirname):
|
|
statinfo = dict()
|
|
ocwd = os.getcwd()
|
|
os.chdir(dirname)
|
|
for root, dirs, files in os.walk('.'):
|
|
for x in files:
|
|
relpath = os.path.join(root, x)
|
|
try:
|
|
statinfo[relpath] = os.stat(relpath)
|
|
except OSError as e:
|
|
if e.errno != errno.ENOENT: # -- ho-hum, race condition
|
|
raise e
|
|
os.chdir(ocwd)
|
|
return statinfo
|
|
|
|
|
|
def att_statinfo(statinfo, attr='st_mtime', func=max):
|
|
if statinfo:
|
|
return func([getattr(v, attr) for v in statinfo.values()])
|
|
else:
|
|
return 0
|
|
|
|
|
|
max_size = functools.partial(att_statinfo, attr='st_size', func=max)
|
|
min_size = functools.partial(att_statinfo, attr='st_size', func=min)
|
|
|
|
max_mtime = functools.partial(att_statinfo, attr='st_mtime', func=max)
|
|
min_mtime = functools.partial(att_statinfo, attr='st_mtime', func=min)
|
|
|
|
max_ctime = functools.partial(att_statinfo, attr='st_ctime', func=max)
|
|
min_ctime = functools.partial(att_statinfo, attr='st_ctime', func=min)
|
|
|
|
max_atime = functools.partial(att_statinfo, attr='st_atime', func=max)
|
|
min_atime = functools.partial(att_statinfo, attr='st_atime', func=min)
|
|
|
|
|
|
def sieve(operand, statinfo, attr='st_mtime', func=operator.gt):
|
|
result = set()
|
|
for fname, stbuf in statinfo.items():
|
|
if func(getattr(stbuf, attr), operand):
|
|
result.add(fname)
|
|
return result
|
|
|
|
mtime_gt = functools.partial(sieve, attr='st_mtime', func=operator.gt)
|
|
mtime_lt = functools.partial(sieve, attr='st_mtime', func=operator.lt)
|
|
|
|
size_gt = functools.partial(sieve, attr='st_size', func=operator.gt)
|
|
size_lt = functools.partial(sieve, attr='st_size', func=operator.lt)
|
|
|
|
#
|
|
# -- end of file
|