超酷小 发表于 2018-7-29 08:15:28

ansible2.4 api调用

AnsibleAPI.py 实现多进程发布代码  

  
#!/usr/bin/python
  
# --*-- coding:utf-8 --*--
  

  
import json
  
import logging
  
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.executor.playbook_executor import PlaybookExecutor
  
from ansible.plugins.callback import CallbackBase
  
from collections import namedtuple
  
from ansible import constants as C
  
import ansible.executor.task_result
  
import multiprocessing
  

  

  
class ResultsCollector(CallbackBase):
  def v2_runner_on_ok(self,result):
  host = result._host
  logging.basicConfig(level=logging.DEBUG,
  format='%(asctime)s %(filename)s %(levelname)s %(message)s',
  datefmt='%a, %d %b %Y %H:%M:%S',
  filename='/root/cmdb/script/publish.log',
  filemode='w'
  )
  logging.warning('===v2_runner_on_ok===host=%s===result=%s' % (host, result._result))
  #print(json.dumps({host.name: result._result}, indent=4))
  def v2_runner_on_failed(self,result,ignore_errors=False):
  host = result._host
  logging.warning('===v2_runner_on_failed====host=%s===result=%s' % (host, result._result))
  

  def v2_runner_on_unreachable(self,result):
  host = result._host
  logging.warning('===v2_runner_on_unreachable====host=%s===result=%s' % (host, result._result))
  

  

  

  
class AnsibleAPI(object):
  def __init__(self,hostlist,image_name,playbooks, *args, **kwargs):
  self.playbooks = playbooks
  self.passwords = None
  self.callback = None
  Options = namedtuple('Options',['connection','remote_user','ask_sudo_pass','verbosity','ack_pass','module_path', 'forks', 'become', 'become_method','become_user','check', 'listhosts', 'listtasks', 'listtags','syntax','sudo_user', 'sudo', 'diff'])
  self.options = Options(connection='smart',remote_user='root',ack_pass=None,sudo_user='root',forks=5, sudo='yes', ask_sudo_pass=False, verbosity=5,module_path=None,become=True, become_method='sudo', become_user='root',check=None, listhosts=False,listtasks=False, listtags=None, syntax=None,diff=False)
  self.loader = DataLoader()
  self.inventory = InventoryManager(loader=self.loader,sources=['hosts'])
  self.variable_manager = VariableManager(loader=self.loader,inventory=self.inventory)
  self.variable_manager.extra_vars = {"image_name": image_name,'host':hostlist}
  def runplaybook(self):
  playbook = PlaybookExecutor(
  playbooks=self.playbooks,
  inventory=self.inventory,
  variable_manager=self.variable_manager,
  loader=self.loader,
  options=self.options,
  passwords=None)
  playbook._tqm._stdout_callback = ResultsCollector()
  playbook.run()
  

  

  
if __name__ == '__main__':
  # 创建对象
  an1 = AnsibleAPI('192.168.194.129,192.168.194.128','common-oss-dc3a25.tar',['/etc/ansible/update.yml'])
  #an2 = AnsibleAPI('192.168.194.128','common-oss-dc3a25.tar',['/etc/ansible/update.yml'])
  #processes = []
  p1 = multiprocessing.Process(name='process_one',target=an1.runplaybook)
  #p2 = multiprocessing.Process(name='process_two',target=an1.runplaybook)
  #processes.append(p1)
  #processes.append(p2)
  #for p in processes:
  #p.start()
  

  # 等待子进程结束,主进程退出
  #for p in processes:
  #p.join()#可以加浮点数参数,等待多久就不等了
  p1.start()
  if p1.is_alive():
  print('正在发布')
  else:
  print('发布结束')
页: [1]
查看完整版本: ansible2.4 api调用