def read_playbook(*args):
args_list = list(args)
cli = PlaybookCLI(args_list)
parser = cli.base_parser(
connect_opts=True,
meta_opts=True,
runas_opts=True,
subset_opts=True,
check_opts=True,
inventory_opts=True,
runtask_opts=True,
vault_opts=True,
fork_opts=True,
module_opts=True,
)
cli.options, cli.args = parser.parse_args(args_list)
cli.normalize_become_options()
cli.options.connection = 'smart'
cli.options.inventory = 'inventory'
loader, inventory, variable_manager = cli._play_prereqs(cli.options)
pb = Playbook.load(cli.args[0], variable_manager, loader)
for play in pb.get_plays():
yield variable_manager.get_vars(play)
def read_playbook(playbook):
cli_args = [os.path.realpath(playbook), testinfra_hosts]
cli = PlaybookCLI(cli_args)
cli.parse()
loader, inventory, variable_manager = cli._play_prereqs()
pb = Playbook.load(cli.args[0], variable_manager, loader)
for play in pb.get_plays():
yield variable_manager.get_vars(play)
def runPlaybooks(playbooks,
_inventory,
params=None,
args=None,
vault_secrets=None):
inventoryArgs = ["-i", _inventory] if _inventory else []
args = ["ansible-playbook"] + inventoryArgs + (args or []) + playbooks
logger.info("running " + " ".join(args))
cli = PlaybookCLI(args)
context._init_global_context = _init_global_context
cli.parse()
# replace C.DEFAULT_STDOUT_CALLBACK with our own so we have control over logging
# config/base.yml sets C.DEFAULT_STDOUT_CALLBACK == 'default' (ansible/plugins/callback/default.py)
# (cli/console.py and cli/adhoc.py sets it to 'minimal' but PlaybookCLI.run() in cli/playbook.py uses the default)
# see also https://github.com/projectatomic/atomic-host-tests/blob/master/callback_plugins/default.py
resultsCB = ResultCallback()
resultsCB.set_options()
C.DEFAULT_STDOUT_CALLBACK = resultsCB
_play_prereqs = cli._play_prereqs
def hook_play_prereqs():
loader, inventory, variable_manager = _play_prereqs()
if vault_secrets:
loader.set_vault_secrets(vault_secrets)
if params:
variable_manager._extra_vars.update(params)
resultsCB.inventoryManager = inventory
resultsCB.variableManager = variable_manager
return loader, inventory, variable_manager
cli._play_prereqs = hook_play_prereqs
oldVerbosity = display.verbosity
if logging.getLogger("unfurl.ansible").getEffectiveLevel() <= 10: # debug
display.verbosity = 2
try:
resultsCB.exit_code = cli.run()
finally:
display.verbosity = oldVerbosity
return resultsCB
def run_playbook(task, inventory='hosts_dev', play_book='fetch_config.yml'):
try:
cli = PlaybookCLI([" ", '-i', INVENTORY_PATH + inventory, play_book])
super(PlaybookCLI, cli).run()
loader, inventory, variable_manager = cli._play_prereqs()
CLI.get_host_list(inventory, context.CLIARGS['subset'])
pbex = PlaybookExecutor(playbooks=context.CLIARGS['args'],
inventory=inventory,
variable_manager=variable_manager,
loader=loader,
passwords=None)
pbex._tqm._stdout_callback = ResultCallback(task)
pbex.run()
return True
except Exception as e:
logger.error(e)
return False
class General(object):
def __init__(self):
self.cli = PlaybookCLI(sys.argv, callback=CallbackModule)
self.cli.parse()
def run(self):
sshpass = None
becomepass = None
passwords = {}
if not self.cli.options.listhosts and not self.cli.options.listtasks and not self.cli.options.listtags and not self.cli.options.syntax:
self.cli.normalize_become_options()
(sshpass, becomepass) = self.cli.ask_passwords()
passwords = {'conn_pass': sshpass, 'become_pass': becomepass}
loader, inventory, variable_manager = self.cli._play_prereqs(
self.cli.options)
for playbook_path in self.cli.args:
pb = Playbook.load(playbook_path,
variable_manager=variable_manager,
loader=loader)
plays = pb.get_plays()
for play in plays:
tqm = None
try:
tqm = TaskQueueManager(
inventory=inventory,
variable_manager=variable_manager,
loader=loader,
options=self.cli.options,
passwords=passwords,
stdout_callback=CallbackModule(),
)
tqm.run(play)
finally:
if tqm is not None:
tqm.cleanup()
playbook_collection = get_collection_name_from_path(b_playbook_dirs[0])
if playbook_collection:
display.warning(
"running playbook inside collection {0}".format(playbook_collection))
AnsibleCollectionLoader().set_default_collection(playbook_collection)
# don't deal with privilege escalation or passwords when we don't need to
if not (context.CLIARGS['listhosts'] or context.CLIARGS['listtasks']
or context.CLIARGS['listtags'] or context.CLIARGS['syntax']):
(sshpass, becomepass) = pb_cli.ask_passwords()
passwords = {'conn_pass': sshpass, 'become_pass': becomepass}
# create base objects
loader, inventory, variable_manager = pb_cli._play_prereqs()
# Fix this when we rewrite inventory by making localhost a real host (and thus show up in list_hosts())
hosts = CLI.get_host_list(inventory, context.CLIARGS['subset'])
# flush fact cache if requplaybook_pathested
if context.CLIARGS['flush_cache']:
pb_cli._flush_cache(inventory, variable_manager)
######################################################################
# Execute the playbook file
pbex = PlaybookExecutor(
playbooks=context.CLIARGS['args'],
inventory=inventory,
variable_manager=variable_manager,