Move backup to module_common

This commit is contained in:
Daniel Hokka Zakrisson 2012-09-04 14:16:30 +02:00
parent 7c2fe3da16
commit dfcb9d3c2d
2 changed files with 15 additions and 18 deletions

View file

@ -47,6 +47,8 @@ import subprocess
import sys import sys
import syslog import syslog
import types import types
import time
import shutil
try: try:
from hashlib import md5 as _md5 from hashlib import md5 as _md5
@ -274,6 +276,18 @@ class AnsibleModule(object):
infile.close() infile.close()
return digest.hexdigest() return digest.hexdigest()
def backuplocal(self, 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:
self.fail_json(msg='Could not make backup of %s to %s: %s' % (fn, backupdest, e))
return backupdest
# == END DYNAMICALLY INSERTED CODE === # == END DYNAMICALLY INSERTED CODE ===

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))