saltstack命令行管理通用脚本

zhltom

#!/usr/bin/env python　　
# -*- coding:utf8 -*-
　　
# create by xmj 20170703
　　

　　
import sys,os
　　
import time
　　
import re
　　
from ConfigParser import ConfigParser
　　
import salt.client
　　
import logging
　　

　　
PATH = os.path.split(os.path.realpath(__file__))[0]
　　

　　
if not os.path.isdir('log'):
　　
    os.makedirs('log')
　　

　　
log = open('log/%s.log' % time.strftime('%Y%m%d_%H%M%S'),'w+')
　　
def WriteLogPrint(comment):
　　
    log.write(comment + '\n')
　　
    print comment
　　

　　
class OperationHostProfile():
　　
    def __init__(self):
　　
        self.hostlist = ConfigParser()
　　
        self.hostlist.read("hostlist.ini")
　　

　　
    def get_sections(self):
　　
        sections = self.hostlist.sections()
　　
        return sections
　　

　　
    def get_hosts(self, section):
　　
        hosts = self.hostlist.get(section, 'hosts').split(',')
　　
        return hosts
　　

　　
class OperationStateProfile():
　　
    def __init__(self):
　　
        self.state = ConfigParser()
　　
        self.state.read('state.ini')
　　

　　
    def get_sections(self):
　　
        sections = self.state.sections()
　　
        return sections
　　

　　
    def get_comment(self,section):
　　
        comment = self.state.get(section, 'comment')
　　
        return comment
　　

　　
    def get_operate(self,section):
　　
        operate = []
　　
        operate.append(self.state.get(section,'operate'))
　　
        return operate
　　

　　

　　

　　
class OperationSalt():
　　
    def __init__(self):
　　
        self.OperationStateProfile = OperationStateProfile()
　　
        self.OperationHostProfile = OperationHostProfile()
　　

　　
    def saltcmd(self,state,hostlist):
　　
        try:
　　
            client = salt.client.LocalClient()
　　
        except Exception, e:
　　
            print "Error: " + str(e)
　　
        for host in hostlist:
　　
            isalive = client.cmd(host,'test.ping')
　　
            WriteLogPrint("--------------------------------------")
　　
            if isalive:
　　
                WriteLogPrint("%s 连通性检查：[\033[32;40m成功\033[0m]" % host)
　　
                jid = client.cmd_async(host,'state.sls',state)
　　
                ret = client.get_cache_returns(jid)
　　
                t = 0
　　
                while not ret:
　　
                    time.sleep(5)
　　
                    if t == 12:
　　
                        WriteLogPrint("Connection Failed!")
　　
                        WriteLogPrint("%s 升级操作 [\033[31;40m失败\033[0m]" % host)
　　
                        break
　　
                    else:
　　
                       ret = client.get_cache_returns(jid)
　　
                       t += 1
　　
                if ret:
　　
                    WriteLogPrint("jid: %s" % jid)
　　
                    WriteLogPrint("%s 升级操作[\033[32;40m成功\033[0m]，以下为详细信息" % host)
　　
                    for operate in ret[host]['ret'].keys():
　　
                        if ret[host]['ret'][operate]['result'] == True:
　　
                            WriteLogPrint(ret[host]['ret'][operate]['comment'] + " [\033[32;40m成功\033[0m]")
　　
                        else:
　　
                            WriteLogPrint(ret[host]['ret'][operate]['comment'] + " [\033[31;40m失败\033[0m]")
　　
                else:
　　
                    WriteLogPrint("%s 升级操作 [\033[31;40m失败\033[0m]" % host)
　　
            else:
　　
                WriteLogPrint("%s 连通性检查：[\033[31;40m失败\033[0m]" % host)
　　
                continue
　　

　　
    def get_state(self):
　　
        sections = self.OperationStateProfile.get_sections()
　　
        print "请输入[]中的数字，选择操作："
　　
        for section in sections:
　　
            comment = self.OperationStateProfile.get_comment(section)
　　
            print "    [%s]: %s" % (section, comment)
　　
        state = raw_input("请输入: ")
　　
        if state in map(str,range(1, len(sections) + 1)):
　　
            comment = self.OperationStateProfile.get_comment(state)
　　
            operate = self.OperationStateProfile.get_operate(state)
　　
        else:
　　
            print "Error Input!!"
　　
            sys.exit()
　　
        return comment, operate
　　

　　
    def set_user_host(self):
　　
        host = raw_input("多个以逗号或者空格分隔: ")
　　
        hosts = re.split(',| ', host)
　　
        if '' in hosts:
　　
            hosts.remove('')
　　
        return hosts
　　

　　

　　
    def get_host(self):
　　
        sections =  self.OperationHostProfile.get_sections()
　　
        print "请输入[]中的数字，选择目标机器："
　　
        i = 1
　　
        for section in sections:
　　
            hosts = self.OperationHostProfile.get_hosts(section)
　　
            if section == 'gray':
　　
                print "    [%d]: %s" % (i , "灰度")
　　
            else:
　　
                print "    [%d]: %s" % (i, section)
　　
            i += 1
　　
            print "        ps:共%s台，他们是%s" % (len(hosts),hosts)
　　
        print "    [%d]: 手动输入" % (i)
　　
        print "请输入你的选项,",
　　
        options = self.set_user_host()
　　
        print options
　　
        hosts = []
　　
        for option in options:
　　
            if option in map(str,range(1, len(sections) + 2)):
　　
                if option == str(i):
　　
                    print "请输入主机名,",
　　
                    hosts = self.set_user_host()
　　
                else:
　　
                    host = self.OperationHostProfile.get_hosts(sections[int(option) - 1])
　　
                    hosts.extend(host)
　　
            else:
　　
                print "Error Input!!"
　　
                sys.exit()
　　
        return hosts
　　

　　
    def main(self):
　　
        os.system('clear')
　　
        WriteLogPrint("Welcom To Use This Script!")
　　
        comment, operate = self.get_state()
　　
        hosts = self.get_host()
　　
        WriteLogPrint("    你选择的操作是： %s" % comment)
　　
        WriteLogPrint("    目标机器共%s台，分别是： %s" % (len(hosts), hosts))
　　
        confirm = raw_input("请输入[Yes]确认： ")
　　
        if confirm == 'Yes':
　　
            self.saltcmd(operate, hosts)
　　
            log.flush()
　　
        else:
　　
            print "Error Input!!"
　　
            sys.exit()
　　

　　
if __name__ == '__main__':
　　
    a = OperationSalt()
　　
    a.main()