Updates for the unarchive module and action_plugin.

There is a bit going on with the changes here. Most of the changes are cleanup of files so that they line up with the standard files.

PR #5136 was merged into the current devel and brought up to working order. A few bug fixes had to be done to get the code to test correctly. Thanks out to @pib!

Issue #5431 was not able to be confirmed as it behaved as expected with a sudo user.

Tests were added via a playbook with archive files to verify functionality.

All tests fire clean including custom playbooks across multiple linux and solaris systems.
pull/5969/head
Richard C Isaacson 11 years ago
parent e6c9705058
commit cb7c2b7524

@ -28,11 +28,9 @@ from ansible.runner.return_data import ReturnData
import sys import sys
reload(sys) reload(sys)
sys.setdefaultencoding("utf8") sys.setdefaultencoding("utf8")
#import base64
#import stat
#import tempfile
import pipes import pipes
class ActionModule(object): class ActionModule(object):
TRANSFERS_FILES = True TRANSFERS_FILES = True
@ -65,7 +63,7 @@ class ActionModule(object):
remote_md5 = self.runner._remote_md5(conn, tmp, dest) remote_md5 = self.runner._remote_md5(conn, tmp, dest)
if remote_md5 != '3': if remote_md5 != '3':
result = dict(failed=True, msg="dest must be an existing dir") result = dict(failed=True, msg="dest must be an existing dir", rc=remote_md5)
return ReturnData(conn=conn, result=result) return ReturnData(conn=conn, result=result)
if copy: if copy:

@ -65,8 +65,9 @@ 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):
@ -76,7 +77,7 @@ class _zipfile(object):
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)
@ -84,12 +85,13 @@ class _zipfile(object):
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):
@ -104,8 +106,8 @@ class _tgzfile(object):
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)
@ -116,9 +118,11 @@ class _tgzfile(object):
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:
if len(out.splitlines(True)) > 0:
return True 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):
@ -127,6 +131,7 @@ class _tarfile(_tgzfile):
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):
@ -135,6 +140,7 @@ class _tarbzip(_tgzfile):
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):
@ -143,6 +149,7 @@ class _tarxz(_tgzfile):
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]
@ -152,6 +159,7 @@ def pick_handler(src,dest,module):
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
@ -166,17 +174,18 @@ def main():
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):
if copy: if copy:
module.fail_json(msg="Source '%s' failed to transfer" % (src)) module.fail_json(msg="Source '%s' failed to transfer" % src)
else: else:
module.fail_json(msg="Source '%s' does not exist" % (src)) 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):
@ -187,16 +196,14 @@ def main():
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))

@ -82,6 +82,7 @@ class TestCallbacks(object):
def on_no_hosts(self): def on_no_hosts(self):
pass pass
class TestPlaybook(unittest.TestCase): class TestPlaybook(unittest.TestCase):
def setUp(self): def setUp(self):
@ -119,8 +120,7 @@ class TestPlaybook(unittest.TestCase):
filename = os.path.join(self.stage_dir, filename) filename = os.path.join(self.stage_dir, filename)
return filename return filename
def _run(self, test_playbook, host_list='test/ansible_hosts', def _run(self, test_playbook, host_list='test/ansible_hosts', extra_vars=None):
extra_vars=None):
''' run a module and get the localhost results ''' ''' run a module and get the localhost results '''
# This ensures tests are independent of eachother # This ensures tests are independent of eachother
global EVENTS global EVENTS
@ -410,6 +410,23 @@ class TestPlaybook(unittest.TestCase):
assert utils.jsonify(expected, format=True) == utils.jsonify(actual,format=True) assert utils.jsonify(expected, format=True) == utils.jsonify(actual,format=True)
def test_unarchive(self):
pb = 'test/playbook-unarchive.yml'
actual = self._run(pb)
expected = {
"localhost": {
"changed": 41,
"failures": 0,
"ok": 45,
"skipped": 0,
"unreachable": 0
}
}
assert utils.jsonify(expected, format=True) == utils.jsonify(actual,format=True)
def _compare_file_output(self, filename, expected_lines): def _compare_file_output(self, filename, expected_lines):
actual_lines = [] actual_lines = []
with open(filename) as f: with open(filename) as f:

@ -0,0 +1,78 @@
---
# To run me manually, use: -i "localhost,"
- hosts: localhost
connection: local
gather_facts: no
vars:
- testdir: /tmp/ansible-unarchive
- filesdir: test_unarchive/files
tasks:
- name: "Simple tar unarchive."
command: rm -rf {{testdir}}
- file: state=directory dest={{testdir}}
- unarchive: src={{filesdir}}/test.tar dest={{testdir}}
register: res
- command: test -f {{testdir}}/foo
- command: test "{{res.changed}}" = "True"
- unarchive: src={{filesdir}}/test.tar dest={{testdir}}
register: res
- command: test "{{res.changed}}" = "False"
- name: "Simple tar.gz unarchive."
command: rm -rf {{testdir}}
- file: state=directory dest={{testdir}}
- unarchive: src={{filesdir}}/test.tar.gz dest={{testdir}}
register: res
- command: test -f {{testdir}}/foo
- command: test "{{res.changed}}" = "True"
- unarchive: src={{filesdir}}/test.tar.gz dest={{testdir}}
register: res
- command: test "{{res.changed}}" = "False"
- name: "Simple zip unarchive."
command: rm -rf {{testdir}}
- file: state=directory dest={{testdir}}
- unarchive: src={{filesdir}}/test.zip dest={{testdir}}
register: res
- command: test -f {{testdir}}/foo
- command: test "{{res.changed}}" = "True"
- unarchive: src={{filesdir}}/test.zip dest={{testdir}}
register: res
- command: test "{{res.changed}}" = "True"
- name: "Unarchive a local tar file."
command : rm -rf {{testdir}}
- file: state=directory dest={{testdir}}
- copy: src={{filesdir}}/test.tar dest={{testdir}}
- unarchive: src={{testdir}}/test.tar dest={{testdir}}
register: res
- command: test -f {{testdir}}/foo
- command: test "{{res.changed}}" = "True"
- unarchive: src={{testdir}}/test.tar dest={{testdir}}
register: res
- command: test "{{res.changed}}" = "False"
- name: "Unarchive a local tar.gz file."
command : rm -rf {{testdir}}
- file: state=directory dest={{testdir}}
- copy: src={{filesdir}}/test.tar.gz dest={{testdir}}
- unarchive: src={{testdir}}/test.tar.gz dest={{testdir}}
register: res
- command: test -f {{testdir}}/foo
- command: test "{{res.changed}}" = "True"
- unarchive: src={{testdir}}/test.tar.gz dest={{testdir}}
register: res
- command: test "{{res.changed}}" = "False"
- name: "Unarchive a local zip file."
command : rm -rf {{testdir}}
- file: state=directory dest={{testdir}}
- copy: src={{filesdir}}/test.zip dest={{testdir}}
- unarchive: src={{testdir}}/test.zip dest={{testdir}}
register: res
- command: test -f {{testdir}}/foo
- command: test "{{res.changed}}" = "True"
- unarchive: src={{testdir}}/test.zip dest={{testdir}}
register: res
- command: test "{{res.changed}}" = "True"
Loading…
Cancel
Save