视频1 视频21 视频41 视频61 视频文章1 视频文章21 视频文章41 视频文章61 推荐1 推荐3 推荐5 推荐7 推荐9 推荐11 推荐13 推荐15 推荐17 推荐19 推荐21 推荐23 推荐25 推荐27 推荐29 推荐31 推荐33 推荐35 推荐37 推荐39 推荐41 推荐43 推荐45 推荐47 推荐49 关键词1 关键词101 关键词201 关键词301 关键词401 关键词501 关键词601 关键词701 关键词801 关键词901 关键词1001 关键词1101 关键词1201 关键词1301 关键词1401 关键词1501 关键词1601 关键词1701 关键词1801 关键词1901 视频扩展1 视频扩展6 视频扩展11 视频扩展16 文章1 文章201 文章401 文章601 文章801 文章1001 资讯1 资讯501 资讯1001 资讯1501 标签1 标签501 标签1001 关键词1 关键词501 关键词1001 关键词1501 专题2001
ansibleapi实现命令异步执行
2020-11-09 13:03:28 责编:小采
文档
代码也可以在这里查看https://github.com/ivonlee/ansible/blob/master/ansible_api_async_run.py

tornado上实现ansible api异步执行,方便php写的运维后台调用,当然php后台还是要做一个类似于队列的东西,将任务存在redis或者mongodb里面,然后有个php进程持续监听任务队列。

相关mysql视频教程推荐:《mysql教程》

下面的脚本运行后,可以用类似POSTMAN工具进行post数据测试,如果你的平台本来就是python的,那更方便了,自己写个简陋的web界面,直接执行了,不用tornado做web容器了。

mongodb里面的表信息,ansible_task是收到的任务,ansible_job里面有任务执行结果

上代码,python比较搓,求大牛带我

import tornado.ioloop
from tornado.options import define, options
import tornado.web
import ansible.runner
from ansible.inventory import Inventory
import simplejson
import hashlib
from pymongo import MongoClient
from bson.objectid import ObjectId
import os
import sys
import time
from psutil import Process
import datetime
from threading import Thread
define("key", default='d41d8cd98f00b204e9800998ecf8427e')
mongoinfo = {"host": "127.0.0.1", "port": "27017", "user":
 "ops", "password": "ops", "dbname": "ansible_log"}
TIME_FORMAT = '%Y-%m-%d %H:%M:%S'
WORKER_TIMEOUT = 5 * 60
NUMBER_OF_TASK_PER_PAGE = 25
ANSIBLE_FORKS = 30
ANSIBLE_INVENTORY = '/etc/ansible/hosts'
def getmd5(str):
 m = hashlib.md5()
 m.update(str)
 return m.hexdigest()
def ConnMongoDB():
 global mongoinfo
 dbhost = mongoinfo['host']
 dbport = mongoinfo['port']
 dbuser = mongoinfo['user']
 dbpwd = mongoinfo['password']
 dbname = mongoinfo['dbname']
 uri = 'mongodb://%s:%s@%s/%s' % (dbuser, dbpwd, dbhost, dbname)
 return uri
class MainHandler(tornado.web.RequestHandler):
 def get(self):
 self.write("Hello, world")
class CommandHandler(tornado.web.RequestHandler):
 def post(self):
 data = simplejson.loads(self.request.body) 
 badcmd = ['reboot','rm','kill','pkill','shutdown','half','mv','dd','mkfs','>','wget']
 type = data['type']
 cmd = data['cmd']
 host = data['host']
 print host
 sign = data['sign']
 isudo = data['sudo']
 cmdinfo = cmd.split(" ",1)
 print type,host,options.key
 hotkey = type+host+options.key
 print hotkey
 result = getmd5(hotkey)
 print result
 if sign != result:
 self.write("Sign is Error")
 else:
 if cmdinfo[0] in badcmd:
 self.write("This is Danger Shell")
 else:
 if "," in host:
 inventory = host.split(",")
 for host in inventory:
 runner = ansible.runner.Runner(
 module_name=type,
 module_args=cmd,
 pattern=host,
 sudo = isudo,
 forks=ANSIBLE_FORKS
 )
 result = runner.run()
 now = datetime.datetime.now()
 true = 'True'
 result['time'] = now.strftime(TIME_FORMAT)
 result['type'] = 'ad-hoc'
 result['sudo'] = isudo
 result['cmd'] = cmd
 result['inventory'] = host
 self.write(result)
 uri = ConnMongoDB()
 client = MongoClient(uri, safe=False)
 db = client.ansible_log
 db.ad_hoc.insert(result)
 else:
 runner = ansible.runner.Runner(
 module_name=type,
 module_args=cmd,
 pattern=host,
 sudo = isudo,
 forks=ANSIBLE_FORKS
 )
 result = runner.run()
 now = datetime.datetime.now()
 true = 'True'
 result['time'] = now.strftime(TIME_FORMAT)
 result['type'] = 'ad-hoc'
 result['sudo'] = isudo
 result['cmd'] = cmd
 result['inventory'] = inventory
 self.write(result)
 uri = ConnMongoDB()
 client = MongoClient(uri, safe=False)
 db = client.ansible_log
 db.ad_hoc.insert(result)
class AsyncTaskHandler(tornado.web.RequestHandler):
 def post(self):
 data = simplejson.loads(self.request.body)
 badcmd = ['reboot', 'rm', 'kill', 'pkill',
 'shutdown', 'half', 'mv', 'dd', 'mkfs', '>', 'wget']
 type = data['type']
 cmd = data['cmd']
 inventory = data['host']
 sign = data['sign']
 isudo = data['sudo']
 cmdinfo = cmd.split(" ", 1)
 print type, inventory, options.key
 hotkey = type + inventory + options.key
 print hotkey
 result = getmd5(hotkey)
 print result
 now = datetime.datetime.now()
 taskinfo = {}
 taskinfo['mode'] = type
 taskinfo['cmd'] = cmd
 taskinfo['inventory'] = inventory
 taskinfo['type'] = 'async ad-hoc'
 taskinfo['start'] = now.strftime(TIME_FORMAT)
 taskinfo['sudo'] = isudo
 uri = ConnMongoDB()
 client = MongoClient(uri, safe=False)
 db = client.ansible_log
 id=db.ansible_task.insert(taskinfo)
 mongoid={"_id":ObjectId(id)}
 print id
 if sign != result:
 self.write("Sign is Error")
 else:
 if cmdinfo[0] in badcmd:
 self.write("This is Danger Shell")
 else:
 runner = ansible.runner.Runner(
 module_name=type,
 module_args=cmd,
 pattern=inventory,
 sudo = isudo,
 forks=ANSIBLE_FORKS
 )
 _, res = runner.run_async(time_limit = WORKER_TIMEOUT)
 now = time.time()
 while True:
 if res.completed or time.time() - now > WORKER_TIMEOUT:
 break
 results = res.poll()
 results = results.get('contacted')
 if results:
 for result in results.items():
 jobinfo = {}
 data = result[1]
 print data
 inventory = result[0]
 jobinfo['inventory']=inventory
 jobinfo['job_id']=data['ansible_job_id']
 jobinfo['cmd']=data['cmd']
 jobinfo['task_id']=id
 uri = ConnMongoDB()
 client = MongoClient(uri, safe=False)
 db = client.ansible_log
 id2 = db.ansible_job.insert(jobinfo)
 mongoid2 = {"_id":ObjectId(id2)}
 if data['rc'] == 0 :
 thisinfo2 = db.ansible_job.find_one(mongoid2)
 thisinfo2['rc']=data['rc']
 thisinfo2['stdout']=data['stdout']
 thisinfo2['stderr']=data['stderr']
 db.ansible_job.save(thisinfo2)
 thisinfo = db.ansible_task.find_one(mongoid)
 thisinfo['end'] = data['end'] 
 thisinfo['rc'] = data['rc']
 db.ansible_task.save(thisinfo)
 elif data['rc'] == 1 :
 thisinfo2 = db.ansible_job.find_one(mongoid2)
 thisinfo2['rc']=data['rc']
 thisinfo2['stderr']=data['stderr']
 db.ansible_job.save(thisinfo2)
 thisinfo = db.ansible_task.find_one(mongoid)
 thisinfo['rc'] = data['rc']
 db.ansible_task.save(thisinfo)
 else:
 thisinfo2 = db.ansible_job.find_one(mongoid2)
 thisinfo2['rc']=data['rc']
 thisinfo2['stderr']=data['msg']
 db.ansible_job.save(thisinfo2)
 thisinfo = db.ansible_task.find_one(mongoid)
 thisinfo['rc'] = data['rc']
 db.ansible_task.save(thisinfo)
 time.sleep(2)
class GetGroupHandler(tornado.web.RequestHandler):
 def get(self):
 i = Inventory()
 groups = i.list_groups()
 self.write('\n'.join(groups))
application = tornado.web.Application([
 (r"/", MainHandler),
 (r"/asynctask", AsyncTaskHandler),
 (r"/command", CommandHandler),
 (r"/getgroup", GetGroupHandler),
])
if __name__ == "__main__":
 application.listen(8888)
 tornado.ioloop.IOLoop.instance().start()

下载本文
显示全文
专题