分享一个python下func module的使用脚本

func是个很好用的运维工具，但很久没对知识做总结。前段时间写了个小工具方便自己及身边同事使用。

主要目的是方便使用，将常用的func功能集中起来；二是返回日志比较靠谱，比如说运行一个'date'命令，返回结果为：

"主机名" ##  "0" "Thu Sep 12 16:41:50 CST 2013"

主机名  ## “命令状态” “命令内容”

并且在出错时也不会大量返回python错误信息：）

fun的python modulelist地址在：https://fedorahosted.org/func/wiki/ModulesList

51cto插入代码换行不正常

#!/usr/bin/env python
#-*- coding: utf-8 -*-
# [email protected]
# 通过func模块连接LINUX服务器，运行指定命令
# ver-0.98
# 2013-09-09
"""
修改日志:
    ver-0.95
        1.调整运行参数
        2.新增推送文件功能
        3.修改日志格式
    ver-0.98
        1.首个靠谱版本
        2.新增修改umask为系统默认"0022"
"""
"""
必须在func master中运行
3种运行方式
    运行命令行
        python FuncExec.py -c "命令"
    运行程序(shell/python or other)
        python FuncExec.py -r 文件名
    传送文件到目标服务器
        python FuncExec.py -s 本地文件路径 远端文件路径
"""
import func.overlord.client as func
import getopt
import sys
import os
hostfile = 'server.csv'
#服务器主机信息文件格式: hostname
worklogfile = 'work.log'
def func_chk(hostname):
    """
        import func.overlord.client as func
        Usage: func_chk(hostname)
        Des: 输入（主机名） --> 输出（主机的func状态func_status，检查信息chkfunclog）
    """
    global func_status
    global chkfunclog
    func_status = ''
    chkfunclog = '"%s" ## "func service ok"' % (hostname)
    chkfunc_dict = {}
    chk_host_cmd = 'grep '+ hostname + ' /etc/hosts' + '|wc -l'
    chk_result = os.popen(chk_host_cmd).read().strip('\n')
    if chk_result == '1':
        try:
            client = func.Client(hostname)
            chk_sys_cmd = 'hostname'
            chkfunc_dict = client.command.run(chk_sys_cmd)
            chkfunc_dict_list = chkfunc_dict[hostname]
            # 列表去空值
            while '' in chkfunc_dict_list:
                chkfunc_dict_list.remove('')
            chkfunc_result = chkfunc_dict_list[1].strip('\n')
            if chkfunc_result == hostname.upper():
                func_status = 'ok'
                chkfunclog = '"%s" ## "%s" "%s"' % (hostname,chkfunc_dict_list[0],chkfunc_dict_list[1].strip('\n'))
            elif len(chkfunc_dict_list) >= 3:
                func_status = 'fail'
                m = 3
                chkfunclog = '"%s" ## ' % (hostname)
                for i in range(0,m):
                    chkfunclog = chkfunclog + ' "%s"' % chkfunc_dict_list[i]
            else:
                func_status = 'fail'
                chkfunclog = '"%s" ## "%s" "%s"' % (hostname,chkfunc_dict_list[0],chkfunc_dict_list[1].strip('\n'))
                #chkfunclog = '"%s" ## "serv hostname isn`t matching"' % (hostname)
        except:
            func_status = 'fail'
            chkfunclog = '"%s" ## "certificate not found"' % (hostname)
    else:
        func_status = 'fail'
        chkfunclog = '"%s" ## "serv info error in /etc/hosts"' % (hostname)
def output_log(logfile, log_content):
    """
        Usage: output_log(日志文件，日志内容)
        Des: 输入（日志文件，日志内容） --> 输出（日志内容追加写入到日志文件）
    """
    log = open(logfile,'a')
    txt_content = log_content + '\n'
    log.write(txt_content)
    log.close()
def usage():
    howto = '''
    Usage: ./FuncExec.py -argument
        -c $your_command   # Execute a command line
        -r $your_script_name    # Execute a script (shell, python or other, running as './$your_script_name')
        -s $local_file $remote_file   # Send a file to target server
        -h    # Print how to
    '''
    print(howto)
"""
MAIN: 开始运行
"""
notice = 'null'
#-- 确认主机名列表文件
pwd = os.getcwd()
hostfile = pwd+'/server.csv'
if os.path.isfile(hostfile):
    print('## hostfiile exist ......')
else:
    print('Unable to find hostfile: '+ hostfile + '.  Quit ......')
    sys.exit()
#-- 确认运行参数&运行种类
try:
    opts,args = getopt.getopt(sys.argv[1:], 'c:r:s');
    len_opts = str(len(opts))
    if len_opts == '0':
        print('## getopt error!')
        usage()
        sys.exit()
    for o,a in opts:
        if o in ('-c'):
            cmd = a
            run_module = 'cmd'
            notice = 'Your command is: '+ cmd
        elif o in ('-r'):
            script_file = a
            run_module = 'script'
            notice = 'Your script file is: '+ script_file
        elif o in ('-s'):
            local_path = args[0]
            remote_path = args[1]
            run_module = 'file'
            notice = 'Local file is: '+ local_path + '\n' + 'Remote file is: '+ remote_path
except getopt.GetoptError:
    print('## getopt error!')
    usage()
    sys.exit()
#-- 用户确认运行内容
running = True
print(notice)
while running:
    run_flag = raw_input('Enter "yes" to continue or Enter "exit" to quit ...... ')
    if run_flag == 'yes':
        print('Command had confirm, start running ...... ')
        running = False
    elif run_flag == 'exit':
        print('Ok , program will be stop ...... ')
        sys.exit()
#-- 打开hostfile，逐台操作主机
f = open(hostfile)
line = f.readline()
while line:
    hostname = 'null'
    hostname = line.strip('\n')
    # 验证主机func状态
    func_chk(hostname)
    if func_status == 'ok':
        client = func.Client(hostname)
        run_dict = {}
        if run_module in ('cmd'):
            # 跑命令
            run_dict = client.command.run('umask 0022;'+ cmd)
        elif run_module in ('script'):
            # 跑脚本
            remote_path = '/tmp/func_tmp_script'
            client.local.copyfile.send(script_file, remote_path)
            client.command.run('chmod u+x '+ remote_path)
            run_dict = client.command.run('umask 0022; ./'+ remote_path)
        elif run_module in ('file'):
            # 抛文件
            run_dict = client.local.copyfile.send(local_path, remote_path)
        # 调整输出
        try:
            run_dict_list = run_dict[hostname]
            while '' in run_dict_list:
                run_dict_list.remove('')
            run_log = '"%s" ## ' % (hostname)
            for i in run_dict_list:
                log_info = str(i).strip('\n')
                run_log = '%s "%s"' % (run_log,log_info)
        except:
            # type(run_dict) = bool; run_dict = True
            if run_dict == True:
                run_log = '"%s" ## "file transfer ok"' % (hostname)
            else:
                run_log = '"%s" ## "file transfer fail"' % (hostname)
        # 输出信息
        output_log(worklogfile,run_log)
    elif func_status == 'fail':
        output_log(worklogfile, chkfunclog)
    print '%s run finished.' % (hostname)
    # 结束
    line = f.readline()
# 关循环
f.close()