Merge pull request #5969 from risaacson/pull_5136
Updates for the unarchive module and action_plugin.
This commit is contained in:
commit
25e12c87c0
1 changed files with 50 additions and 33 deletions
|
@ -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))
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue