測試、預發(fā)布后用python檢測網(wǎng)頁是否有日常鏈接

更新時間：2014年06月03日 09:02:32 作者：

難免會碰到秀逗了把測試的鏈接發(fā)布到線上的情況，一般這種都是通過一些測試的檢查工具來檢查鏈接來規(guī)避風險的，下面為大家簡述下大概的實現(xiàn)思路

在大的互聯(lián)網(wǎng)公司干技術的基本都會碰到測試、預發(fā)布、線上這種多套環(huán)境的，來實現(xiàn)測試和線上正式環(huán)境的隔離，這種情況下，就難免會碰到秀逗了把測試的鏈接發(fā)布到線上的情況，一般這種都是通過一些測試的檢查工具來檢查鏈接來規(guī)避風險的。前兩天跟了一個問題也是這個情況，開發(fā)疏忽把日常url發(fā)布到線上了。但是測試那邊沒有自動化的監(jiān)控工具，導致沒有及時發(fā)現(xiàn)，由于最近正好在看python，后來處理完回家就想用python做個簡單的監(jiān)控。

大略思路是：用python寫一個腳本來分析網(wǎng)頁里的所有url，看是否包含日常的鏈接，然后把腳本放到crontab里跑定時任務，10分鐘跑一次檢查。如果發(fā)現(xiàn)非法鏈接，就發(fā)告警郵件給相關人員。腳本代碼100行左右，比較好理解，貼上代碼。

本來想用beautifulsoup，不過考慮到安裝三方庫麻煩，所以還是用自帶的sgmllib來，不需要關心庫。發(fā)郵件函數(shù)沒有實現(xiàn)，根據(jù)各自的smtp服務器實現(xiàn)以下即可。

復制代碼代碼如下:

 
#!/usr/bin/env python 
#coding:UTF-8 

import urllib2 
from sgmllib import SGMLParser 
import smtplib 
import time 
#from email.mime.text import MIMEText 
#from bs4 import BeautifulSoup 
#import re 

class UrlParser(SGMLParser): 
urls = [] 
def do_a(self,attrs): 
'''''parse tag a''' 
for name,value in attrs: 
if name=='href': 
self.urls.append(value) 
else: 
continue 

def do_link(self,attrs): 
'''''parse tag link''' 
for name,value in attrs: 
if name=='href': 
self.urls.append(value); 
else: 
continue 

def checkUrl(checkurl, isDetail): 
'''''檢查checkurl對應的網(wǎng)頁源碼是否有非法url''' 
parser = UrlParser() 
page = urllib2.urlopen(checkurl) 
content = page.read() 
#content = unicode(content, "gb2312").encode("utf8") 
parser.feed(content) 
urls = parser.urls 

dailyUrls = [] 
detailUrl = "" 
for url in urls: 
if 'daily' in url: 
dailyUrls.append(url); 
if not detailUrl and not isDetail and 'www.bc5u.com' in url: 
detailUrl = url 

page.close() 
parser.close() 

if isDetail: 
return dailyUrls 
else: 
return dailyUrls,detailUrl 

def sendMail(): 
'''''發(fā)送提醒郵件''' 
pass 

def log(content): 
'''''記錄執(zhí)行日志''' 
logFile = 'checkdailyurl.log' 
f = open(logFile,'a') 
f.write(str(time.strftime("%Y-%m-%d %X",time.localtime()))+content+'\n') 
f.flush() 
f.close() 

def main(): 
'''''入口方法''' 
#檢查ju 
url = "www.bc5u.com" 

dailyUrls,detailUrl=checkUrl(url, False) 
if dailyUrls: 
#檢查到daily鏈接，發(fā)送告警郵件 
sendMail() 
log('check: find daily url') 
else: 
#沒檢查到daily鏈接，不處理 
log('check: not find daily url') 

#檢查judetail 
dailyUrls=checkUrl(detailUrl, True) 
if dailyUrls: 
#檢查到daily鏈接，發(fā)送告警郵件 
log('check: find daily url') 
sendMail() 
else: 
#沒檢查到daily鏈接，不處理 
log('check: not find daily url') 

if __name__ == '__main__': 
main() 