Python實現的檢測web服務器健康狀況的小程序

#!/usr/bin/env python
#coding=utf8
 
"""
使用socket方式來檢查服務器的監控狀況
"""
 
from optparse import OptionParser
 
import socket
 
import sys
 
import re
 
from StringIO import StringIO
 
class check_server:
 """
 該類主要是利用socket建立一個連接以后，發送一個http請求，然后根據返回的狀態碼，判斷主機的健康狀況
 """
 def __init__(self,address,port,resource):
 self.address = address
 self.port = port
 self.resource = resource
 
 
 def check(self):
 """
 該方法也是該類的主要方法，包括構建請求資源，解析返回結果等
 """
 if not self.resource.startswith('/'):
 self.resource = '/' + self.resource
 
 request = "GET %s HTTP/1.1
Host:%s

" %(self.resource,self.address)
 
 #建立一個socket連接
 
 s = socket.socket()
 #設置連接超時時間
 s.settimeout(10)
 
 print "現在開始對 %s 上的 %s 端口連接......" %(self.address,self.port)
 
 try:
 s.connect((self.address,self.port))
 print "連接 %s 上端口 %s 成功" %(self.address,self.port)
 s.send(request)
 response = s.recv(100)
 
 except socket.error,e:
 print "連接%s 上端口 %s 失敗 ,原因為:%s" %(self.address,self.port,e)
 return False
 finally:
 print "關閉連接"
 s.close()
 
 
 line = StringIO(response).readline()
 
 try:
 (http_version,status,messages) = re.split(r's+',line,2)
 except ValueError:
 print "分割響應碼失敗"
 return False
 print "返回的狀態碼是%s" %(status)
 
 if status in ['200','301','302']:
 
 print "服務器的監控狀況良好"
 else:
 
 print "乖乖，趕快上線看看，咋回事"
 
 
if __name__ == '__main__':
 """
 處理參數
 """
 parser =OptionParser()
 parser.add_option("-a","--address",dest="address" ,default='localhost',help="要檢查主機的地址或者主機名")
 parser.add_option('-p','--port',dest="port",type=int,default=80,help="要檢查主機的端口")
 parser.add_option('-r','--resource',dest="resource",default="/",help="要檢查的資源，比如")
 (options,args) = parser.parse_args()
 
#開始檢測鳥
checks = check_server(options.address,options.port,options.resource)
 
checks.check()