diff options
Diffstat (limited to 'recipes/stage-manager/files/stage-manager')
-rwxr-xr-x | recipes/stage-manager/files/stage-manager | 156 |
1 files changed, 156 insertions, 0 deletions
diff --git a/recipes/stage-manager/files/stage-manager b/recipes/stage-manager/files/stage-manager new file mode 100755 index 0000000000..536d1afda0 --- /dev/null +++ b/recipes/stage-manager/files/stage-manager @@ -0,0 +1,156 @@ +#!/usr/bin/env python + +# Copyright (C) 2006-2007 Richard Purdie +# +# This program is free software; you can redistribute it and/or modify it under +# the terms of the GNU General Public License version 2 as published by the Free +# Software Foundation; +# +# 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. + +import optparse +import os, sys, stat + +__version__ = "0.0.1" + + +def write_cache(cachefile, cachedata): + fd = open(cachefile, 'w') + for f in cachedata: + s = f + '|' + str(cachedata[f]['ts']) + '|' + str(cachedata[f]['size']) + fd.write(s + '\n') + fd.close() + +def read_cache(cachefile): + cache = {} + f = open(cachefile, 'r') + lines = f.readlines() + f.close() + for l in lines: + data = l.split('|') + cache[data[0]] = {} + cache[data[0]]['ts'] = int(data[1]) + cache[data[0]]['size'] = int(data[2]) + cache[data[0]]['seen'] = False + return cache + +def mkdirhier(dir): + """Create a directory like 'mkdir -p', but does not complain if + directory already exists like os.makedirs + """ + try: + os.makedirs(dir) + except OSError, e: + if e.errno != 17: raise e + +if __name__ == "__main__": + parser = optparse.OptionParser( version = "Metadata Stage Manager version %s" % ( __version__ ), + usage = """%prog [options]\n\nPerforms mamagement tasks on a metadata staging area.""" ) + + parser.add_option( "-p", "--parentdir", help = "the path to the metadata parent directory", + action = "store", dest = "parentdir", default = None) + + parser.add_option( "-c", "--cachefile", help = "the cache file to use", + action = "store", dest = "cachefile", default = None) + + parser.add_option( "-d", "--copydir", help = "copy changed files to this directory", + action = "store", dest = "copydir", default = None) + + parser.add_option( "-u", "--update", help = "update the cache file", + action = "store_true", dest = "update", default = False) + + (options, args) = parser.parse_args() + + if options.parentdir is None: + print("Error, --parentdir option not supplied") + sys.exit(1) + + if options.cachefile is None: + print("Error, --cachefile option not supplied") + sys.exit(1) + + if not options.parentdir.endswith('/'): + options.parentdir = options.parentdir + '/' + + cache = {} + if os.access(options.cachefile, os.F_OK): + cache = read_cache(options.cachefile) + + found_difference = False + + def updateCache(path, fstamp): + cache[path] = {} + cache[path]['ts'] = fstamp[stat.ST_MTIME] + cache[path]['size'] = fstamp[stat.ST_SIZE] + cache[path]['seen'] = True + found_difference = True + + def copyfile(path): + if options.copydir: + copypath = os.path.join(options.copydir, path.replace(options.parentdir, '', 1)) + mkdirhier(os.path.split(copypath)[0]) + os.system("cp -dp " + path + " " + copypath) + + def copydir(path, fstamp): + if options.copydir: + copypath = os.path.join(options.copydir, path.replace(options.parentdir, '', 1)) + if os.path.exists(copypath): + os.system("rm -rf " + copypath) + if os.path.islink(path): + os.symlink(os.readlink(path), copypath) + else: + mkdirhier(copypath) + os.utime(copypath, (fstamp[stat.ST_ATIME], fstamp[stat.ST_MTIME])) + + for root, dirs, files in os.walk(options.parentdir): + for f in files: + path = os.path.join(root, f) + if not os.access(path, os.R_OK): + continue + fstamp = os.lstat(path) + if path not in cache: + print "new file %s" % path + updateCache(path, fstamp) + copyfile(path) + else: + if cache[path]['ts'] != fstamp[stat.ST_MTIME] or cache[path]['size'] != fstamp[stat.ST_SIZE]: + print "file %s changed" % path + updateCache(path, fstamp) + copyfile(path) + cache[path]['seen'] = True + for d in dirs: + path = os.path.join(root, d) + fstamp = os.lstat(path) + if path not in cache: + print "new dir %s" % path + updateCache(path, fstamp) + copydir(path, fstamp) + else: + if cache[path]['ts'] != fstamp[stat.ST_MTIME]: + print "dir %s changed" % path + updateCache(path, fstamp) + copydir(path, fstamp) + cache[path]['seen'] = True + + todel = [] + for path in cache: + if not cache[path]['seen']: + print "%s removed" % path + found_difference = True + todel.append(path) + + if options.update: + print "Updating" + for path in todel: + del cache[path] + mkdirhier(os.path.split(options.cachefile)[0]) + write_cache(options.cachefile, cache) + + if found_difference: + sys.exit(5) + sys.exit(0) + + + |