Simplify ansible-pull to just invoke ansible and ansible-playbook
This eliminates the creation of runner and playbook instances and just invokes the processes ansible and ansible-playbook.
This commit is contained in:
parent
23ff967f38
commit
84c9caa805
1 changed files with 25 additions and 108 deletions
133
bin/ansible-pull
133
bin/ansible-pull
|
@ -16,132 +16,49 @@
|
||||||
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
import os
|
import os
|
||||||
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
import socket
|
from optparse import OptionParser
|
||||||
import logging
|
|
||||||
|
|
||||||
import ansible.playbook
|
|
||||||
import ansible.runner
|
|
||||||
import ansible.constants as C
|
|
||||||
from ansible import errors
|
|
||||||
from ansible import callbacks
|
|
||||||
from ansible import utils
|
|
||||||
from ansible import inventory
|
|
||||||
|
|
||||||
DEFAULT_PLAYBOOK = 'local.yml'
|
DEFAULT_PLAYBOOK = 'local.yml'
|
||||||
|
|
||||||
|
def _run(cmd):
|
||||||
|
cmd = subprocess.Popen(cmd, shell=True,
|
||||||
|
stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||||||
|
(out, err) = cmd.communicate()
|
||||||
|
print out
|
||||||
|
if cmd.returncode != 0:
|
||||||
|
print err
|
||||||
|
return cmd.returncode
|
||||||
|
|
||||||
def main(args):
|
def main(args):
|
||||||
""" Set up and run a local playbook """
|
""" Set up and run a local playbook """
|
||||||
usage = "%prog [options]"
|
usage = "%prog [options]"
|
||||||
parser = utils.base_parser(constants=C, usage=usage,
|
parser = OptionParser()
|
||||||
connect_opts=False, runas_opts=False)
|
|
||||||
parser.set_defaults(module_name='git', transport='local',
|
|
||||||
one_line=False, tree=None)
|
|
||||||
parser.add_option('-d', '--directory', dest='dest', default=None,
|
parser.add_option('-d', '--directory', dest='dest', default=None,
|
||||||
help='Directory to checkout git repository')
|
help='Directory to checkout git repository')
|
||||||
parser.add_option('-U', '--url', dest='url', default=None,
|
parser.add_option('-U', '--url', dest='url',
|
||||||
|
default=None,
|
||||||
help='URL of git repository')
|
help='URL of git repository')
|
||||||
parser.add_option('-C', '--checkout', dest='checkout', default="HEAD",
|
parser.add_option('-C', '--checkout', dest='checkout',
|
||||||
|
default="HEAD",
|
||||||
help='Branch/Tag/Commit to checkout. Defaults to HEAD.')
|
help='Branch/Tag/Commit to checkout. Defaults to HEAD.')
|
||||||
parser.remove_option('-k') # Remove ssh password option
|
|
||||||
parser.remove_option('-K') # Remove sudo password option
|
|
||||||
parser.remove_option('-T') # Remove ssh timeout option
|
|
||||||
options, args = parser.parse_args(args)
|
options, args = parser.parse_args(args)
|
||||||
|
|
||||||
clirunner_cb = callbacks.CliRunnerCallbacks()
|
|
||||||
clirunner_cb.options = options
|
|
||||||
|
|
||||||
# ----------------------------------------------
|
|
||||||
# First git clone/pull
|
|
||||||
|
|
||||||
git_opts = "repo=%s dest=%s version=%s" % (options.url, options.dest, options.checkout)
|
git_opts = "repo=%s dest=%s version=%s" % (options.url, options.dest, options.checkout)
|
||||||
pattern = "localhost"
|
cmd = 'ansible all -c local -m git -a "%s"' % git_opts
|
||||||
inventory_manager = inventory.Inventory([pattern])
|
rc = _run(cmd)
|
||||||
|
if rc != 0:
|
||||||
|
return rc
|
||||||
|
|
||||||
"""
|
|
||||||
Ideally, changes should be reported via logging and not to STDOUT
|
|
||||||
"""
|
|
||||||
|
|
||||||
runner = ansible.runner.Runner(
|
|
||||||
module_name=options.module_name,
|
|
||||||
module_args=git_opts,
|
|
||||||
module_path=options.module_path,
|
|
||||||
inventory=inventory_manager,
|
|
||||||
forks=options.forks,
|
|
||||||
pattern=pattern,
|
|
||||||
callbacks=clirunner_cb,
|
|
||||||
transport=options.transport,
|
|
||||||
debug=options.debug
|
|
||||||
)
|
|
||||||
try:
|
|
||||||
runner.run()
|
|
||||||
except errors.AnsibleError, e:
|
|
||||||
print >>sys.stderr, "ERROR: %s" % e
|
|
||||||
return 1
|
|
||||||
|
|
||||||
# ----------------------------------------------
|
|
||||||
# Second, run the playbook
|
|
||||||
"""
|
|
||||||
Change to the directory where the git checkout is located.
|
|
||||||
Insert 'local.yml' as the first playbook to be run. This
|
|
||||||
supports multiple playbooks being supplied on the CLI, similar
|
|
||||||
to ansible-playbook. This then loops on all the playbooks,
|
|
||||||
instantiates and runs a playbook. A couple things of note:
|
|
||||||
* The transport uses the default set above, local
|
|
||||||
* The host_list argument to Playbook is set to a list of
|
|
||||||
names. These are localhost, the fqdn, and hostname.
|
|
||||||
This last point is problematic because it will run a playbook
|
|
||||||
3 times if the playbook is for 'all' hosts. We do not necessarily
|
|
||||||
want to override 'hosts' in the playbook because they may be generic
|
|
||||||
across the entire infrastructure -- not host specific.
|
|
||||||
|
|
||||||
Finally, this should use the logging module in some manner and
|
|
||||||
not print data to STDOUT.
|
|
||||||
"""
|
|
||||||
|
|
||||||
if os.path.exists("%s/%s" % (options.dest, DEFAULT_PLAYBOOK)):
|
|
||||||
args.insert(0, DEFAULT_PLAYBOOK)
|
|
||||||
os.chdir(options.dest)
|
os.chdir(options.dest)
|
||||||
hostname = socket.getfqdn()
|
cmd = 'ansible-playbook -c local %s' % DEFAULT_PLAYBOOK
|
||||||
stats = callbacks.AggregateStats()
|
rc = _run(cmd)
|
||||||
playbook_cb = callbacks.PlaybookCallbacks()
|
return rc
|
||||||
pbrunner_cb = callbacks.PlaybookRunnerCallbacks(stats)
|
|
||||||
local_host = [pattern, hostname, hostname.split('.')[0]]
|
|
||||||
|
|
||||||
for playbook in args:
|
|
||||||
pb = ansible.playbook.PlayBook(
|
|
||||||
playbook=playbook,
|
|
||||||
host_list=local_host,
|
|
||||||
module_path=options.module_path,
|
|
||||||
debug=options.debug,
|
|
||||||
runner_callbacks=pbrunner_cb,
|
|
||||||
callbacks=playbook_cb,
|
|
||||||
transport=options.transport,
|
|
||||||
stats=stats
|
|
||||||
)
|
|
||||||
"""
|
|
||||||
This just takes the reporting from ansible-playbook.
|
|
||||||
Ideally, this should use logging to report success/failure/changes.
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
pb.run()
|
|
||||||
hosts = sorted(pb.stats.processed.keys())
|
|
||||||
print "RECAP\n\n"
|
|
||||||
for h in hosts:
|
|
||||||
t = pb.stats.summarize(h)
|
|
||||||
print "%-30s : ok=%4s changed=%4s unreachable=%4s failed=%4s " % (h,
|
|
||||||
t['ok'], t['changed'], t['unreachable'], t['failures']
|
|
||||||
)
|
|
||||||
print "\n"
|
|
||||||
except errors.AnsibleError, e:
|
|
||||||
print >>sys.stderr, "ERROR: %s" % e
|
|
||||||
return 1
|
|
||||||
|
|
||||||
return 0
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
try:
|
try:
|
||||||
sys.exit(main(sys.argv[1:]))
|
sys.exit(main(sys.argv[1:]))
|
||||||
except errors.AnsibleError, e:
|
except KeyboardInterrupt, e:
|
||||||
print >>sys.stderr, "ERROR: %s" % e
|
print >>sys.stderr, "Exit on user request.\n"
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
Loading…
Reference in a new issue