Merge pull request #5969 from risaacson/pull_5136

Updates for the unarchive module and action_plugin.
This commit is contained in:
Richard Isaacson 2014-02-12 10:10:25 -06:00
commit 25e12c87c0

View file

@ -37,6 +37,12 @@ options:
- Remote absolute path where the archive should be unpacked - Remote absolute path where the archive should be unpacked
required: true required: true
default: null default: null
copy:
description:
- Should the file be copied from the local to the remote machine?
required: false
choices: [ "yes", "no" ]
default: "yes"
author: Dylan Martin author: Dylan Martin
todo: todo:
- detect changed/unchanged for .zip files - detect changed/unchanged for .zip files
@ -59,35 +65,37 @@ EXAMPLES = '''
- unarchive: src=foo.tgz dest=/var/lib/foo - unarchive: src=foo.tgz dest=/var/lib/foo
''' '''
import os import os
# class to handle .zip files # class to handle .zip files
class _zipfile(object): class ZipFile(object):
def __init__(self,src,dest,module): def __init__(self, src, dest, module):
self.src = src self.src = src
self.dest = dest self.dest = dest
self.module = module self.module = module
def is_unarchived(self): def is_unarchived(self):
return dict(bool = False) return dict(unarchived=False)
def unarchive(self): def unarchive(self):
cmd = 'unzip -o "%s" -d "%s"' % (self.src,self.dest) cmd = 'unzip -o "%s" -d "%s"' % (self.src, self.dest)
rc, out, err = self.module.run_command(cmd) rc, out, err = self.module.run_command(cmd)
return dict(cmd = cmd, rc=rc, out=out, err=err) return dict(cmd=cmd, rc=rc, out=out, err=err)
def can_handle_archive(self): def can_handle_archive(self):
cmd = 'unzip -l "%s"' % (self.src) cmd = 'unzip -l "%s"' % self.src
rc, out, err = self.module.run_command(cmd) rc, out, err = self.module.run_command(cmd)
if rc == 0: if rc == 0:
return True return True
return False return False
# class to handle gzipped tar files # class to handle gzipped tar files
class _tgzfile(object): class TgzFile(object):
def __init__(self,src,dest,module): def __init__(self, src, dest, module):
self.src = src self.src = src
self.dest = dest self.dest = dest
self.module = module self.module = module
@ -96,56 +104,62 @@ class _tgzfile(object):
def is_unarchived(self): def is_unarchived(self):
dirof = os.path.dirname(self.dest) dirof = os.path.dirname(self.dest)
destbase = os.path.basename(self.dest) destbase = os.path.basename(self.dest)
cmd = 'tar -v -C "%s" --diff -%sf "%s"' % (self.dest, self.zipflag,self.src) cmd = 'tar -v -C "%s" --diff -%sf "%s"' % (self.dest, self.zipflag, self.src)
rc, out, err = self.module.run_command(cmd) rc, out, err = self.module.run_command(cmd)
bool = (rc == 0) unarchived = (rc == 0)
return dict( bool = bool, rc = rc , out = out, err = err, cmd = cmd) return dict(unarchived=unarchived, rc=rc, out=out, err=err, cmd=cmd)
def unarchive(self): def unarchive(self):
cmd = 'tar -C "%s" -x%sf "%s"' % (self.dest,self.zipflag,self.src) cmd = 'tar -C "%s" -x%sf "%s"' % (self.dest, self.zipflag, self.src)
rc, out, err = self.module.run_command(cmd) rc, out, err = self.module.run_command(cmd)
return dict(cmd = cmd, rc=rc, out=out, err=err) return dict(cmd=cmd, rc=rc, out=out, err=err)
def can_handle_archive(self): def can_handle_archive(self):
cmd = 'tar -t%sf "%s"' % (self.zipflag,self.src) cmd = 'tar -t%sf "%s"' % (self.zipflag, self.src)
rc, out, err = self.module.run_command(cmd) rc, out, err = self.module.run_command(cmd)
if rc == 0: if rc == 0:
return True if len(out.splitlines(True)) > 0:
return True
return False return False
# class to handle tar files that aren't compressed # class to handle tar files that aren't compressed
class _tarfile(_tgzfile): class TarFile(TgzFile):
def __init__(self,src,dest,module): def __init__(self, src, dest, module):
self.src = src self.src = src
self.dest = dest self.dest = dest
self.module = module self.module = module
self.zipflag = '' self.zipflag = ''
# class to handle bzip2 compressed tar files # class to handle bzip2 compressed tar files
class _tarbzip(_tgzfile): class TarBzip(TgzFile):
def __init__(self,src,dest,module): def __init__(self, src, dest, module):
self.src = src self.src = src
self.dest = dest self.dest = dest
self.module = module self.module = module
self.zipflag = 'j' self.zipflag = 'j'
# class to handle xz compressed tar files # class to handle xz compressed tar files
class _tarxz(_tgzfile): class TarXz(TgzFile):
def __init__(self,src,dest,module): def __init__(self, src, dest, module):
self.src = src self.src = src
self.dest = dest self.dest = dest
self.module = module self.module = module
self.zipflag = 'J' self.zipflag = 'J'
# try handlers in order and return the one that works or bail if none work # try handlers in order and return the one that works or bail if none work
def pick_handler(src,dest,module): def pick_handler(src, dest, module):
handlers = [_tgzfile, _zipfile, _tarfile, _tarbzip, _tarxz] handlers = [TgzFile, ZipFile, TarFile, TarBzip, TarXz]
for handler in handlers: for handler in handlers:
obj = handler(src,dest,module) obj = handler(src, dest, module)
if obj.can_handle_archive(): if obj.can_handle_archive():
return obj return obj
raise RuntimeError('Failed to find handler to unarchive "%s"' % src) raise RuntimeError('Failed to find handler to unarchive "%s"' % src)
def main(): def main():
module = AnsibleModule( module = AnsibleModule(
# not checking because of daisy chain to file module # not checking because of daisy chain to file module
@ -153,40 +167,43 @@ def main():
src = dict(required=True), src = dict(required=True),
original_basename = dict(required=False), # used to handle 'dest is a directory' via template, a slight hack original_basename = dict(required=False), # used to handle 'dest is a directory' via template, a slight hack
dest = dict(required=True), dest = dict(required=True),
copy = dict(default=True, type='bool'),
), ),
add_file_common_args=True, add_file_common_args=True,
) )
src = os.path.expanduser(module.params['src']) src = os.path.expanduser(module.params['src'])
dest = os.path.expanduser(module.params['dest']) dest = os.path.expanduser(module.params['dest'])
copy = module.params['copy']
# did tar file arrive? # did tar file arrive?
if not os.path.exists(src): if not os.path.exists(src):
module.fail_json(msg="Source '%s' failed to transfer" % (src)) if copy:
module.fail_json(msg="Source '%s' failed to transfer" % src)
else:
module.fail_json(msg="Source '%s' does not exist" % src)
if not os.access(src, os.R_OK): if not os.access(src, os.R_OK):
module.fail_json(msg="Source '%s' not readable" % (src)) module.fail_json(msg="Source '%s' not readable" % src)
# is dest OK to recieve tar file? # is dest OK to receive tar file?
if not os.path.exists(os.path.dirname(dest)): if not os.path.exists(os.path.dirname(dest)):
module.fail_json(msg="Destination directory '%s' does not exist" % (os.path.dirname(dest))) module.fail_json(msg="Destination directory '%s' does not exist" % (os.path.dirname(dest)))
if not os.access(os.path.dirname(dest), os.W_OK): if not os.access(os.path.dirname(dest), os.W_OK):
module.fail_json(msg="Destination '%s' not writable" % (os.path.dirname(dest))) module.fail_json(msg="Destination '%s' not writable" % (os.path.dirname(dest)))
handler = pick_handler(src,dest,module) handler = pick_handler(src, dest, module)
res_args = dict( handler=handler.__class__.__name__, dest = dest, src = src ) res_args = dict(handler=handler.__class__.__name__, dest=dest, src=src)
# do we need to do unpack? # do we need to do unpack?
namelist = ['bool','rc','out','err','cmd']
res_args['check_results'] = handler.is_unarchived() res_args['check_results'] = handler.is_unarchived()
if res_args['check_results']['bool']: if res_args['check_results']['unarchived']:
res_args['changed'] = False res_args['changed'] = False
module.exit_json(**res_args) module.exit_json(**res_args)
# do the unpack # do the unpack
try: try:
results = handler.unarchive() results = handler.unarchive()
#results = (src,dest,module)
except IOError: except IOError:
module.fail_json(msg="failed to unpack %s to %s" % (src, dest)) module.fail_json(msg="failed to unpack %s to %s" % (src, dest))