发博文
公告：【5月博客评选】欢迎提交本月原创作品
吃鱼翅的鱼

xglcr.blog.chinaunix.net

   
首页 | 博文目录 | 相册 | 博客圈 | 关于我 | 留言
个人资料
xglcr
微博 论坛
发纸条 打招呼 加关注 加好友
  • 博客访问：53523
  • 博文数量：36
  • 博客积分：2124
  • 博客等级：大尉
  • 注册时间：2007-03-21 10:30:14
订阅我的博客
  • 订阅
  • 订阅到鲜果
  • 订阅到抓虾
  • 订阅到Google
字体大小：大 中 小博文
wxPython和paramiko实现批量管理linux服务器 (2011-11-25 15:54)
标签:  服务器  linux  分类： python


    运维工作中经常需要通过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, "此程序为服务器批量管理工具，如果您在使用中发现问题或是有什么建议，请联系作者：李才荣 Email：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()
 ssh批量管理.part01.rar    ssh批量管理.part02.rar    ssh批量管理.part03.rar  
 ssh批量管理.part04.rar  


0
   顶
热门产品推荐
阅读(315)┊ 评论 (3)┊收藏(1)┊举报┊打印
前一篇：博客已升级，请注意变更地址
[发评论] 评论 重要提示：警惕虚假中奖信息!
  • 追忆的风筝 2012-05-06 19:25
    师傅，猜猜我是谁？
  • wolf5729 2012-05-05 11:04
    博主你好!
    我在博文 http://www.danoking.com/2012/05/05/开源节流之批量远程关闭linux-方案篇/ 提到了你的此篇文章，特此相告。
    如果侵犯了你，请知会我，我立刻进行处理。

    感谢博主的分享。
    我也经常把玩python，略玩过wxpython，经常使用linux。
  • puluto 2011-11-25 16:16
    支持分享..
亲，您还没有登录,请[登录]或[注册]后再进行评论