saltstack二次开发（三）

以下代码实现的功能：

通过saltstack的httpapi在minion上安装软件。

vim saltapi.py

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

import json

import requests


class SaltServer(object):
    def __init__(self):
        self.session = requests.session()
        self.token = self.getToken()

    # 获取token
    def getToken(self):
        url =  "http://192.168.127.165:8000/login"
        headers = {"Accept": "application/json"}
        data = {
            "username": "saltapi",
            "password": "saltapi",
            "eauth": "pam"
        }
        res = self.session.post(url=url, headers=headers, data=data)
        text = res.text
        result = json.loads(text)
        token = result.get("return")[0].get("token")
        return token
    
    # 执行modules
    def runModules(self, minionid, fun, arg=None):
        url = "http://192.168.127.165:8000"
        data = {
            "client": "local",
            "tgt": minionid,
            "fun": fun,
            "arg": arg
        }
        resultBean = dict()
        try:
            res = self.session.post(url=url,  data=data)
            text = res.text
            data = json.loads(text).get("return")
            resultBean['code'] = 0
            resultBean['message'] = "success"
            resultBean['data'] = data
        except Exception as e:
            resultBean['code'] = 0
            resultBean['message'] = "success"
            resultBean['data'] = e
        finally:
            return resultBean

    # 执行runner
    def runRunner(self, fun, arg=None):
        url = "http://192.168.127.165:8000"
        data = {
            "client": "runner",
            "fun": fun,
            "arg": arg
        }
        resultBean = dict()
        try:
            res = self.session.post(url=url,  data=data)
            text = res.text
            data = json.loads(text).get("return")
            resultBean['code'] = 0
            resultBean['message'] = "success"
            resultBean['data'] = data
        except Exception as e:
            resultBean['code'] = 0
            resultBean['message'] = "success"
            resultBean['data'] = e
        finally:
            return resultBean

vim installAPP.py

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

# nginx.sh里面就写了一句简单的yum install -y nginx

import SaltServer

saltServer = SaltServer()

result1 = saltServer.runModules('192.168.127.166', 'cp.get_file', ['salt://_shell/app/nginx.sh', '/usr/local/src/installnginx.sh']) # 把安装nginx脚本同步到minion端
print(result1)
result2 = saltServer.runModules('192.168.127.166', 'installApp.nginx')
print(result2)

执行 python installApp.py之后，就可以在192.168.127.166上面看到nginx已经安装好了。