远程执行命令通用脚本

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

import os
import sys
import time
import argparse
import datetime
import paramiko

reload(sys)
sys.setdefaultencoding("utf-8")

username = 'root'
password = ''

#change return color
def G(s):
return "%s[32;2m%s%s[0m"%(chr(27), s, chr(27))
def R(s):
return "%s[31;2m%s%s[0m"%(chr(27), s, chr(27))

def cmd_exc(ip, username, password):
conn = paramiko.SSHClient()
conn.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
conn.connect(hostname = ip, username = username, password = password, timeout = 10)
stdin, stdout, stderr = conn.exec_command(cmd)
result = stdout.readlines()
except:
print R("无法连接")
conn.close()
try:
return G(result)
except UnboundLocalError:
pass

if __name__ == "__main__":
parser=argparse.ArgumentParser(description='ssh_cmd', usage='%(prog)s [options]')
parser.add_argument('-H','--host', nargs='?', dest='listhost', help='主机/多个主机用","分割')
parser.add_argument('-f','--file', nargs='?', dest='filehost', help='主机列表文件')
parser.add_argument('-m','--command', nargs='?', dest='command', help='执行命令')
if len(sys.argv)==1:
parser.print_help()
else:
args=parser.parse_args()
cmd = args.command
if args.listhost is not None and args.filehost is None and args.command is not None:
for ip in args.listhost.split(','):
print G(ip)
print G('-'*80)
print cmd_exc(ip, username, password)
print
elif args.listhost is None and args.filehost is not None and args.command is not None:
try:
with open(args.filehost) as f:
for ips in f.readlines():
ip = ips.replace('\n', '')
print G(ip)
print G('-'*80)
print cmd_exc(ip, username, password)
print
except IOError:
print R('主机列表文件不存在')
else:
print R('主机或命令不能为空')