Created
May 3, 2019 18:54
-
-
Save tsaridas/0c79d1f3e63a878be30645be08dea94f to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python | |
| import json | |
| import shutil | |
| from collections import namedtuple | |
| from ansible.parsing.dataloader import DataLoader | |
| from ansible.vars.manager import VariableManager | |
| from ansible.inventory.manager import InventoryManager | |
| from ansible.playbook.play import Play | |
| from ansible.executor.task_queue_manager import TaskQueueManager | |
| from ansible.plugins.callback import CallbackBase | |
| import ansible.constants as C | |
| class ResultCallback(CallbackBase): | |
| """A sample callback plugin used for performing an action as results come in | |
| If you want to collect all results into a single object for processing at | |
| the end of the execution, look into utilizing the ``json`` callback plugin | |
| or writing your own custom callback plugin | |
| """ | |
| def v2_runner_on_ok(self, result, **kwargs): | |
| """Print a json representation of the result | |
| This method could store the result in an instance attribute for retrieval later | |
| """ | |
| host = result._host | |
| print(json.dumps({host.name: result._result}, indent=4)) | |
| # since API is constructed for CLI it expects certain options to always be set, named tuple 'fakes' the args parsing options object | |
| Options = namedtuple('Options', ['connection', 'module_path', 'forks', 'become', 'become_method', 'become_user', 'check', 'diff']) | |
| options = Options(connection='local', module_path=['/to/mymodules'], forks=10, become=None, become_method=None, become_user=None, check=False, diff=False) | |
| # initialize needed objects | |
| loader = DataLoader() # Takes care of finding and reading yaml, json and ini files | |
| passwords = dict(vault_pass='secret') | |
| # Instantiate our ResultCallback for handling results as they come in. Ansible expects this to be one of its main display outlets | |
| results_callback = ResultCallback() | |
| # create inventory, use path to host config file as source or hosts in a comma separated string | |
| inventory = InventoryManager(loader=loader, sources='/etc/ansible/inventory') | |
| # variable manager takes care of merging all the different sources to give you a unifed view of variables available in each context | |
| variable_manager = VariableManager(loader=loader, inventory=inventory) | |
| # create datastructure that represents our play, including tasks, this is basically what our YAML loader does internally. | |
| play_source = dict( | |
| name = "Ansible Play", | |
| hosts = 'salt-minion-01', | |
| gather_facts = 'no', | |
| tasks = [ | |
| dict(action=dict(module='shell', args='ls'), register='shell_out'), | |
| dict(action=dict(module='debug', args=dict(msg='{{shell_out.stdout}}'))) | |
| ] | |
| ) | |
| play_source_2 = dict( | |
| name = "Ansible Play", | |
| hosts = 'salt-minion-02', | |
| gather_facts = 'no', | |
| tasks = [ | |
| dict(action=dict(module='shell', args='ls'), register='shell_out'), | |
| dict(action=dict(module='debug', args=dict(msg='{{shell_out.stdout}}'))) | |
| ] | |
| ) | |
| # Create play object, playbook objects use .load instead of init or new methods, | |
| # this will also automatically create the task objects from the info provided in play_source | |
| play = Play().load(play_source, variable_manager=variable_manager, loader=loader) | |
| play2 = Play().load(play_source_2, variable_manager=variable_manager, loader=loader) | |
| # Run it - instantiate task queue manager, which takes care of forking and setting up all objects to iterate over host list and tasks | |
| tqm = None | |
| try: | |
| tqm = TaskQueueManager( | |
| inventory=inventory, | |
| variable_manager=variable_manager, | |
| loader=loader, | |
| options=options, | |
| passwords=passwords, | |
| stdout_callback=results_callback, # Use our custom callback instead of the ``default`` callback plugin, which prints to stdout | |
| ) | |
| result2 = tqm.run(play2) # most interesting data for a play is actually sent to the callback's methods | |
| result = tqm.run(play) # most interesting data for a play is actually sent to the callback's methods | |
| finally: | |
| # we always need to cleanup child procs and the structres we use to communicate with them | |
| if tqm is not None: | |
| tqm.cleanup() | |
| # Remove ansible tmpdir | |
| shutil.rmtree(C.DEFAULT_LOCAL_TMP, True) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment