Chinaunix首页 | 论坛 | 博客

吃鱼翅的鱼xglcr.blog.chinaunix.net

首页　| 　博文目录　| 　关于我

xglcr

  • 博客访问： 302982
  • 博文数量： 32
  • 博客积分： 2179
  • 博客等级： 大尉
  • 技术积分： 443
  • 用 户 组： 普通用户
  • 注册时间： 2007-03-21 10:30
文章分类

全部博文（32）

文章存档

2014年（1）

2013年（4）

2011年（4）

2009年（11）

2008年（5）

2007年（7）

我的朋友
最近访客
推荐博文
相关博文
wxPython和paramiko实现批量管理linux服务器

分类： Python/Ruby

2011-11-25 15:54:48

   

《SSH服务器批量管理工具》使用说明

 

此工具是基于SSH协议对服务器进行批量管理的小工具，支持批量执行命令、批量上传文件、批量获取服务器当前状态信息等功能，无需在服务器端安装任何服务，同时可方便的将执行返回的结果（单行结果或完整结果）导出到文件中保存，使用IP列表文件进行服务器管理，根据服务器名称可方便的进行分类筛选。  

新增密钥支持，可通过密钥连接至服务器进行管理操作。



    

针对“状态信息”查看不太方便，增加表格形式查看服务器基本状态信息功能，点击“状态信息”按钮后可以默认弹出当前勾选的第一台服务器状态窗口，如需要查看其它服务器信息，可通过右键点击服务器所在行弹出表格形式状态窗口（右键点击只对勾选并执行了“状态信息”的服务器生效）。

    


程序主界面截图如下：


 

界面功能说明：


执行命令：用于对服务器列表中勾选的服务器执行指定的命令

发送文件：用于对服务器列表中勾选的服务器发送本地的文件，方便将本地写好的脚本等文件上传至服务器

四个文本输入框分别是：端口、用户名、密码、命令输入框


         以上是服务器列表框，默认加载程序目录下的iplist.txt文件，可通过“重载IP列表”下拉框选择其它IP列表文件


         四个选择按钮不用多解释了，“批量选择”用于对服务器列表中高亮的服务器进行勾选

         关键字：此下拉框默认一个“错误”项，选择错误项的话将去掉所有勾选状态，只勾选服务器列表中“备注”为“错误”的服务器；在服务器列表中任意一行点击鼠标右键，会把所点行的服务器名称以“_”为分隔符拆解成关键字填入关键字下拉框中，从下拉框中选择关键字即可将包含所选关键字的服务器置为勾选状态

         重载IP列表：此下拉框列出程序所在目录下所有扩展名为txt的文件，可以将不同类型的服务器分别放入不同的txt文件中，需要操作的时候从下拉框中选择相应IP列表文件即可加载，IP列表文件的格式如下，IP地址和服务器名称中间可用空格或tab隔开：


         编辑当前IP列表：此按钮打开当前加载的IP列表文件进行编辑，编辑保存后重新在下拉框中选择列表名称重载一下即可

         历史命令：在结果框中列出本次开启程序后执行过的所有历史命令及执行时间

         状态：执行过程中相关信息会显示在“状态：”后面


         完整结果显示框：服务器列表中只显示返回结果的第一行，完整的结果将显示在结果框中，鼠标双击服务器列表中的服务器即可在结果框中显示该服务器的完整结果

         状态信息：用于获取服务器当前的状态信息，结果显示在结果框中，主要包括以下信息：

                   Cpu/内存信息

                   磁盘信息

                   网络设备及流量信息

                   服务器当前运行的服务

                   网络连接信息

                   平均负载、开启时间、连接用户

                   Selinux、io_wait状态

                   占用CPU最多的前10个进程

                   占用内存最多的前10个里程

         导出结果：将服务器列表中的内容导出至文件，可以选择导出单选结果或完整结果，也可以选择导出所有服务器或当前勾选的服务器结果
（最新打包下载）

ssh_tool.part01.rar        ssh_tool.part02.rar        ssh_tool.part03.rar        ssh_tool.part04.rar        ssh_tool.part05.rar

   
==================================================================
     运维工作中经常需要通过ssh连接服务器进行管理，在大批量执行相同的指令时就比较麻烦，一般都需要自己编写脚本来实现，下面是我利用wxPython和paramiko写的批量管理小工具。
    其中paramiko默认没有启用超时设置，参考网上的方法对paramiko的client.py文件做了小小的修改，让其recv的时候支持超时，改动如下（粗体加下划线的部分为增加的代码）：

  1. def exec_command(self, command, bufsize=-1, timeout=None):
  2.         chan = self._transport.open_session()
  3.         if timeout is not None:
  4.             chan.settimeout(timeout)
  5.         chan.exec_command(command)
  6.         stdin = chan.makefile('wb', bufsize)
  7.         stdout = chan.makefile('rb', bufsize)
  8.         stderr = chan.makefile_stderr('rb', bufsize)
  9.         return stdin, stdout, stderr

    附件是打包好的exe程序，win系统下载解压后可以直接使用

  1. # -*- coding: cp936 -*-
  2. # ==========================================================
  3. # $Name: ssh_manage.py
  4. # $Revision: 0.1.1
  5. # $Function: server manager tool
  6. # $Author: LiCaiRong
  7. # $Email: xglcr@163.com
  8. # $Create Date: 2009-12-25 12:30
  9. # $Modify Date: 2009-12-25 12:30
  10. # ==========================================================

  12. import wx,threading,os,glob,paramiko,time
  13. from wx.lib.mixins.listctrl import CheckListCtrlMixin, ListCtrlAutoWidthMixin

  15. TIMEOUT = 5
  16. historycmd = []
  17. textdata = []
  18. lock = threading.RLock()
  19. for x in range(1000):
  20.     textdata.append(x)

  22. class CheckListCtrl(wx.ListCtrl, CheckListCtrlMixin):
  23.     def __init__(self, parent):
  24.         wx.ListCtrl.__init__(self, parent, -1, (15, 50), (890, 360), style=wx.LC_REPORT | wx.LC_HRULES | wx.LC_VRULES)
  25.         CheckListCtrlMixin.__init__(self)

  27. class InsertFrame(wx.Frame):
  28.     def __init__(self, parent, id):
  29.         wx.Frame.__init__(self, parent, id, 'SSH批量连接工具', pos=(100, 50),
  30.                 size=(930, 690), style=wx.DEFAULT_FRAME_STYLE ^ (wx.RESIZE_BORDER | wx.MAXIMIZE_BOX))
  31.         self.SetIcon(wx.Icon("fish.ico", wx.BITMAP_TYPE_ICO))
  32.         font=wx.Font(9,wx.MODERN, wx.NORMAL, wx.NORMAL)
  33.         self.panel = wx.Panel(self)
  34.         self.button1 = wx.Button(self.panel, label="执行命令", pos=(15, 15), size=(80, 25))
  35.         self.porttext = wx.TextCtrl(self.panel, -1, "22", pos=(110,17), size=(50, 21))
  36.         self.usertext = wx.TextCtrl(self.panel, -1, "root", pos=(170,17), size=(50, 21))
  37.         self.pwdtext = wx.TextCtrl(self.panel, -1, "", pos=(230,17), size=(70, 21), style=wx.TE_PASSWORD)
  38.         self.pwdtext.SetFont(font)
  39.         self.cmdtext = wx.TextCtrl(self.panel, -1, "uptime", pos=(310,17), size=(550, 21))
  40.         self.cmdtext.SetFont(font)
  41.         self.list1 = CheckListCtrl(self.panel)
  42.         self.list1.SetBackgroundColour('#B0C4DE')
  43.         sizedata = [25,34,120,120,490,80]
  44.         labelname = ['','id','服务器名称','服务器IP','执行结果','备注']
  45.         for i in range(6):
  46.             if i != 1:
  47.                 self.list1.InsertColumn(i, labelname[i], format=wx.LIST_FORMAT_LEFT, width=int(sizedata[i]))
  48.             else:
  49.                 self.list1.InsertColumn(i, labelname[i], format=wx.LIST_FORMAT_CENTRE, width=int(sizedata[i]))
  50.         button2 = wx.Button(self.panel, label="全选", pos=(15, 420), size=(60, 25))
  51.         button3 = wx.Button(self.panel, label="全清", pos=(80, 420), size=(60, 25))
  52.         button4 = wx.Button(self.panel, label="反选", pos=(145, 420), size=(60, 25))
  53.         button5 = wx.Button(self.panel, label="批量选择", pos=(210, 420), size=(60, 25))
  54.         self.cmbox = wx.ComboBox(self.panel, -1, "关键字", pos=(275, 422), size=(70, 30), choices=["错误"])
  55.         button7 = wx.Button(self.panel, label="配置文件", pos=(350, 420), size=(60, 25))
  56.         ipfile = glob.glob('*.txt')
  57.         self.cmbox2 = wx.ComboBox(self.panel, -1, "重载配置", pos=(415, 422), size=(70, 30), choices=ipfile)
  58.         button9 = wx.Button(self.panel, label="历史命令", pos=(490, 420), size=(60, 25))
  59.         label = wx.StaticText(self.panel, -1, "状态：", pos=(565, 425))
  60.         self.multitext = wx.TextCtrl(self.panel, -1, pos=(15, 455), size=(890, 175), style=wx.TE_MULTILINE)
  61.         self.msg = wx.StaticText(self.panel, -1, label="", pos=(610, 425))
  62.         self.Bind(wx.EVT_BUTTON, self.run, self.button1)
  63.         self.Bind(wx.EVT_LIST_ITEM_ACTIVATED,self.printtext, self.list1)
  64.         self.Bind(wx.EVT_LIST_ITEM_RIGHT_CLICK,self.listadd, self.list1)
  65.         self.Bind(wx.EVT_BUTTON, self.select_all, button2)
  66.         self.Bind(wx.EVT_BUTTON, self.clear_all, button3)
  67.         self.Bind(wx.EVT_BUTTON, self.inverse, button4)
  68.         self.Bind(wx.EVT_BUTTON, self.batch_select, button5)
  69.         self.Bind(wx.EVT_COMBOBOX, self.type, self.cmbox)
  70.         self.Bind(wx.EVT_BUTTON, self.profile, button7)
  71.         self.Bind(wx.EVT_COMBOBOX, self.reload, self.cmbox2)
  72.         self.Bind(wx.EVT_BUTTON, self.history, button9)
  73.         self.multitext.SetFont(font)
  74.         wx.StaticText(self.panel, -1, "此程序为服务器批量管理工具，如果您在使用中发现问题或是有什么建议，请联系作者：xglcr@163.com", pos=(100, 635))
  75.         self.openfile(self,'iplist.txt')

  77.     def run(self,event):
  78.         self.indexip = []
  79.         curselection = []
  80.         self.done_count = []
  81.         self.fail_count = []
  82.         self.port = int(self.porttext.GetValue())
  83.         self.username = self.usertext.GetValue()
  84.         self.password = self.pwdtext.GetValue()
  85.         cmd = self.cmdtext.GetValue()
  86.         num = self.list1.GetItemCount()
  87.         for x in range(num):
  88.             if self.list1.IsChecked(x):
  89.                 tmpdata = "%s %s" % (x,iplist[x].rstrip().split()[0])
  90.                 self.indexip.append(tmpdata)
  91.                 curselection.append(x)
  92.         if (len(self.indexip) == 0):
  93.             self.update_msg("没有选择执行命令的服务器，请选择！")
  94.         elif not self.port or not self.username or not self.password or not cmd:
  95.             self.update_msg("请输入端口、用户名、密码以及命令！")
  96.         else:
  97.             self.button1.Enable(False)
  98.             historycmd.append("%s %s\n" % (time.strftime("%Y-%m-%d %H:%M:%S"),cmd))
  99.             threads=[]
  100.             loops=range(len(self.indexip))
  101.             for i in loops:
  102.                 t = threading.Thread(target=self.sockets,args=(self,self.indexip[i],cmd))
  103.                 threads.append(t)
  104.             for i in loops:
  105.                 threads[i].start()
  106.             self.update_msg("[ 0/%3d]个连接线程已完成, [%3d]个错误" % (len(self.indexip), len(self.fail_count)))
  107.             
  108.     def sockets(self,event,index,data):
  109.         try:
  110.             HOST = index.split()[1]
  111.             idx = index.split()[0]
  112.             self.list1.SetStringItem(int(idx), 5, "正在执行...")
  113.             s = paramiko.SSHClient()
  114.             s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  115.             s.connect(HOST, self.port, username=self.username, password=self.password)
  116.             stdin, stdout , stderr = s.exec_command(data, timeout=TIMEOUT)
  117.             if data.strip().split()[-1] == "&":
  118.                 return_data = "no return values"
  119.             else:
  120.                 return_data = stdout.read()
  121.             dataline = return_data.split('\n')[0]
  122.             self.list1.SetStringItem(int(idx), 4, unicode(dataline,"gbk"))
  123.             self.list1.SetStringItem(int(idx), 5, "完成")
  124.             svrname = unicode(iplist[int(idx)].rstrip().split()[1],'gbk')
  125.             value = unicode("执行结果","gbk")
  126.             memo = unicode("备注：完成","gbk")
  127.             textdata[int(idx)] = "%s %s %s:\n%s\n%s" % (svrname,HOST,value,unicode(return_data,"gbk"),memo)
  128.             lock.acquire()
  129.             self.multitext.SetValue("%s" % textdata[int(idx)])
  130.             lock.release()
  131.             s.close()
  132.             self.done_count.append(HOST)
  133.             self.update_msg("[%3d/%3d]个连接线程已完成, [%3d]个错误" % (len(self.done_count)+len(self.fail_count),len(self.indexip), len(self.fail_count)))
  134.             if len(self.done_count) + len(self.fail_count) == len(self.indexip):
  135.                 self.button1.Enable(True)
  136.         except Exception, e:
  137.             dataline = "%s" % e
  138.             self.list1.SetStringItem(int(idx), 4, dataline)
  139.             self.list1.SetStringItem(int(index.split()[0]), 5, "错误")
  140.             svrname = unicode(iplist[int(idx)].rstrip().split()[1],'gbk')
  141.             value = unicode("执行结果","gbk")
  142.             memo = unicode("备注：错误","gbk")
  143.             textdata[int(idx)] = "%s %s %s:\n%s\n%s" % (svrname,HOST,value,dataline,memo)
  144.             s.close()
  145.             self.fail_count.append(index.split()[0])
  146.             self.update_msg("[%3d/%3d]个连接线程已完成, [%3d]个错误" % (len(self.done_count)+len(self.fail_count),len(self.indexip), len(self.fail_count)))
  147.             if len(self.done_count) + len(self.fail_count) == len(self.indexip):
  148.                 self.button1.Enable(True)

  150.     def update_msg(self, msg):
  151.         self.msg.SetLabel(msg)
  152.         
  153.     def printtext(self,event):
  154.         cur = self.list1.GetFirstSelected()
  155.         self.multitext.SetValue("%s" % textdata[int(cur)])

  157.     def reload(self,event):
  158.         IPLIST = self.cmbox2.GetValue()
  159.         self.openfile(self,IPLIST)

  161.     def openfile(self,event,IP):
  162.         num = self.list1.GetItemCount()
  163.         if (num != 0):
  164.             self.list1.DeleteAllItems()
  165.         f = open(IP,'r')
  166.         global iplist
  167.         iplist = []
  168.         row = 0
  169.         for line in f:
  170.             iplist.append(line)
  171.             ip,notes = line.split()
  172.             self.list1.InsertStringItem(row,'')
  173.             self.list1.SetStringItem(row, 1, str(row))
  174.             self.list1.SetStringItem(row, 2, notes)
  175.             self.list1.SetStringItem(row, 3, ip)
  176.             row += 1
  177.         f.close()
  178.     
  179.     def listadd(self,event):
  180.         self.cmbox.Clear()
  181.         self.cmbox.Append("错误")
  182.         cur = self.list1.GetFirstSelected()
  183.         notes = unicode(iplist[cur].rstrip().split()[1],'gbk')
  184.         for i in range(len(notes.split('_'))):
  185.             self.cmbox.Append(notes.split('_')[i])
  186.             self.cmbox.Append(notes.split('_')[i][0:-1])
  187.             self.cmbox.Append(notes.split('_')[i][0:-2])
  188.     
  189.     def type(self,event):
  190.         selection = []
  191.         match = self.cmbox.GetValue()
  192.         num = self.list1.GetItemCount()
  193.         if match == u"错误":
  194.             self.clear_all(self)
  195.             for i in self.fail_count:
  196.                 self.list1.CheckItem(int(i))
  197.                 selection.append(int(i))
  198.         else:
  199.             for i in range(num):
  200.                 tmp = (unicode(iplist[i],"gbk").rstrip().split()[1].lower()).find(match.lower())
  201.                 if (tmp != -1):
  202.                     self.list1.CheckItem(i)
  203.                     selection.append(i)
  204.         self.update_msg("已勾选符合条件的服务器[ %s ]台！" % len(selection))
  205.     
  206.     def select_all(self,event):
  207.         num = self.list1.GetItemCount()
  208.         for i in range(num):
  209.             self.list1.CheckItem(i)
  210.         self.update_msg("当前已选择服务器[ %s ]台！" % num)

  212.     def clear_all(self,event):
  213.         num = self.list1.GetItemCount()
  214.         for i in range(num):
  215.             self.list1.CheckItem(i,False)
  216.         self.update_msg("当前未选择任何服务器！")

  218.     def inverse(self,event):
  219.         selection = []
  220.         num = self.list1.GetItemCount()
  221.         for i in range(num):
  222.             if self.list1.IsChecked(i):
  223.                 self.list1.CheckItem(i,False)
  224.             else:
  225.                 self.list1.CheckItem(i)
  226.                 selection.append(i)
  227.         self.update_msg("当前已选择服务器[ %s ]台！" % len(selection))

  229.     def batch_select(self,event):
  230.         selection = []
  231.         i = self.list1.GetFirstSelected()
  232.         while i != -1:
  233.             self.list1.CheckItem(i)
  234.             i = self.list1.GetNextSelected(i)
  235.             selection.append(i)
  236.         self.update_msg("已勾选高亮的服务器[ %s ]台！" % len(selection))

  238.     def profile(self,event):
  239.         wx.Execute('notepad admin.ini')

  241.     def history(self,event):
  242.         self.multitext.Clear()
  243.         for i in historycmd:
  244.             self.multitext.AppendText(i)
  245.         self.update_msg("总共[ %s ]条历史命令！" % len(historycmd))


  248. if __name__ == '__main__':
  249.     app = wx.PySimpleApp()
  250.     frame = InsertFrame(parent=None, id=-1)
  251.     frame.Show()
  252.     app.MainLoop()


阅读(6823) | 评论(7) | 转发(1) |
1

上一篇：Ubuntu 字体设置

下一篇：调整错误级别，关闭MySQLdb的恼人的Warning信息

给主人留下些什么吧！~~

wolf57292012-05-05 11:04:34

博主你好!
我在博文 http://www.danoking.com/2012/05/05/开源节流之批量远程关闭linux-方案篇/ 提到了你的此篇文章，特此相告。
如果侵犯了你，请知会我，我立刻进行处理。

感谢博主的分享。
我也经常把玩python，略玩过wxpython，经常使用linux。

回复 | 举报

puluto2011-11-25 16:16:31

支持分享..

回复 | 举报
关于我们 | 关于IT168 | 联系方式 | 广告合作 | 法律声明 | 免费注册

Copyright 2001-2010 ChinaUnix.net All Rights Reserved 北京皓辰网域网络信息技术有限公司. 版权所有

感谢所有关心和支持过ChinaUnix的朋友们

16024965号-6