多线程ssh工具

2011年7月18日 | 分类: 乱七八糟 | 标签: , , ,

多线程SSh工具Tssh-1.3

程序执行前请安装pextect模块安装很简单 python setup.py install 我的python版本 2.5 和 2.4.3测试程序都没有问题
推荐大家使用 2.5及以上版本 2.5以下下版本不支持  except:  finally: 混合使用 可能会有问题

Tssh-1.3 修补了前面版本的一个小Bug就是连接服务器超时的时候会报出一大堆错误
更新了密钥使用的方式前版本对使用密钥的人是一个困扰
支持主机列表中开头#号注释
建议使用Python 2.5 以上版本
使用帮助
如果密码不同就将密码写在列表中格式内容为 IP:密码
如果使用密钥做认证就直接使用 -f 指定一个 IP 列表就可以 别的就不用在指定了 但是别忘了指定 command 哦
支持 sudo  直接在 -c “sudo 命令” 就可以密码会以当密码做认证
-h 会得到程序帮助
-f 指定一个linux主机列表
-u 指定用户名
-p 指定密码  当你的列表中已经存在密码请不要尝试输入-p 参数 谢谢
-c 指定要指定的命令 多个命令使用 ; 分号或者 && 来连接
-t 指定线程数目如果机器比较多的话建议指定一下 默认程序会以列表中机器数目来创建线程
-s 将执行结果输出到标准输出.我不建议这样使用因为线程的处理比较复杂你并不知道谁快谁慢,导致输出的结果很乱
默认情况程序会将每个IP的执行结果保存在脚本执行目录下Tssh/log/中大家可以查看执行结果

原文：http://bbs.linuxtone.org/thread-7170-1-1.html

pssh—多线程ssh客户端工具 1.1

在执行wget  这样的操作的时候有问题。下载成功，但是显示没成功。其他目前还正常

  1. #!/usr/bin/env python
  2. # coding:utf-8
  3. import paramiko,argparse,getpass,sys,os
  4. import threading,Queue,logging
  5. from colors import light_red,light_green
  6. hostqueue=Queue.Queue()
  7. MAXTHREAD=10
  8. logging.disable(10)
  9. def logdog():
  10.     logger = logging.getLogger()
  11.     hand = logging.FileHandler(“exec.log”,’a’,’utf-8′)
  12.     formatter = logging.Formatter(‘%(asctime)s %(levelname)s %(message)s’)
  13.     hand.setFormatter(formatter)
  14.     logger.addHandler(hand)
  15.     logger.setLevel(logging.INFO)
  16.     logger.disabled
  17.     return logger,hand
  18. def funcssh(optargs,host):
  19.     connect=paramiko.SSHClient()
  20.     connect.load_system_host_keys()
  21.     connect.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  22.     if optargs[‘certificate’]:
  23.         if optargs[‘ktype’]==’rsa’:
  24.             try:
  25.                 key=paramiko.RSAKey.from_private_key_file(optargs[‘certificate’],password=optargs[‘passwd’])
  26.             except paramiko.SSHException,e:
  27.                 return host+”\t\t”+str(e)
  28.         else:
  29.             try:
  30.                 key=paramiko.DSSKey.from_private_key_file(optargs[‘certificate’],password=optargs[‘passwd’])
  31.             except paramiko.SSHException,e:
  32.                 return host+”\t\t”+str(e)
  34.         try:
  35.             connect.connect(host,optargs[‘port’],pkey=key)
  36.         except BaseException,e:
  37.             return host+”\t\t”+str(e)
  38.     else:
  39.         try:
  40.             connect.connect(host,optargs[‘port’],optargs[‘user’],optargs[‘passwd’])
  41.         except BaseException,e:
  42.             return host+str(e)
  43.     if optargs[‘command’]:
  44.         stdin,stdout,stderr=connect.exec_command(“%s” % optargs[‘command’])
  46.         s=stdout.read()
  47.         if s:
  48.             sout=’\n’+host.center(50,’#’)+’\n’+s+’EXEC COMMAD OK\n’
  49.             logger,hand=logdog()
  50.             logger.log(20, sout)
  51.             hand.flush()
  52.             logger.removeHandler(hand)
  53.             s=host.center(50,’#’)+’\n’+s+light_green(‘EXEC COMMAD OK\n’)
  54.             connect.close()
  55.             return s
  56.         e=stderr.read()
  57.         if e:
  58.             e=’\n’+host.center(50,’#’)+’\n’+e+’EXEC COMMAD Faild\n’
  59.             logging.error(e,exc_info=False)
  60.             logger,hand=logdog()
  61.             logger.log(40, e)
  62.             hand.flush()
  63.             logger.removeHandler(hand)
  64.             connect.close()
  65.             return light_red(e)
  66.         if not e and not s:
  67.             e=’\n’+host.center(50,’#’)+’\n’+’EXEC COMMAD Sucess But No resolut\n’
  68.             return e
  71.     else:
  72.         return “No Command To Exec”
  74. class pssh(threading.Thread):
  75.     global hostqueue
  76.     paramiko.util.log_to_file(‘ssh.log’)
  77.     def __init__(self,**optargs):
  78.         threading.Thread.__init__(self)
  79.         self.optargs=optargs
  80.         paramiko.util.log_to_file(‘ssh.log’)
  81.         #self.daemon=True
  82.     def run(self):
  83.         while not hostqueue.empty():
  84.             try:
  85.                 self.host=hostqueue.get_nowait()
  86.             except:
  87.                 sys.exit()
  88.             resoult=funcssh(self.optargs,self.host)
  89.             print resoult
  90.             hostqueue.task_done()
  91. class pscp(threading.Thread):
  92.     global hostqueue
  93.     paramiko.util.log_to_file(‘ssh.log’)
  94.     def __init__(self,**optargs):
  95.         threading.Thread.__init__(self)
  96.         self.optargs=optargs
  97.     def walk(self,dirs,todir):
  98.         try:
  99.             self.sftp.mkdir(todir+dirs.lstrip(self.sd))
  100.         except IOError,e:
  101.             pass
  102.         dirs=dirs.rstrip(os.sep)+os.sep
  103.         for i in os.listdir(dirs):
  104.             if os.path.isdir(dirs+i):
  105.                 print dirs+i
  106.                 self.walk(dirs+i,todir)
  107.             else:
  108.                 print dirs+i,todir+dirs.lstrip(self.sd)+i
  109.                 self.sftp.put(dirs+i,todir+dirs.lstrip(self.sd)+i)
  110.     def run(self):
  111.         #while not hostqueue.empty():
  112.         self.host=hostqueue.get()
  113.         try:
  114.             self.connect=paramiko.Transport((self.host,self.optargs[‘port’]))
  115.         except paramiko.SSHException,e:
  116.                 print e
  117.                 sys.exit()
  118.         if self.optargs[‘certificate’]:
  119.             if args.ktype==’rsa’:
  120.                 try:
  121.                     self.key=paramiko.RSAKey.from_private_key_file(self.optargs[‘certificate’],password=self.optargs[‘passwd’])
  122.                 except paramiko.SSHException,e:
  123.                     print e
  124.                     sys.exit()
  125.             else:
  126.                 try:
  127.                     self.key=paramiko.DSSKey.from_private_key_file(self.optargs[‘certificate’],password=self.optargs[‘passwd’])
  128.                 except paramiko.SSHException,e:
  129.                     print e
  130.                     sys.exit()
  131.             try:
  132.                 self.connect.connect(username=self.optargs[‘user’],pkey=self.key)
  133.             except paramiko.SSHException,e:
  134.                 print e
  135.                 sys.exit()
  136.         else:
  137.             try:
  138.                 self.connect.connect(username=self.optargs[‘user’],password=self.optargs[‘passwd’])
  139.             except paramiko.SSHException,e:
  140.                 print e
  141.                 sys.exit()
  142.         self.sftp = paramiko.SFTPClient.from_transport(self.connect)
  143.         self.sd=self.optargs[‘sdir’].rstrip(os.sep)+os.sep
  144.         self.td=self.optargs[‘ddir’].rstrip(os.sep)+os.sep
  145.         if not os.path.isdir(self.optargs[‘sdir’]):
  146.             try:
  147.                 self.sftp.put(self.optargs[‘sdir’],self.optargs[‘ddir’])
  148.             except paramiko.SSHException,e:
  149.                 print e
  150.                 sys.exit()
  151.         else:
  152.             self.walk(self.sd,self.td)
  153.         hostqueue.task_done()
  154. def getargs():
  155.     parser = argparse.ArgumentParser(description=’ssh.’,formatter_class=argparse.ArgumentDefaultsHelpFormatter,prog=’PSSH’,epilog=’Create by dennis’,usage=’%(prog)s [options]’)
  156.     parser.add_argument(‘-i’,action=’store’,metavar=’certificate’,dest=’certificate’,help=’The Certificate file’)
  157.     parser.add_argument(‘-t’,action=’store’,choices=[‘ssh’,’scp’],metavar=’ctype’,default=’ssh’,dest=’ctype’,help=”default is ssh, the value is ssh or scp “)
  158.     parser.add_argument(‘-u’,action=’store’,default=getpass.getuser(),metavar=’user’,dest=’user’,help=’The Certificate’)
  159.     parser.add_argument(‘-p’,action=’store_true’,default=False,dest=’passwd’,help=”Passwd .if use the -i option.the passwd is for the certificate”)
  160.     #parser.add_argument(‘-c’,action=’store_true’,dest=’command’,default=False,help=’Turn On Commandline Input’)
  161.     parser.add_argument(‘-c’,action=’store’,metavar=’command’,dest=’command’,help=’ Command Input’)
  162.     parser.add_argument(‘-P’,action=’store’,default=22,type=int,metavar=’port ‘,dest=’port’,help=’ssh Port .Default is 22′)
  163.     parser.add_argument(‘-sd’,action=’store’,metavar=’sdir ‘,dest=’sdir’,help=’source file/dir .’)
  164.     parser.add_argument(‘-f’,action=’store’,metavar=’filename’,dest=’filename’,help=’filename include all host .’,required=True)
  165.     parser.add_argument(‘-td’,action=’store’,metavar=’ddir ‘,dest=’ddir’,help=’dst flle/dir .’)
  166.     parser.add_argument(‘-k’,action=’store’,choices=[‘rsa’,’dsa’],metavar=’ktype’,default=’rsa’,dest=’ktype’,help=”key type .default is rsa.the value is rsa or dsa”)
  167.     parser.add_argument(‘-v’,’-version’,action=’version’ ,version=’%(prog)s 1.0′)
  168.     if len(sys.argv)==1:
  169.         parser.print_help()
  170.         sys.exit(1)
  171.     args = parser.parse_args()
  172.     return args
  173. if __name__ == ‘__main__’:
  174.     args=getargs()
  175.     ctype=args.ctype
  176.     if args.passwd:
  177.         passwd=getpass.getpass()
  178.     else:
  179.         passwd=None
  180.     try:
  181.         with open(args.filename,’r’) as f:
  182.             for i in f:
  183.                 hostqueue.put(i.strip(os.linesep))
  184.     except IOError,e:
  185.         print e
  186.         sys.exit()
  187.     if hostqueue.qsize() < MAXTHREAD:
  188.         MAXTHREAD=hostqueue.qsize()
  189.     if ctype==’ssh’:
  190.         for  i in xrange(0,MAXTHREAD):
  191.             ssh=pssh(ktype=args.ktype,command=args.command,user=args.user,passwd=passwd,certificate=args.certificate,port=args.port,filename=args.filename)
  192.             ssh.start()
  193.         ssh.join()
  195.     else:
  196.         for  i in xrange(0,MAXTHREAD):
  197.             scp=pscp(ktype=args.ktype,sdir=args.sdir,ddir=args.ddir,user=args.user,passwd=passwd,certificate=args.certificate,port=args.port,filename=args.filename)
  198.             scp.start()
  199.         scp.join()

原文：http://queryer.cn/post/122/

以上是些什么东西我也不怎么懂，我只是想要有一个简单方便的ssh多线程代理客户端那ssh翻墙速度不就会更快了。

发表评论
  1. 匿名
    2011年7月19日08:30
    回复 | 引用 | #1

    连我要做什么都没看懂，杯具了。。。。