102 lines
2.9 KiB
Python
Executable file
102 lines
2.9 KiB
Python
Executable file
#!/usr/bin/python
|
|
|
|
# (c) 2012, Michael DeHaan <michael.dehaan@gmail.com>
|
|
#
|
|
# This file is part of Ansible
|
|
#
|
|
# Ansible is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Ansible is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
import sys
|
|
import os
|
|
import shlex
|
|
import syslog
|
|
|
|
# ===========================================
|
|
# convert arguments of form a=b c=d
|
|
# to a dictionary
|
|
# FIXME: make more idiomatic
|
|
|
|
def dump_kv(vars):
|
|
return " ".join("%s='%s'" % (k,v) for (k,v) in vars.items())
|
|
|
|
def exit_kv(rc=0, **kwargs):
|
|
if 'path' in kwargs:
|
|
add_path_info(kwargs)
|
|
print dump_kv(kwargs)
|
|
sys.exit(rc)
|
|
|
|
def md5_sum(f):
|
|
md5sum = os.popen("/usr/bin/md5sum %(file)s 2>/dev/null || /sbin/md5 -q %(file)s 2>/dev/null || /usr/bin/digest -a md5 -v %(file)s 2>/dev/null" % {"file": f}).read().split()[0]
|
|
return md5sum
|
|
|
|
if len(sys.argv) == 1:
|
|
exit_kv(rc=1, failed=1, msg="incorrect number of arguments given")
|
|
|
|
argfile = sys.argv[1]
|
|
if not os.path.exists(argfile):
|
|
exit_kv(rc=1, failed=1, msg="file %s does not exist" % (argfile))
|
|
|
|
args = open(argfile, 'r').read()
|
|
items = shlex.split(args)
|
|
syslog.openlog('ansible-%s' % os.path.basename(__file__))
|
|
syslog.syslog(syslog.LOG_NOTICE, 'Invoked with %s' % args)
|
|
|
|
|
|
params = {}
|
|
for x in items:
|
|
(k, v) = x.split("=")
|
|
params[k] = v
|
|
|
|
src = params['src']
|
|
dest = params['dest']
|
|
if src:
|
|
src = os.path.expanduser(src)
|
|
if dest:
|
|
dest = os.path.expanduser(dest)
|
|
|
|
# raise an error if there is no src file
|
|
if not os.path.exists(src):
|
|
exit_kv(rc=1, failed=1, msg="Source %s failed to transfer" % (src))
|
|
|
|
# raise an error if there is no dest file
|
|
if not os.path.exists(dest):
|
|
exit_kv(rc=1, failed=1, msg="Destination %s does not exist" % (dest))
|
|
|
|
# raise an error if copy has no permission on dest
|
|
if not os.access(dest, os.W_OK):
|
|
exit_kv(rc=1, failed=1, msg="Destination %s not writable" % (dest))
|
|
elif not os.access(dest, os.R_OK):
|
|
exit_kv(rc=1, failed=1, msg="Destination %s not readable" % (dest))
|
|
|
|
md5sum_src = None
|
|
md5sum_src = md5_sum(src)
|
|
|
|
md5sum_dest = None
|
|
try:
|
|
md5sum_dest = md5_sum(dest)
|
|
except IndexError:
|
|
# File doesn't exist yet, so clear to continue
|
|
pass
|
|
|
|
if md5sum_src != md5sum_dest:
|
|
os.system("cp %s %s" % (src, dest))
|
|
changed = True
|
|
else:
|
|
changed = False
|
|
|
|
# mission accomplished
|
|
#print "md5sum=%s changed=%s" % (md5sum_dest, changed)
|
|
exit_kv(dest=dest, src=src, changed="md5sum=%s changed=%s" % (md5sum_dest, changed))
|
|
|