Python第40天 salt-api

2019-1-21 09:08:05

salt-api

#!/usr/bin/env python
# _*_ coding:utf-8 _*_
import requests
import json
try:
    import cookielib
except:
    import http.cookiejar as cookielib
# 使用urllib2请求https出错，做的设置
import ssl
context = ssl._create_unverified_context()
# 使用requests请求https出现警告，做的设置
# from requests.packages.urllib3.exceptions import *
# requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
requests.packages.urllib3.disable_warnings()
'''需要自行修改'''
salt_api = "https://192.168.127.78:8888/"
username = "saltapi"
password = "password"
class SaltApi(object):
    def __init__(self, url):
        self.url = url
        '''需要自行修改'''
        self.username = username
        '''需要自行修改'''
        self.password = password
        self.headers = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/50.0.2661.102 Safari/537.36",
            "Content-type": "application/json"
        }
        self.params = {'client': 'local', 'fun': '', 'tgt': ''}
        self.login_url = salt_api + "login"
        self.login_params = {'username': self.username, 'password': self.password, 'eauth': 'pam'}
        # print('------->',self.login_params)
        self.token = self.get_data(self.login_url, self.login_params)['token']
        self.headers['X-Auth-Token'] = self.token

    def get_data(self, url, params):
        send_data = json.dumps(params)
        request = requests.post(url, data=send_data, headers=self.headers, verify=False)
        response = request.json()
        result = dict(response)
        return result['return'][0]

    def salt_command(self, tgt, method, arg=None):
        """远程执行命令，相当于salt 'client1' cmd.run 'free -m'"""
        if arg:
            params = {'client': 'local', 'fun': method, 'tgt': tgt, 'arg': arg,}
        else:
            params = {'client': 'local', 'fun': method, 'tgt': tgt}
        result = self.get_data(self.url, params)
        return result

#自调用
def main():
    salt = SaltApi(salt_api)
    salt_client = '*'
    salt_test = 'grains.items'
    result = salt.salt_command(salt_client, salt_test)
    return result

if __name__ == '__main__':
    main()