Merge branch 'backup-common' of https://github.com/dhozac/ansible into devel

This commit is contained in:
Michael DeHaan 2012-09-04 19:45:39 -04:00
commit de96d04093
3 changed files with 15 additions and 22 deletions

View file

@ -50,6 +50,7 @@ def main():
argument_spec = dict( argument_spec = dict(
src = dict(required=True), src = dict(required=True),
dest = dict(required=True), dest = dict(required=True),
backup=dict(default=False, choices=BOOLEANS),
) )
) )
@ -58,6 +59,7 @@ def main():
destmd5 = None destmd5 = None
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'])
backup = module.boolean(module.params.get('backup', False))
if not os.path.exists(src): if not os.path.exists(src):
module.fail_json(msg="Source (%s) does not exist" % src) module.fail_json(msg="Source (%s) does not exist" % src)
@ -72,6 +74,8 @@ def main():
destmd5 = module.md5(dest) destmd5 = module.md5(dest)
if pathmd5 != destmd5: if pathmd5 != destmd5:
if backup and destmd5 is not None:
module.backuplocal(dest)
shutil.copy(path, dest) shutil.copy(path, dest)
changed = True changed = True

19
copy
View file

@ -20,19 +20,6 @@
import os import os
import shutil import shutil
import time
def backuplocal(fn):
"""make a date-marked backup of the specified file, return True or False on success or failure"""
# backups named basename-YYYY-MM-DD@HH:MM~
ext = time.strftime("%Y-%m-%d@%H:%M~", time.localtime(time.time()))
backupdest = '%s.%s' % (fn, ext)
try:
shutil.copy2(fn, backupdest)
except shutil.Error, e:
return False, 'Could not make backup of %s to %s: %s' % (fn, backupdest, e)
return True, backupdest
def main(): def main():
@ -76,11 +63,7 @@ def main():
try: try:
if backup: if backup:
if os.path.exists(dest): if os.path.exists(dest):
success, msg = backuplocal(dest) backup_file = module.backuplocal(dest)
if not success:
module.fail_jason(msg=msg)
else:
backup_file = msg
shutil.copyfile(src, dest) shutil.copyfile(src, dest)
except shutil.Error: except shutil.Error:
module.fail_json(msg="failed to copy: %s and %s are the same" % (src, dest)) module.fail_json(msg="failed to copy: %s and %s are the same" % (src, dest))

View file

@ -21,7 +21,7 @@
import re import re
import os import os
def present(module, name, regexp, line, insertafter): def present(module, name, regexp, line, insertafter, backup):
f = open(name, 'rb') f = open(name, 'rb')
lines = f.readlines() lines = f.readlines()
f.close() f.close()
@ -69,13 +69,15 @@ def present(module, name, regexp, line, insertafter):
changed = True changed = True
if changed: if changed:
if backup:
module.backuplocal(name)
f = open(name, 'wb') f = open(name, 'wb')
f.writelines(lines) f.writelines(lines)
f.close() f.close()
module.exit_json(changed=changed, msg=msg) module.exit_json(changed=changed, msg=msg)
def absent(module, name, regexp): def absent(module, name, regexp, backup):
f = open(name, 'rb') f = open(name, 'rb')
lines = f.readlines() lines = f.readlines()
f.close() f.close()
@ -90,6 +92,8 @@ def absent(module, name, regexp):
lines = filter(matcher, lines) lines = filter(matcher, lines)
changed = len(found) > 0 changed = len(found) > 0
if changed: if changed:
if backup:
module.backuplocal(name)
f = open(name, 'wb') f = open(name, 'wb')
f.writelines(lines) f.writelines(lines)
f.close() f.close()
@ -103,18 +107,20 @@ def main():
regexp=dict(required=True), regexp=dict(required=True),
line=dict(aliases=['value']), line=dict(aliases=['value']),
insertafter=dict(default='EOF'), insertafter=dict(default='EOF'),
backup=dict(default=False, choices=BOOLEANS),
), ),
) )
params = module.params params = module.params
backup = module.boolean(module.params.get('backup', False))
if params['state'] == 'present': if params['state'] == 'present':
if 'line' not in params: if 'line' not in params:
module.fail_json(msg='line= is required with state=present') module.fail_json(msg='line= is required with state=present')
present(module, params['name'], params['regexp'], params['line'], present(module, params['name'], params['regexp'], params['line'],
params['insertafter']) params['insertafter'], backup)
else: else:
absent(module, params['name'], params['regexp']) absent(module, params['name'], params['regexp'], backup)
# this is magic, see lib/ansible/module_common.py # this is magic, see lib/ansible/module_common.py
#<<INCLUDE_ANSIBLE_MODULE_COMMON>> #<<INCLUDE_ANSIBLE_MODULE_COMMON>>