Python查看多臺(tái)服務(wù)器進(jìn)程的腳本分享

更新時(shí)間：2014年06月11日 09:03:03 作者：

這篇文章主要介紹了Python查看多臺(tái)服務(wù)器進(jìn)程的腳本分享,需要的朋友可以參考下

最近做自己開(kāi)發(fā)用相關(guān)服務(wù)的一個(gè)checklist，就寫(xiě)了這個(gè)腳本，用來(lái)在跳板機(jī)去檢查各個(gè)服務(wù)器上面的相關(guān)服務(wù)是否正常

使用expect登錄每個(gè)機(jī)器(因?yàn)榘踩珕?wèn)題，不能直接使用ssh信任),然后根據(jù)yaml文件的配置讀取服務(wù)名字以及啟動(dòng)的進(jìn)程數(shù)量去檢查每個(gè)服務(wù)是否正常 PS：難點(diǎn)是沒(méi)有用端口轉(zhuǎn)發(fā)也只有普通用戶權(quán)限

checklist.py

復(fù)制代碼代碼如下:

#coding=utf-8
import sys
#因?yàn)槲疫@個(gè)腳本要讓很多人能運(yùn)行，但是不能給他們看見(jiàn)我的密碼算法,所以是pyc
#我這個(gè)腳本要給很多其他普通用戶去用，是用我的ssh登錄操作，不能放在我的home目錄，所以放在tmp
sys.path.append('/tmp/local/lib/python2.6/site-packages/PyYAML-3.10-py2.6-linux-x86_64.egg') #依賴yaml
sys.path.append('/tmp/local/lib/python2.6/site-packages/pexpect-2.4-py2.6.egg') #依賴pexpect
import yaml
import pexpect
dataDict = yaml.load(open('/tmp/config.yaml')) #將我的yaml配置load進(jìn)來(lái)

def myprint(color,mes): #以前寫(xiě)的一個(gè)終端彩色打印的函數(shù)
    '''使用ANSI控制碼終端顯示彩色'''
    d = dict(r=31, g=32, gb=36, y=33, b=34, p=35, o=37)
    color = "\x1B[%d;%dm" % (1, d[color])
    print "%s%s\x1B[0m" % (color, mes)

def main():
    list = ['g', 'b', 'y', 'gb', 'p']
    light = 0
    for k in dataDict:
        if k.startswith('bj-'):
        color = list[light%5] #根據(jù)服務(wù)器對(duì)顏色輪循
            SERVER = dataDict[k]
        #我這是使用了-F 是因?yàn)槲覜](méi)有root權(quán)限不能修改hosts文件，但是我在config.yaml使用了別名，
        而這個(gè)定義就是自定義了sshconfig，默認(rèn)是~/.ssh/config
        child = pexpect.spawn('ssh -F /tmp/sshconfig dongwm@{0}'.format(SERVER['host']))
        #因?yàn)橛衅渌脩?，可能他還沒(méi)有鏈接過(guò)某服務(wù)器，最開(kāi)始會(huì)讓你確認(rèn)服務(wù)器標(biāo)識(shí)，需要點(diǎn)yes
        f = child.expect(['Password: ', 'password: ', 'continue connecting (yes/no)?'])
        if f == 2:
            #當(dāng)這個(gè)flag為2 表示那個(gè)用戶沒(méi)有登錄過(guò)某服務(wù)器
            child.sendline('yes')
            child.expect('password:')
            child.sendline('{0}'.format(mypasswd(SERVER['host']))) #mypasswd是加密我服務(wù)器權(quán)限的函數(shù)，每個(gè)服務(wù)器密碼不同
        if f == 1:
            child.sendline('{0}'.format(mypasswd(SERVER['host'])))
        child.expect('~')
        for service in SERVER['service']:
        flag = 0
        #我在配置里面會(huì)加服務(wù),一般會(huì)指定服務(wù)的進(jìn)程數(shù)來(lái)對(duì)比是否正常
        if isinstance(service, dict):
            data =service.items()[0]
            service = data[0]
            num = data[1]
        else:
        #假如我在配置只指定服務(wù)，不指定進(jìn)程數(shù)，那么只要確定跑了進(jìn)程不在乎進(jìn)程數(shù)
            num = 0
            flag = 1
            child.expect('~')
            child.sendline('ps -ef|grep {0}|grep -v grep|wc -l'.format(
            service))
            child.readline()
            #進(jìn)程數(shù)
            pro_num = child.readline().split('\r\n')[0]
        if int(pro_num) == num or flag:
            #進(jìn)程數(shù)符合配置標(biāo)注的數(shù)值
            myprint(color, '[{0}] [{1}] [{2}] [{3}]'.format(k.center(12),
            SERVER['ip'].center(14), service.center(20), 'ok'.center(4)))
        else:
            myprint('r', '[{0}] [{1}] [{2}] [{3}] [{4}!={5}]'.format(k.center(12),
            SERVER['ip'].center(14), service.center(20), 'fail',
            pro_num, num))
        light += 1
            child.sendline('exit')

if __name__ == '__main__':
main()

config.yaml 我這里只截取了其中一段

復(fù)制代碼代碼如下:

bj-2:
  host: s233 #這個(gè)s233在sshconfig指定
  ip: XXX.XXX.XXX.233 #只是為了顯示出ip 好確認(rèn)
  service: #服務(wù)load后是一個(gè)列表
  #給XX用
  - nginx: 5
  - uwsgi: 25
  - supervisord: 1
  #給本機(jī)XX提供mysql服務(wù)
  - mysql: 3 #django
  #給本機(jī)XX提供XX
  - celery: 12 
  #給本機(jī)XX提供XX
  - rabbitmq: 9
  - redis: 1
  - mongod: 2