前几天某站2个小时无法访问，最后终于触发了XX业务部门的人工报警，然后就悲剧了，领导大怒，问：怎么回事？答：公司站点太多，有些小站点访问访问量比较小也不重要，关注度下降，所以。。。，讨论最后领导下指示要尽快加所有站点的监控，不管站点大小，如果不可访问立刻要进行报警，绝对不能再次触发人工报警，不能让这种不专业的事再次发生，所以就有了今天的这个脚本，脚本如下：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 |
#!/usr/bin/env python import pickle, os, sys, logging from httplib import HTTPConnection, socket from smtplib import SMTP def email_alert(message, status): fromaddr = 'abc@163.com' toaddrs = 'def@abc.com' server = SMTP('smtp.163.com:25') server.starttls() server.login('xxxxx', 'xxxx') server.sendmail(fromaddr, toaddrs, 'Subject: %s\r\n%s' % (status, message)) server.quit() def get_site_status(url): response = get_response(url) try: if getattr(response, 'status') == 200: return 'up' except AttributeError: pass return 'down' def get_response(url): try: conn = HTTPConnection(url) conn.request('HEAD', '/') return conn.getresponse() except socket.error: return None except: logging.error('Bad URL:', url) exit(1) def get_headers(url): response = get_response(url) try: return getattr(response, 'getheaders')() except AttributeError: return 'Headers unavailable' def compare_site_status(prev_results): def is_status_changed(url): status = get_site_status(url) friendly_status = '%s is %s' % (url, status) print friendly_status if url in prev_results and prev_results[url] != status: logging.warning(status) email_alert(str(get_headers(url)), friendly_status) prev_results[url] = status return is_status_changed def is_internet_reachable(): if get_site_status('www.baidu.com') == 'down' and get_site_status('www.qq.com') == 'down': return False return True def load_old_results(file_path): pickledata = {} if os.path.isfile(file_path): picklefile = open(file_path, 'rb') pickledata = pickle.load(picklefile) picklefile.close() return pickledata def store_results(file_path, data): output = open(file_path, 'wb') pickle.dump(data, output) output.close() def main(urls): logging.basicConfig(level=logging.WARNING, filename='checksites.log', format='%(asctime)s %(levelname)s: %(message)s', datefmt='%Y-%m-%d %H:%M:%S') pickle_file = 'data.pkl' pickledata = load_old_results(pickle_file) print pickledata if is_internet_reachable(): status_checker = compare_site_status(pickledata) map(status_checker, urls) else: logging.error('Either the world ended or we are not connected to the net.') store_results(pickle_file, pickledata) if __name__ == '__main__': main(sys.argv[1:]) |
脚本逻辑就是获取站点HEAD，如果正常就返回up，有问题就返回down,getattr()是python内置函数，接收一个对象，可以根据对象的属性返回对象的值，比较有意思的是compare_site_status()函数中又定义了一个函数，最后return了这个内部函数，这个函数在map()函数中进行了调用，map()函数接收2个参数，一个是函数，一个系列，在这个脚本里就是检测每个url的站点状态，如果有问题就发邮件报警，检测过的站点状态用pickle模块存在data.pkl文件中，在检测站点前先做网络连通性检测，这里定义的is_internet_reachable()函数实现比较简单，判断是否网络故障就是访问下baidu和qq,如果这2个站点都不能访问，就表示网络出问题了，不一定是站点出现问题，如果网络正常，然后才进行站点状态验证。