通过flask框架写成http接口调用Ansible

534523

#*******一、flask_ansible.py文件
#!/usr/bin/env python
#_*_ coding:utf-8 _*_
import json
from flask import Flask,request
from ansible_api_job import Ansibles

app = Flask(__name__)

@app.route('/ansible/command', methods=['GET','POST'])
def command():
    if request.method == 'POST':
        jsondata = request.get_data()
        dictdata = json.loads(jsondata)
        ansible = Ansibles(dictdata['host']) #实例化Ansible对象
        res = ansible.shell(dictdata['command'])
        return res
    else:
        return '<h1>Access error</h1>'

@app.route('/ansible/copyfile', methods=['GET','POST'])
def copyfile():
    if request.method == 'POST':
        jsondata = request.get_data()
        dictdata = json.loads(jsondata)
        ansible = Ansibles(dictdata['targethost']) #实例化Ansible对象
        res = ansible.copyfile(str(dictdata['srcfile']),str(dictdata['dectdir']))
        return res
    else:
        return '<h1>Access error</h1>'

@app.route('/ansible/pullfile',methods=['GET','POST'])
def pullfile():
    if request.method == 'POST':
        jsondata = request.get_data()
        dictdata = json.loads(jsondata)
        ansible = Ansibles(dictdata['srchost'])
        res = ansible.pull(dictdata['pullsrcfile'], dictdata['savelocaldir'])
        return res
    else:
        return '<h1>Access error</h1>'
if __name__ =='__main__':
    app.run(debug=True,host='0.0.0.0')

#************二、ansible_api_job.py文件
#!/user/bin/eng python
# -*- coding=utf-8 -*-
import ansible.runner

class Ansibles(object):
    def __init__(self,hostname):
        self.hostlist = "host.txt"
        self.user = "root"
        self.passwd = "1qaz#EDC"
        self.pattern = "vmserver"

        self.hostname = hostname
        hostlist = self.hostname.split(',')
        hostfile = open('host.txt', 'w+')
        hostfile.writelines('[vmserver]' + '\n')
        for line in hostlist:
            hostfile.writelines(line + '\n')
        hostfile.close()

    def shell(self,command):
        results = ansible.runner.Runner(
            host_list= self.hostlist,
            remote_user = self.user,
            remote_pass = self.passwd,
            module_name = 'shell',
            module_args = command,
            pattern = self.pattern,
            forks = 10
            ).run()
        for (hostname,result) in results['contacted'].items():
            if result['stdout'] == "":
                return "HOST:%s, ERROR>>(%s)" % (hostname, result['stderr'])
            else:
                return "HOST:%s, RESULTS>>(%s)" % (hostname, result['stdout'])

    def copyfile(self,srcfile,dectdir):
        results = ansible.runner.Runner(
            host_list=self.hostlist,
            remote_user=self.user,
            remote_pass=self.passwd,
            module_name='copy',
            module_args='src=%s dest=%s' % (srcfile,dectdir),
            pattern=self.pattern,
            forks=10
        ).run()
        for (hostname,result) in results['contacted'].items():
            if 'failed' in result:
                return "HOST:%s, ERROR>>(%s)" % (hostname,result['msg'])
            else:
                return "HOST:%s, copy ok" % (hostname)

    # 文件拉取到本地
    def pull(self,pullsrcfile,savelocaldir):
        pullfileres = ansible.runner.Runner(
            host_list=self.hostlist,
            remote_user=self.user,
            remote_pass=self.passwd,
            module_name='fetch',
            module_args='src=%s dest=%s flat=yes' % (pullsrcfile, savelocaldir),
            pattern=self.pattern,
            forks=10
        ).run()
        for (hostname,result) in pullfileres['contacted'].items():
            if 'msg' in result:
                return "HOST:%s ERROR>>(%s)" % (hostname,result['msg'])
            else:
                return 'HOST:%s pull ok' % (hostname)

if __name__ == "__main__":
    pass


     
#*********三、urlib2调用http接口
#_*_coding:utf-8_*_
import urllib2
import json

#调用执行命令HTTP接口
def ansible_http_post_command(host,command):
    url = 'http://192.168.89.8:5000/ansible/command'
    data = {'host': host, 'command': command}
    headers = {'Content-Type': 'application/json'}
    req = urllib2.Request(url=url, headers=headers, data=json.dumps(data))
    response = urllib2.urlopen(req)
    return response.read()
#res = ansible_http_post_command('vm5',"uname -a")
#print res

#调用传送文件到目标主机的HTTP接口
def ansible_http_post_copyfile(srcfile,targethost,dectdir):
    url = 'http://192.168.89.8:5000/ansible/copyfile'
    data = {'srcfile': srcfile, 'targethost': targethost,'dectdir':dectdir}
    headers = {'Content-Type': 'application/json'}
    req = urllib2.Request(url=url, headers=headers, data=json.dumps(data))
    response = urllib2.urlopen(req)
    return response.read()
res = ansible_http_post_copyfile('/tmp/12345.txt','vm4','/root/')
print res

#调用从目标主机中拉取文件到本地的HTTP接口
def ansible_http_post_pullsend(srchost,pullsrcfile,savelocaldir):
    url = 'http://192.168.89.8:5000/ansible/pullfile'
    data = {'srchost':srchost,'pullsrcfile':pullsrcfile,'savelocaldir':savelocaldir}
    headers = {'Content-Type': 'application/json'}
    req = urllib2.Request(url=url, headers=headers, data=json.dumps(data))
    response = urllib2.urlopen(req)
    return response.read()
#res = ansible_http_post_pullsend('vm2','/tmp/12345.txt','/tmp/')
#print res