CMDB_salt-api引數請求等
摘要：
具體的專案目錄結構如下：
目錄詳情：
/func/main.py 程式主入口；
/conf/settings.py 配置檔案，用於儲存使用者名稱、密碼、url 等資料；
/func/hosts/.py 定義各個引數獲取的方法；
/...
具體的專案目錄結構如下：
目錄詳情：
- /func/main.py 程式主入口；
- /conf/settings.py 配置檔案，用於儲存使用者名稱、密碼、url 等資料；
- /func/hosts/.py 定義各個引數獲取的方法；
- /func/task/.py 呼叫 salt-api 執行命令；
settings.py
HOST_FUNC_DIC = {
'disk': 'func.hosts.disk.Disk',
'mem': 'func.hosts.mem.Mem',
'cpu': 'func.hosts.cpu.Cpu',
'os': 'func.hosts.os.Os',
'ip': 'func.hosts.ip.Ip',
}
HOST_LI = ['mem']
SALT_API_CON = "https://192.168.171.173:8001/"
SALT_USERNAME = 'saltapi'
SALT_PASSWORD = 'XXXXXX'
run.py作為獲取資源的主程式，程式碼如下：
import importlib
import os, sys
BASEDIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASEDIR)##將檔案path加入環境變數
from conf.settings import HOST_FUNC_DIC
from conf.settings import HOST_LI
from func.task.get_data import send_main
for host in HOST_LI:
path = HOST_FUNC_DIC.get(host)
module_path, class_name = path.rsplit('.', maxsplit=1)
module = importlib.import_module(module_path)
disk_class = getattr(module, class_name)
JG = disk_class()
func = JG.run()
JG1 = send_main(func)
/func/task/get_data.py
import requests
import json
try:
import cookielib
except:
import http.cookiejar as cookielib
# 使用urllib2請求https出錯，做的設定
import ssl
context = ssl._create_unverified_context()
from func.task.salt_client import SaltApi
from conf.settings import SALT_API_CON
# 使用requests請求https出現警告，做的設定
from requests.packages.urllib3.exceptions import InsecureRequestWarning
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
def send_main(params):
salt = SaltApi(SALT_API_CON, params)
result1 = salt.salt_command()
if __name__ == '__main__':
send_main({'client': 'local', 'fun': '192.168.171.172', 'tgt': 'test.ping'})
/func/task/salt_client.py
import json
import requests
# from conf.settings import SALT_API_CON
from conf.settings import SALT_USERNAME
from conf.settings import SALT_PASSWORD
class SaltApi:
def __init__(self, url, params):##定義salt api介面的類，初始化獲得token
self.url = url
self.username = SALT_USERNAME
self.password = SALT_PASSWORD
self.params = params
self.headers = {##設定初始化header頭資訊
"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.login_url = url + "login"##地址拼接
self.login_params = {'username': self.username, 'password': self.password, 'eauth': 'pam'}
# self.token = self.get_data(self.login_url, self.login_params)['token']
self.token = self.get_data(self.login_url, self.login_params).get('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)
# print result
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
mem.py
from func.hosts.main import Base#載入自定義的一個類
class Disk(Base):
def run(self):
self.method="cmd.run"
self.tgt="*"
self.arg="free -m"
return {"client":"local", "fun":self.method, "tgt":self.tgt, "arg":self.arg}
cpu.py
from func import Main
Class Cpu(Mian)
def run(self)
self.method = 'cmd.run'
self.tgt = 'hc-02'
self.agr = 'cat /proc/cpuinfo'
#print ('run...cpu')
retrun {'client':'local', 'func':self.method, 'tgt':self.tgt, 'arg':self.arg}
/func/hosts/main.py
main.py 定義基類，該類中定義 run 方法，其餘的引數類均繼承該類 若其餘類未定義 run 方法，則會呼叫父類的 run 方法，從而給出提示資訊；class Base(object):
def __init__(self):
self.method = ''
self.tgt = ''
self.arg = ''
def run(self):
raise NotImplementedError('Not run func')