8d97c8c222
Previous version initialized the `TaskQueueManager` after calling `Play.load()` while advertising a way to inject a custom library location path. This caused the tasks loader not to find any custom modules because it was triggered before the path was actually added to the module loader. This patch changes the order of the operations to ensure that the customized `context.CLIARGS` actually influences things. Resolves https://github.com/ansible/ansible/issues/69758.
104 lines
3.6 KiB
Python
Executable file
104 lines
3.6 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
from ansible.executor.task_queue_manager import TaskQueueManager
|
|
from ansible.inventory.manager import InventoryManager
|
|
from ansible.parsing.dataloader import DataLoader
|
|
from ansible.playbook.play import Play
|
|
from ansible.plugins.callback import CallbackBase
|
|
from ansible.vars.manager import VariableManager
|
|
from ansible import context
|
|
from ansible.module_utils.common.collections import ImmutableDict
|
|
|
|
|
|
# Create a callback object so we can capture the output
|
|
class ResultsCollector(CallbackBase):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(ResultsCollector, self).__init__(*args, **kwargs)
|
|
self.host_ok = {}
|
|
self.host_unreachable = {}
|
|
self.host_failed = {}
|
|
|
|
def v2_runner_on_unreachable(self, result):
|
|
self.host_unreachable[result._host.get_name()] = result
|
|
|
|
def v2_runner_on_ok(self, result, *args, **kwargs):
|
|
self.host_ok[result._host.get_name()] = result
|
|
|
|
def v2_runner_on_failed(self, result, *args, **kwargs):
|
|
self.host_failed[result._host.get_name()] = result
|
|
|
|
|
|
def main():
|
|
host_list = ['localhost', 'www.example.com', 'www.google.com']
|
|
# since the API is constructed for CLI it expects certain options to always be set in the context object
|
|
context.CLIARGS = ImmutableDict(connection='smart', module_path=['/usr/share/ansible'], forks=10, become=None,
|
|
become_method=None, become_user=None, check=False, diff=False)
|
|
|
|
# required for
|
|
# https://github.com/ansible/ansible/blob/devel/lib/ansible/inventory/manager.py#L204
|
|
sources = ','.join(host_list)
|
|
if len(host_list) == 1:
|
|
sources += ','
|
|
|
|
# initialize needed objects
|
|
loader = DataLoader()
|
|
passwords = dict()
|
|
|
|
# Instantiate our ResultsCollector for handling results as
|
|
# they come in. Ansible expects this to be one of its main
|
|
# display outlets.
|
|
callback = ResultsCollector()
|
|
|
|
# create inventory and pass to var manager
|
|
inventory = InventoryManager(loader=loader, sources=sources)
|
|
variable_manager = VariableManager(loader=loader, inventory=inventory)
|
|
|
|
# Instantiate task queue manager, which takes care of forking
|
|
# and setting up all objects to iterate over host list and tasks.
|
|
# IMPORTANT: This also adds library dirs paths to the module loader
|
|
# IMPORTANT: and so it must be initialized before calling `Play.load()`.
|
|
tqm = TaskQueueManager(
|
|
inventory=inventory,
|
|
variable_manager=variable_manager,
|
|
loader=loader,
|
|
passwords=passwords,
|
|
stdout_callback=callback,
|
|
)
|
|
|
|
# create play with tasks
|
|
play_source = dict(
|
|
name="Ansible Play",
|
|
hosts=host_list,
|
|
gather_facts='no',
|
|
tasks=[dict(action=dict(module='command', args=dict(cmd='/usr/bin/uptime')))]
|
|
)
|
|
play = Play().load(play_source, variable_manager=variable_manager, loader=loader)
|
|
|
|
# actually run it
|
|
try:
|
|
result = tqm.run(play)
|
|
finally:
|
|
if tqm is not None:
|
|
tqm.cleanup()
|
|
if loader:
|
|
loader.cleanup_all_tmp_files()
|
|
|
|
print("UP ***********")
|
|
for host, result in callback.host_ok.items():
|
|
print('{0} >>> {1}'.format(host, result._result['stdout']))
|
|
|
|
print("FAILED *******")
|
|
for host, result in callback.host_failed.items():
|
|
print('{0} >>> {1}'.format(host, result._result['msg']))
|
|
|
|
print("DOWN *********")
|
|
for host, result in callback.host_unreachable.items():
|
|
print('{0} >>> {1}'.format(host, result._result['msg']))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|