#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2016/10/18 9:50
# @Author  : Beam
# @Site    : 實(shí)現(xiàn)探測(cè)Web服務(wù)質(zhì)量
# @File    : demo_pycurl.py
# @Software: PyCharm
import pycurl
import os,sys,time
def getInfo(URL):
    """
    :param URL:用戶(hù)輸入需要檢測(cè)的URL地址
    info 字典主要用于映射dic字典
    dic字典主要存下curl結(jié)果
    :return:沒(méi)return卜范，直接print沽瞭，函數(shù)可以改寫(xiě)，可以用于定時(shí)檢測(cè)多個(gè)域名辰狡，增加一個(gè)需要檢測(cè)的url列表即可
    """
    c = pycurl.Curl()
    c.setopt(pycurl.URL,URL) #定義請(qǐng)求的URL常量
    c.setopt(pycurl.CONNECTTIMEOUT,5) #請(qǐng)求等待時(shí)間最多5秒
    c.setopt(pycurl.TIMEOUT,5)   #定義請(qǐng)求超時(shí)時(shí)間（服務(wù)器沒(méi)回應(yīng)）
    c.setopt(pycurl.NOPROGRESS,1) #屏蔽下載進(jìn)度條
    c.setopt(pycurl.FORBID_REUSE,1) #交互完成后強(qiáng)制斷開(kāi)連接，不重用
    c.setopt(pycurl.MAXREDIRS,1)  #指定HTTP重定向的最大數(shù)為1
    c.setopt(pycurl.DNS_CACHE_TIMEOUT,30) #設(shè)置DNS信息保存時(shí)間為30秒
    c.setopt(pycurl.USERAGENT,"Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/53.0.2785.116 Safari/537.36 OPR/40.0.2308.81 (Edition Baidu)")
    dic = {}
    info ={'NAMELOOKUP_TIME':'DNS解析時(shí)間','CONNECT_TIME':'建立連接時(shí)間','PRETRANSFER_TIME':'建立到準(zhǔn)備傳輸所消耗的時(shí)間','STARTTRANSFER_TIME':'建立連接到傳輸開(kāi)始消耗的時(shí)間','TOTAL_TIME':'傳輸總時(shí)間',
           'HTTP_CODE':'HTTP狀態(tài)碼','SIZE_DOWNLOAD':'下載數(shù)據(jù)包大小','HEADER_SIZE':'HTTP頭部大小','SPEED_DOWNLOAD':'平均下載速度'}
    with open(os.path.dirname(os.path.realpath(__file__))+"/content.txt","wb") as indexfile:
        c.setopt(pycurl.WRITEHEADER,indexfile)   #將返回的HTTP HEADER定向到indexfile文件對(duì)象
        c.setopt(pycurl.WRITEDATA,indexfile)     #將返回的HTML內(nèi)容定向到indexfile文件對(duì)象
        try:
            c.perform()
        except Exception,e:
            print "Connection error:" +str(e)
            c.close()
            sys.exit()
        dic['NAMELOOKUP_TIME'] = '%.2f ms' % (c.getinfo(c.NAMELOOKUP_TIME)*1000)     #獲取DNS解析時(shí)間
        dic['CONNECT_TIME'] = '%.2f ms' % (c.getinfo(c.CONNECT_TIME)*1000)          #獲取建立連接時(shí)間
        dic['PRETRANSFER_TIME'] = '%.2f ms' % (c.getinfo(c.PRETRANSFER_TIME)*1000)  #獲取從建立到準(zhǔn)備傳輸所消耗的時(shí)間
        dic['STARTTRANSFER_TIME'] = '%.2f ms' % (c.getinfo(c.STARTTRANSFER_TIME)*1000)  #獲取從建立連接到傳輸開(kāi)始消耗的時(shí)間
        dic['TOTAL_TIME'] = '%.2f ms' % (c.getinfo(c.TOTAL_TIME)*1000)              #獲取傳輸總時(shí)間
        dic['HTTP_CODE'] = c.getinfo(c.HTTP_CODE)                #獲取HTTP狀態(tài)碼
        dic['SIZE_DOWNLOAD'] = '%d bytes/s' % (c.getinfo(c.SIZE_DOWNLOAD))        #獲取下載數(shù)據(jù)包大小
        dic['HEADER_SIZE'] = '%d bytes/s' % (c.getinfo(c.HEADER_SIZE))            #獲取HTTP頭部大小
        dic['SPEED_DOWNLOAD'] = '%d bytes/s' % (c.getinfo(c.SPEED_DOWNLOAD))      #獲取平均下載速度
    for key in info:
        print info[key],':',dic[key]
def main():
    while True:
        URL = raw_input("請(qǐng)輸入一個(gè)URL地址(Q for exit)：")
        if URL.lower() == 'q':
            sys.exit()
        else:
            getInfo(URL)
if __name__ == '__main__':
    main()