#!/usr/bin/env python
#coding=utf8
 
"""
使用socket方式來(lái)檢查服務(wù)器的監(jiān)控狀況
"""
 
from optparse import OptionParser
 
import socket
 
import sys
 
import re
 
from StringIO import StringIO
 
class check_server:
  """
  該類主要是利用socket建立一個(gè)連接以后，發(fā)送一個(gè)http請(qǐng)求，然后根據(jù)返回的狀態(tài)碼，判斷主機(jī)的健康狀況
  """
  def __init__(self,address,port,resource):
    self.address = address
    self.port = port
    self.resource = resource
 
 
  def check(self):
    """
    該方法也是該類的主要方法，包括構(gòu)建請(qǐng)求資源，解析返回結(jié)果等
    """
    if not self.resource.startswith('/'):
      self.resource = '/' + self.resource
 
    request = "GET %s HTTP/1.1\r\nHost:%s\r\n\r\n" %(self.resource,self.address)
 
    #建立一個(gè)socket連接
 
    s = socket.socket()
    #設(shè)置連接超時(shí)時(shí)間
    s.settimeout(10)
 
    print "現(xiàn)在開始對(duì) %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 失敗 ,原因?yàn)?%s" %(self.address,self.port,e)
      return False
    finally:
      print "關(guān)閉連接"
      s.close()
 
 
    line = StringIO(response).readline()
 
    try:
      (http_version,status,messages) = re.split(r'\s+',line,2)
    except ValueError:
      print "分割響應(yīng)碼失敗"
      return False
    print "返回的狀態(tài)碼是%s" %(status)
 
    if status in ['200','301','302']:
 
      print "服務(wù)器的監(jiān)控狀況良好"
    else:
 
      print "乖乖，趕快上線看看，咋回事"
 
 
if __name__ == '__main__':
  """
  處理參數(shù)
  """
  parser =OptionParser()
  parser.add_option("-a","--address",dest="address" ,default='localhost',help="要檢查主機(jī)的地址或者主機(jī)名")
  parser.add_option('-p','--port',dest="port",type=int,default=80,help="要檢查主機(jī)的端口")
  parser.add_option('-r','--resource',dest="resource",default="/",help="要檢查的資源，比如")
  (options,args) = parser.parse_args()
 
#開始檢測(cè)鳥
checks = check_server(options.address,options.port,options.resource)
 
checks.check()