remove no used ansible api with != 2.6.2
This commit is contained in:
pengtao
parent
8e0fb41973
commit
3715268329
@ -1,194 +0,0 @@
|
||||
import sys
|
||||
import os
|
||||
import json
|
||||
import shutil
|
||||
from ansible.module_utils.common.collections import ImmutableDict
|
||||
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
|
||||
from ansible import context
|
||||
import ansible.constants as C
|
||||
|
||||
|
||||
class ResultCallback(CallbackBase):
|
||||
"""
|
||||
重写callbackBase类的部分方法
|
||||
"""
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
super().__init__(*args, **kwargs)
|
||||
self.host_ok = {}
|
||||
self.host_unreachable = {}
|
||||
self.host_failed = {}
|
||||
self.task_ok = {}
|
||||
|
||||
def v2_runner_on_unreachable(self, result):
|
||||
self.host_unreachable[result._host.get_name()] = result
|
||||
|
||||
def v2_runner_on_ok(self, result, **kwargs):
|
||||
self.host_ok[result._host.get_name()] = result
|
||||
|
||||
def v2_runner_on_failed(self, result, **kwargs):
|
||||
self.host_failed[result._host.get_name()] = result
|
||||
|
||||
|
||||
class MyAnsiable2():
|
||||
def __init__(self,
|
||||
connection='local', # 连接方式 local 本地方式，smart ssh方式
|
||||
remote_user=None, # ssh 用户
|
||||
remote_password=None, # ssh 用户的密码，应该是一个字典, key 必须是 conn_pass
|
||||
private_key_file=None, # 指定自定义的私钥地址
|
||||
sudo=None, sudo_user=None, ask_sudo_pass=None,
|
||||
module_path=None, # 模块路径，可以指定一个自定义模块的路径
|
||||
become=None, # 是否提权
|
||||
become_method=None, # 提权方式 默认 sudo 可以是 su
|
||||
become_user=None, # 提权后，要成为的用户，并非登录用户
|
||||
check=False, diff=False,
|
||||
listhosts=None, listtasks=None, listtags=None,
|
||||
verbosity=3,
|
||||
syntax=None,
|
||||
start_at_task=None,
|
||||
inventory=None):
|
||||
|
||||
# 函数文档注释
|
||||
"""
|
||||
初始化函数，定义的默认的选项值，
|
||||
在初始化的时候可以传参，以便覆盖默认选项的值
|
||||
"""
|
||||
context.CLIARGS = ImmutableDict(
|
||||
connection=connection,
|
||||
remote_user=remote_user,
|
||||
private_key_file=private_key_file,
|
||||
sudo=sudo,
|
||||
sudo_user=sudo_user,
|
||||
ask_sudo_pass=ask_sudo_pass,
|
||||
module_path=module_path,
|
||||
become=become,
|
||||
become_method=become_method,
|
||||
become_user=become_user,
|
||||
verbosity=verbosity,
|
||||
listhosts=listhosts,
|
||||
listtasks=listtasks,
|
||||
listtags=listtags,
|
||||
syntax=syntax,
|
||||
start_at_task=start_at_task,
|
||||
)
|
||||
|
||||
# 三元表达式，假如没有传递 inventory, 就使用 "localhost,"
|
||||
# 指定 inventory 文件
|
||||
# inventory 的值可以是一个 资产清单文件
|
||||
# 也可以是一个包含主机的元组，这个仅仅适用于测试
|
||||
# 比如 ： 1.1.1.1, # 如果只有一个 IP 最后必须有英文的逗号
|
||||
# 或者： 1.1.1.1, 2.2.2.2
|
||||
|
||||
self.inventory = inventory if inventory else "localhost,"
|
||||
|
||||
# 实例化数据解析器
|
||||
self.loader = DataLoader()
|
||||
|
||||
# 实例化 资产配置对象
|
||||
self.inv_obj = InventoryManager(
|
||||
loader=self.loader, sources=self.inventory)
|
||||
|
||||
# 设置密码
|
||||
self.passwords = remote_password
|
||||
|
||||
# 实例化回调插件对象
|
||||
self.results_callback = ResultCallback()
|
||||
|
||||
# 变量管理器
|
||||
self.variable_manager = VariableManager(self.loader, self.inv_obj)
|
||||
|
||||
def run(self, hosts='localhost', gether_facts="no", module="ping", args='', task_time=0):
|
||||
"""
|
||||
参数说明：
|
||||
task_time -- 执行异步任务时等待的秒数，这个需要大于 0 ，等于 0 的时候不支持异步（默认值）。这个值应该等于执行任务实际耗时时间为好
|
||||
"""
|
||||
play_source = dict(
|
||||
name="Ad-hoc",
|
||||
hosts=hosts,
|
||||
gather_facts=gether_facts,
|
||||
tasks=[
|
||||
# 这里每个 task 就是这个列表中的一个元素，格式是嵌套的字典
|
||||
# 也可以作为参数传递过来，这里就简单化了。
|
||||
{"action": {"module": module, "args": args}, "async": task_time, "poll": 0}])
|
||||
|
||||
play = Play().load(play_source, variable_manager=self.variable_manager, loader=self.loader)
|
||||
|
||||
tqm = None
|
||||
try:
|
||||
tqm = TaskQueueManager(
|
||||
inventory=self.inv_obj,
|
||||
variable_manager=self.variable_manager,
|
||||
loader=self.loader,
|
||||
passwords=self.passwords,
|
||||
stdout_callback=self.results_callback)
|
||||
|
||||
result = tqm.run(play)
|
||||
|
||||
finally:
|
||||
if tqm is not None:
|
||||
tqm.cleanup()
|
||||
shutil.rmtree(C.DEFAULT_LOCAL_TMP, True)
|
||||
|
||||
def playbook(self, playbooks, run_data):
|
||||
"""
|
||||
Keyword arguments:
|
||||
playbooks -- 需要是一个列表类型
|
||||
"""
|
||||
from ansible.executor.playbook_executor import PlaybookExecutor
|
||||
self.variable_manager.extra_vars = run_data
|
||||
playbook = PlaybookExecutor(playbooks=playbooks,
|
||||
inventory=self.inv_obj,
|
||||
variable_manager=self.variable_manager,
|
||||
loader=self.loader,
|
||||
passwords=self.passwords)
|
||||
|
||||
# 使用回调函数
|
||||
playbook._tqm._stdout_callback = self.results_callback
|
||||
|
||||
result = playbook.run()
|
||||
|
||||
def get_result(self):
|
||||
result_raw = {'success': {}, 'failed': {}, 'unreachable': {}}
|
||||
|
||||
# print(self.results_callback.host_ok)
|
||||
for host, result in self.results_callback.host_ok.items():
|
||||
result_raw['success'][host] = result._result
|
||||
for host, result in self.results_callback.host_failed.items():
|
||||
result_raw['failed'][host] = result._result
|
||||
for host, result in self.results_callback.host_unreachable.items():
|
||||
result_raw['unreachable'][host] = result._result
|
||||
|
||||
# 最终打印结果，并且使用 JSON 继续格式化
|
||||
print(json.dumps(result_raw, indent=4))
|
||||
|
||||
return json.dumps(result_raw)
|
||||
|
||||
|
||||
def test_run_cmd():
|
||||
hosts = "192.168.100.42"
|
||||
ansible3 = MyAnsiable2(
|
||||
inventory='hosts', connection='smart') # 创建资源库对象
|
||||
ansible3.run(hosts=hosts, module="shell",
|
||||
args=' ps -ef |grep python')
|
||||
stdout_dict = json.loads(ansible3.get_result())
|
||||
print(stdout_dict, type(stdout_dict))
|
||||
print(stdout_dict['success'][hosts]['stdout'], '######wc')
|
||||
|
||||
|
||||
def test_run_playbook():
|
||||
ansible3 = MyAnsiable2(
|
||||
inventory='hosts', connection='smart')
|
||||
run_data = {
|
||||
"dest_paths": "/data/apps/temp",
|
||||
"source": "/root/temp/host",
|
||||
"dest_filename": "/data/apps/temp/host.new"
|
||||
}
|
||||
ansible3.playbook(playbooks=['test.yml'])
|
||||
stdout_dict = json.loads(ansible3.get_result())
|
||||
print(stdout_dict, type(stdout_dict))
|
||||
# print(stdout_dict['success']['192.168.0.94']['stdout'])
|
||||
@ -1,137 +0,0 @@
|
||||
#!/usr/bin/env python
|
||||
# pip3 install ansible
|
||||
# pip3 install ansible_runner
|
||||
# pip3 install ansible_inventory
|
||||
# pip3 install ansible_playbook
|
||||
# Ansible will require Python 3.8 or newer on the controller starting with Ansible 2.12
|
||||
from __future__ import (absolute_import, division, print_function)
|
||||
__metaclass__ = type
|
||||
|
||||
import json
|
||||
import shutil
|
||||
|
||||
import ansible.constants as C
|
||||
from ansible.executor.task_queue_manager import TaskQueueManager
|
||||
from ansible.module_utils.common.collections import ImmutableDict
|
||||
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
|
||||
|
||||
|
||||
# Create a callback plugin so we can capture the output
|
||||
class ResultsCollectorJSONCallback(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 __init__(self, *args, **kwargs):
|
||||
super(ResultsCollectorJSONCallback, self).__init__(*args, **kwargs)
|
||||
self.host_ok = {}
|
||||
self.host_unreachable = {}
|
||||
self.host_failed = {}
|
||||
|
||||
def v2_runner_on_unreachable(self, result):
|
||||
host = result._host
|
||||
self.host_unreachable[host.get_name()] = result
|
||||
|
||||
def v2_runner_on_ok(self, result, *args, **kwargs):
|
||||
"""Print a json representation of the result.
|
||||
|
||||
Also, store the result in an instance attribute for retrieval later
|
||||
"""
|
||||
host = result._host
|
||||
self.host_ok[host.get_name()] = result
|
||||
print(json.dumps({host.name: result._result}, indent=4))
|
||||
|
||||
def v2_runner_on_failed(self, result, *args, **kwargs):
|
||||
host = result._host
|
||||
self.host_failed[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=['/to/mymodules', '/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() # Takes care of finding and reading yaml, json and ini files
|
||||
passwords = dict(vault_pass='secret')
|
||||
|
||||
# Instantiate our ResultsCollectorJSONCallback for handling results as they come in. Ansible expects this to be one of its main display outlets
|
||||
results_callback = ResultsCollectorJSONCallback()
|
||||
|
||||
# create inventory, use path to host config file as source or hosts in a comma separated string
|
||||
inventory = InventoryManager(loader=loader, sources=sources)
|
||||
|
||||
# variable manager takes care of merging all the different sources to give you a unified view of variables available in each context
|
||||
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,
|
||||
# Use our custom callback instead of the ``default`` callback plugin, which prints to stdout
|
||||
stdout_callback=results_callback,
|
||||
)
|
||||
|
||||
# create data structure that represents our play, including tasks, this is basically what our YAML loader does internally.
|
||||
play_source = dict(
|
||||
name="Ansible Play",
|
||||
hosts=host_list,
|
||||
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}}'))),
|
||||
dict(action=dict(module='command', args=dict(cmd='/usr/bin/uptime'))),
|
||||
]
|
||||
)
|
||||
|
||||
# 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)
|
||||
|
||||
# Actually run it
|
||||
try:
|
||||
# most interesting data for a play is actually sent to the callback's methods
|
||||
result = tqm.run(play)
|
||||
finally:
|
||||
# we always need to cleanup child procs and the structures we use to communicate with them
|
||||
tqm.cleanup()
|
||||
if loader:
|
||||
loader.cleanup_all_tmp_files()
|
||||
|
||||
# Remove ansible tmpdir
|
||||
shutil.rmtree(C.DEFAULT_LOCAL_TMP, True)
|
||||
|
||||
print("UP ***********")
|
||||
for host, result in results_callback.host_ok.items():
|
||||
print('{0} >>> {1}'.format(host, result._result['stdout']))
|
||||
|
||||
print("FAILED *******")
|
||||
for host, result in results_callback.host_failed.items():
|
||||
print('{0} >>> {1}'.format(host, result._result['msg']))
|
||||
|
||||
print("DOWN *********")
|
||||
for host, result in results_callback.host_unreachable.items():
|
||||
print('{0} >>> {1}'.format(host, result._result['msg']))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
@ -1,198 +0,0 @@
|
||||
from datetime import datetime
|
||||
from ansible.plugins.callback import CallbackBase
|
||||
|
||||
# A custom object to store to the database
|
||||
from some_project.storage import Logs
|
||||
|
||||
|
||||
class PlayLogger:
|
||||
"""Store log output in a single object.
|
||||
We create a new object per Ansible run
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.log = ''
|
||||
self.runtime = 0
|
||||
|
||||
def append(self, log_line):
|
||||
"""append to log"""
|
||||
self.log += log_line+"\n\n"
|
||||
|
||||
def banner(self, msg):
|
||||
"""Output Trailing Stars"""
|
||||
width = 78 - len(msg)
|
||||
if width < 3:
|
||||
width = 3
|
||||
filler = "*" * width
|
||||
return "\n%s %s " % (msg, filler)
|
||||
|
||||
|
||||
class CallbackModule(CallbackBase):
|
||||
"""
|
||||
Reference: https://github.com/ansible/ansible/blob/v2.0.0.2-1/lib/ansible/plugins/callback/default.py
|
||||
"""
|
||||
|
||||
CALLBACK_VERSION = 2.0
|
||||
CALLBACK_TYPE = 'stored'
|
||||
CALLBACK_NAME = 'database'
|
||||
|
||||
def __init__(self):
|
||||
super(CallbackModule, self).__init__()
|
||||
self.logger = PlayLogger()
|
||||
self.start_time = datetime.now()
|
||||
|
||||
def v2_runner_on_failed(self, result, ignore_errors=False):
|
||||
delegated_vars = result._result.get('_ansible_delegated_vars', None)
|
||||
|
||||
# Catch an exception
|
||||
# This may never be called because default handler deletes
|
||||
# the exception, since Ansible thinks it knows better
|
||||
if 'exception' in result._result:
|
||||
# Extract the error message and log it
|
||||
error = result._result['exception'].strip().split('\n')[-1]
|
||||
self.logger.append(error)
|
||||
|
||||
# Remove the exception from the result so it's not shown every time
|
||||
del result._result['exception']
|
||||
|
||||
# Else log the reason for the failure
|
||||
if result._task.loop and 'results' in result._result:
|
||||
# item_on_failed, item_on_skipped, item_on_ok
|
||||
self._process_items(result)
|
||||
else:
|
||||
if delegated_vars:
|
||||
self.logger.append("fatal: [%s -> %s]: FAILED! => %s" % (result._host.get_name(
|
||||
), delegated_vars['ansible_host'], self._dump_results(result._result)))
|
||||
else:
|
||||
self.logger.append("fatal: [%s]: FAILED! => %s" % (
|
||||
result._host.get_name(), self._dump_results(result._result)))
|
||||
|
||||
def v2_runner_on_ok(self, result):
|
||||
self._clean_results(result._result, result._task.action)
|
||||
delegated_vars = result._result.get('_ansible_delegated_vars', None)
|
||||
if result._task.action == 'include':
|
||||
return
|
||||
elif result._result.get('changed', False):
|
||||
if delegated_vars:
|
||||
msg = "changed: [%s -> %s]" % (result._host.get_name(),
|
||||
delegated_vars['ansible_host'])
|
||||
else:
|
||||
msg = "changed: [%s]" % result._host.get_name()
|
||||
else:
|
||||
if delegated_vars:
|
||||
msg = "ok: [%s -> %s]" % (result._host.get_name(),
|
||||
delegated_vars['ansible_host'])
|
||||
else:
|
||||
msg = "ok: [%s]" % result._host.get_name()
|
||||
|
||||
if result._task.loop and 'results' in result._result:
|
||||
# item_on_failed, item_on_skipped, item_on_ok
|
||||
self._process_items(result)
|
||||
else:
|
||||
self.logger.append(msg)
|
||||
|
||||
def v2_runner_on_skipped(self, result):
|
||||
if result._task.loop and 'results' in result._result:
|
||||
# item_on_failed, item_on_skipped, item_on_ok
|
||||
self._process_items(result)
|
||||
else:
|
||||
msg = "skipping: [%s]" % result._host.get_name()
|
||||
self.logger.append(msg)
|
||||
|
||||
def v2_runner_on_unreachable(self, result):
|
||||
delegated_vars = result._result.get('_ansible_delegated_vars', None)
|
||||
if delegated_vars:
|
||||
self.logger.append("fatal: [%s -> %s]: UNREACHABLE! => %s" % (result._host.get_name(
|
||||
), delegated_vars['ansible_host'], self._dump_results(result._result)))
|
||||
else:
|
||||
self.logger.append("fatal: [%s]: UNREACHABLE! => %s" % (
|
||||
result._host.get_name(), self._dump_results(result._result)))
|
||||
|
||||
def v2_runner_on_no_hosts(self, task):
|
||||
self.logger.append("skipping: no hosts matched")
|
||||
|
||||
def v2_playbook_on_task_start(self, task, is_conditional):
|
||||
self.logger.append("TASK [%s]" % task.get_name().strip())
|
||||
|
||||
def v2_playbook_on_play_start(self, play):
|
||||
name = play.get_name().strip()
|
||||
if not name:
|
||||
msg = "PLAY"
|
||||
else:
|
||||
msg = "PLAY [%s]" % name
|
||||
|
||||
self.logger.append(msg)
|
||||
|
||||
def v2_playbook_item_on_ok(self, result):
|
||||
delegated_vars = result._result.get('_ansible_delegated_vars', None)
|
||||
if result._task.action == 'include':
|
||||
return
|
||||
elif result._result.get('changed', False):
|
||||
if delegated_vars:
|
||||
msg = "changed: [%s -> %s]" % (result._host.get_name(),
|
||||
delegated_vars['ansible_host'])
|
||||
else:
|
||||
msg = "changed: [%s]" % result._host.get_name()
|
||||
else:
|
||||
if delegated_vars:
|
||||
msg = "ok: [%s -> %s]" % (result._host.get_name(),
|
||||
delegated_vars['ansible_host'])
|
||||
else:
|
||||
msg = "ok: [%s]" % result._host.get_name()
|
||||
|
||||
msg += " => (item=%s)" % (result._result['item'])
|
||||
|
||||
self.logger.append(msg)
|
||||
|
||||
def v2_playbook_item_on_failed(self, result):
|
||||
delegated_vars = result._result.get('_ansible_delegated_vars', None)
|
||||
if 'exception' in result._result:
|
||||
# Extract the error message and log it
|
||||
error = result._result['exception'].strip().split('\n')[-1]
|
||||
self.logger.append(error)
|
||||
|
||||
# Remove the exception from the result so it's not shown every time
|
||||
del result._result['exception']
|
||||
|
||||
if delegated_vars:
|
||||
self.logger.append("failed: [%s -> %s] => (item=%s) => %s" % (result._host.get_name(
|
||||
), delegated_vars['ansible_host'], result._result['item'], self._dump_results(result._result)))
|
||||
else:
|
||||
self.logger.append("failed: [%s] => (item=%s) => %s" % (result._host.get_name(
|
||||
), result._result['item'], self._dump_results(result._result)))
|
||||
|
||||
def v2_playbook_item_on_skipped(self, result):
|
||||
msg = "skipping: [%s] => (item=%s) " % (
|
||||
result._host.get_name(), result._result['item'])
|
||||
self.logger.append(msg)
|
||||
|
||||
def v2_playbook_on_stats(self, stats):
|
||||
run_time = datetime.now() - self.start_time
|
||||
# returns an int, unlike run_time.total_seconds()
|
||||
self.logger.runtime = run_time.seconds
|
||||
|
||||
hosts = sorted(stats.processed.keys())
|
||||
for h in hosts:
|
||||
t = stats.summarize(h)
|
||||
|
||||
msg = "PLAY RECAP [%s] : %s %s %s %s %s" % (
|
||||
h,
|
||||
"ok: %s" % (t['ok']),
|
||||
"changed: %s" % (t['changed']),
|
||||
"unreachable: %s" % (t['unreachable']),
|
||||
"skipped: %s" % (t['skipped']),
|
||||
"failed: %s" % (t['failures']),
|
||||
)
|
||||
|
||||
self.logger.append(msg)
|
||||
|
||||
def record_logs(self, user_id, success=False):
|
||||
"""
|
||||
Special callback added to this callback plugin
|
||||
Called by Runner objet
|
||||
:param user_id:
|
||||
:return:
|
||||
"""
|
||||
|
||||
log_storage = Logs()
|
||||
return log_storage.save_log(user_id, self.logger.log, self.logger.runtime, success)
|
||||
@ -1,23 +0,0 @@
|
||||
from .task import Runner
|
||||
# You may want this to run as user root instead
|
||||
# or make this an environmental variable, or
|
||||
# a CLI prompt. Whatever you want!
|
||||
become_user_password = 'foo-whatever'
|
||||
run_data = {
|
||||
"user_id": 12345,
|
||||
'foo': 'bar',
|
||||
'baz': 'cux-or-whatever-this-one-is'
|
||||
}
|
||||
|
||||
|
||||
runner = Runner(hostnames='192.168.10.233'
|
||||
playbook='run.yaml',
|
||||
private_key='/home/user/.ssh/id_whatever',
|
||||
run_data=run_data,
|
||||
become_pass=become_user_password,
|
||||
verbosity=0)
|
||||
|
||||
|
||||
stats = runner.run()
|
||||
|
||||
# Maybe do something with stats here? If you want!
|
||||
@ -1,159 +0,0 @@
|
||||
import os
|
||||
from tempfile import NamedTemporaryFile
|
||||
from ansible.inventory import Inventory
|
||||
from ansible.vars import VariableManager
|
||||
from ansible.parsing.dataloader import DataLoader
|
||||
from ansible.executor import playbook_executor
|
||||
from ansible.utils.display import Display
|
||||
|
||||
|
||||
class Options(object):
|
||||
"""
|
||||
Options class to replace Ansible OptParser
|
||||
"""
|
||||
|
||||
def __init__(self, verbosity=None, inventory=None, listhosts=None, subset=None, module_paths=None, extra_vars=None,
|
||||
forks=None, ask_vault_pass=None, vault_password_files=None, new_vault_password_file=None,
|
||||
output_file=None, tags=None, skip_tags=None, one_line=None, tree=None, ask_sudo_pass=None, ask_su_pass=None,
|
||||
sudo=None, sudo_user=None, become=None, become_method=None, become_user=None, become_ask_pass=None,
|
||||
ask_pass=None, private_key_file=None, remote_user=None, connection=None, timeout=None, ssh_common_args=None,
|
||||
sftp_extra_args=None, scp_extra_args=None, ssh_extra_args=None, poll_interval=None, seconds=None, check=None,
|
||||
syntax=None, diff=None, force_handlers=None, flush_cache=None, listtasks=None, listtags=None, module_path=None):
|
||||
self.verbosity = verbosity
|
||||
self.inventory = inventory
|
||||
self.listhosts = listhosts
|
||||
self.subset = subset
|
||||
self.module_paths = module_paths
|
||||
self.extra_vars = extra_vars
|
||||
self.forks = forks
|
||||
self.ask_vault_pass = ask_vault_pass
|
||||
self.vault_password_files = vault_password_files
|
||||
self.new_vault_password_file = new_vault_password_file
|
||||
self.output_file = output_file
|
||||
self.tags = tags
|
||||
self.skip_tags = skip_tags
|
||||
self.one_line = one_line
|
||||
self.tree = tree
|
||||
self.ask_sudo_pass = ask_sudo_pass
|
||||
self.ask_su_pass = ask_su_pass
|
||||
self.sudo = sudo
|
||||
self.sudo_user = sudo_user
|
||||
self.become = become
|
||||
self.become_method = become_method
|
||||
self.become_user = become_user
|
||||
self.become_ask_pass = become_ask_pass
|
||||
self.ask_pass = ask_pass
|
||||
self.private_key_file = private_key_file
|
||||
self.remote_user = remote_user
|
||||
self.connection = connection
|
||||
self.timeout = timeout
|
||||
self.ssh_common_args = ssh_common_args
|
||||
self.sftp_extra_args = sftp_extra_args
|
||||
self.scp_extra_args = scp_extra_args
|
||||
self.ssh_extra_args = ssh_extra_args
|
||||
self.poll_interval = poll_interval
|
||||
self.seconds = seconds
|
||||
self.check = check
|
||||
self.syntax = syntax
|
||||
self.diff = diff
|
||||
self.force_handlers = force_handlers
|
||||
self.flush_cache = flush_cache
|
||||
self.listtasks = listtasks
|
||||
self.listtags = listtags
|
||||
self.module_path = module_path
|
||||
|
||||
|
||||
class Runner(object):
|
||||
|
||||
def __init__(self, hostnames, playbook, private_key_file, run_data, become_pass, verbosity=0):
|
||||
|
||||
self.run_data = run_data
|
||||
|
||||
self.options = Options()
|
||||
self.options.private_key_file = private_key_file
|
||||
self.options.verbosity = verbosity
|
||||
self.options.connection = 'ssh' # Need a connection type "smart" or "ssh"
|
||||
self.options.become = True
|
||||
self.options.become_method = 'sudo'
|
||||
self.options.become_user = 'root'
|
||||
|
||||
# Set global verbosity
|
||||
self.display = Display()
|
||||
self.display.verbosity = self.options.verbosity
|
||||
# Executor appears to have it's own
|
||||
# verbosity object/setting as well
|
||||
playbook_executor.verbosity = self.options.verbosity
|
||||
|
||||
# Become Pass Needed if not logging in as user root
|
||||
passwords = {'become_pass': become_pass}
|
||||
|
||||
# Gets data from YAML/JSON files
|
||||
self.loader = DataLoader()
|
||||
self.loader.set_vault_password(os.environ['VAULT_PASS'])
|
||||
|
||||
# All the variables from all the various places
|
||||
self.variable_manager = VariableManager()
|
||||
self.variable_manager.extra_vars = self.run_data
|
||||
|
||||
# Parse hosts, I haven't found a good way to
|
||||
# pass hosts in without using a parsed template :(
|
||||
# (Maybe you know how?)
|
||||
self.hosts = NamedTemporaryFile(delete=False)
|
||||
self.hosts.write("""[run_hosts]
|
||||
%s
|
||||
""" % hostnames)
|
||||
self.hosts.close()
|
||||
|
||||
# This was my attempt to pass in hosts directly.
|
||||
#
|
||||
# Also Note: In py2.7, "isinstance(foo, str)" is valid for
|
||||
# latin chars only. Luckily, hostnames are
|
||||
# ascii-only, which overlaps latin charset
|
||||
## if isinstance(hostnames, str):
|
||||
## hostnames = {"customers": {"hosts": [hostnames]}}
|
||||
|
||||
# Set inventory, using most of above objects
|
||||
self.inventory = Inventory(
|
||||
loader=self.loader, variable_manager=self.variable_manager, host_list=self.hosts.name)
|
||||
self.variable_manager.set_inventory(self.inventory)
|
||||
|
||||
# Playbook to run. Assumes it is
|
||||
# local to this python file
|
||||
pb_dir = os.path.dirname(__file__)
|
||||
playbook = "%s/%s" % (pb_dir, playbook)
|
||||
|
||||
# Setup playbook executor, but don't run until run() called
|
||||
self.pbex = playbook_executor.PlaybookExecutor(
|
||||
playbooks=[playbook],
|
||||
inventory=self.inventory,
|
||||
variable_manager=self.variable_manager,
|
||||
loader=self.loader,
|
||||
options=self.options,
|
||||
passwords=passwords)
|
||||
|
||||
def run(self):
|
||||
# Results of PlaybookExecutor
|
||||
self.pbex.run()
|
||||
stats = self.pbex._tqm._stats
|
||||
|
||||
# Test if success for record_logs
|
||||
run_success = True
|
||||
hosts = sorted(stats.processed.keys())
|
||||
for h in hosts:
|
||||
t = stats.summarize(h)
|
||||
if t['unreachable'] > 0 or t['failures'] > 0:
|
||||
run_success = False
|
||||
|
||||
# Dirty hack to send callback to save logs with data we want
|
||||
# Note that function "record_logs" is one I created and put into
|
||||
# the playbook callback file
|
||||
self.pbex._tqm.send_callback(
|
||||
'record_logs',
|
||||
user_id=self.run_data['user_id'],
|
||||
success=run_success
|
||||
)
|
||||
|
||||
# Remove created temporary files
|
||||
os.remove(self.hosts.name)
|
||||
|
||||
return stats
|
||||
Loading…
x
Reference in New Issue
Block a user