python获取接口响应时间_Python使用pycurl获取http的响应时间

最近需要对节点到源站自己做个监控,简单的ping可以检测到一些东西,但是http请求的检查也要进行,于是就研究了下pycurl

pycurl是个用c语言实现的python 库,虽然据说不是那么pythonic,但是却很高效,它支持的协议居多:

supporting FTP, FTPS, HTTP, HTTPS, GOPHER, TELNET, DICT, FILE and LDAP. libcurl supports HTTPS certificates, HTTP POST, HTTP PUT, FTP uploading, kerberos, HTTP form based upload, proxies, cookies, user+password authentication, file transfer resume, http proxy tunneling and more!

这一堆协议已经很多了,我需要就是http一个,相对urlib来说,这个库可能更快些。

以下这个脚本是对某一个给定的url进行检查,并打印出http相应码,响应大小,建立连接时间,准备传输时间,传输第一个字节时间,完成时间

#!/usr/bin/python

# coding: UTF-8

import StringIO

import pycurl

import sys

import os

class Test:

def __init__(self):

self.contents = ''

def body_callback(self,buf):

self.contents = self.contents + buf

def test_gzip(input_url):

t = Test()

#gzip_test = file("gzip_test.txt", 'w')

c = pycurl.Curl()

c.setopt(pycurl.WRITEFUNCTION,t.body_callback)

c.setopt(pycurl.ENCODING, 'gzip')

c.setopt(pycurl.URL,input_url)

c.perform()

http_code = c.getinfo(pycurl.HTTP_CODE)

http_conn_time =  c.getinfo(pycurl.CONNECT_TIME)

http_pre_tran =  c.getinfo(pycurl.PRETRANSFER_TIME)

http_start_tran =  c.getinfo(pycurl.STARTTRANSFER_TIME)

http_total_time = c.getinfo(pycurl.TOTAL_TIME)

http_size = c.getinfo(pycurl.SIZE_DOWNLOAD)

print 'http_code http_size conn_time pre_tran start_tran total_time'

print "%d %d %f %f %f %f"%(http_code,http_size,http_conn_time,http_pre_tran,http_start_tran,http_total_time)

if __name__ == '__main__':

input_url = sys.argv[1]

test_gzip(input_url)

脚本运行效果

xu:~/curl$ python pycurl_test.py http://daxuxu.info/

http_code http_size conn_time pre_tran start_tran total_time

200 8703 0.748147 0.748170 1.632642 1.636552

pycurl 的一些响应信息:

(参考: http://curl.haxx.se/libcurl/c/curl_easy_getinfo.html )

pycurl.NAMELOOKUP_TIME 域名解析时间

pycurl.CONNECT_TIME 远程服务器连接时间

pycurl.PRETRANSFER_TIME 连接上后到开始传输时的时间

pycurl.STARTTRANSFER_TIME 接收到第一个字节的时间

pycurl.TOTAL_TIME 上一请求总的时间

pycurl.REDIRECT_TIME 如果存在转向的话,花费的时间

pycurl.EFFECTIVE_URL

pycurl.HTTP_CODE HTTP 响应代码

pycurl.REDIRECT_COUNT 重定向的次数

pycurl.SIZE_UPLOAD 上传的数据大小

pycurl.SIZE_DOWNLOAD 下载的数据大小

pycurl.SPEED_UPLOAD 上传速度

pycurl.HEADER_SIZE 头部大小

pycurl.REQUEST_SIZE 请求大小

pycurl.CONTENT_LENGTH_DOWNLOAD 下载内容长度

pycurl.CONTENT_LENGTH_UPLOAD 上传内容长度

pycurl.CONTENT_TYPE 内容的类型

pycurl.RESPONSE_CODE 响应代码

pycurl.SPEED_DOWNLOAD 下载速度

pycurl.SSL_VERIFYRESULT

pycurl.INFO_FILETIME 文件的时间信息

pycurl.HTTP_CONNECTCODE HTTP 连接代码

pycurl.HTTPAUTH_AVAIL

pycurl.PROXYAUTH_AVAIL

pycurl.OS_ERRNO

pycurl.NUM_CONNECTS

pycurl.SSL_ENGINES

pycurl.INFO_COOKIELIST

pycurl.LASTSOCKET

pycurl.FTP_ENTRY_PATH

########################################################

前几天看完《Python简明教程》,预想练手,想起同事的一个 ruby 代码,尝试改写成 python,顺便看看两个语言的简练程度。下面是原始的 ruby 代码:

#!/usr/bin/env ruby

require 'rubygems'

require 'net/http'

urls = ["http://icyleaf.com"]

.times do

urls.each do |url|

start_at = Time.now

Net::HTTP.get URI.parse(url)

end_at = Time.now

diff = end_at - start_at

if diff < . then

color_code =

elsif diff > . then

color_code =

else

color_code =

end

puts "#{url}\n time: \033[#{color_code}m#{diff}\033[0m seconds"

end

end

改写 python 的同时,考虑脚本的灵活性准备增加两个参数,第一个是请求测试次数,第二个是请求测试的 URL,而 python 默认提供了argparse 库,可以很方便的生成 --help 的帮助和解析传递的参数:

#!/usr/bin/env python

import urllib2

import time

import sys

import argparse

def benchmark(url, count):

for i in range(count):

s = time.time()

r = urllib2.urlopen(urllib2.Request(url))

e = time.time()

diff = e - s

if diff < 0.3:

color_code =

elif diff > 0.8:

color_code =

else:

color_code =

print '# %d' % (i + )

print '\tStauts: %s' % r.getcode()

print '\tTime: \033[%dm%f\033[0m second(s)' % (color_code, diff)

def main(argv):

parser = argparse.ArgumentParser(description='url request time test')

parser.add_argument('URL', help='request url')

parser.add_argument('-t', '--time', action='store', dest='count', type=int, default=, help='request times')

args = parser.parse_args(argv)

benchmark(args.URL, args.count)

if __name__ == '__main__':

main(sys.argv[:])

当然,我主要是为了练手 python 才去写的,ruby 本身也有 optparse 库用于解析参数,但是需要自己手写生成 --help 的输出,而且需要对每个参数做相应的 callback。

效果如下:

Posted by icyleaf on 2012-08-02

#########################################################

用Python编一个抓网页的程序是非常快的,下面就是一个例子:

importurllib2

html=urllib2.urlopen('http://www.server110.com').read()

但是在实际工作中,这种写法是远远不够的,至少会遇到下面几个问题:

网络会出错,任何错误都可能。例如机器宕了,网线断了,域名出错了,网络超时了,页面没有了,网站跳转了,服务被禁了,主机负载不够了…

服务器加上了限制,只让常见浏览器访问

服务器加上了防盗链的限制

某些2B网站不管你HTTP请求里有没有Accept-Encoding头部,也不管你头部具体内容是什么,反正总给你发gzip后的内容

URL链接千奇百怪,带汉字的也罢了,有的甚至还有回车换行

某些网站HTTP头部里有一个Content-Type,网页里有好几个Content-Type,更过分的是,各个Content-Type还不一样,最过分的是,这些Content-Type可能都不是正文里使用的Content-Type,从而导致乱码

网络链接很慢,乘分析几千个页面的时间,建议你可以好好吃顿饭去了

Python本身的接口有点糙

好吧,这么一大箩筐问题,我们来一个个搞定。

错误处理和服务器限制

首先是错误处理。由于urlopen本身将大部分错误,甚至包括4XX和5XX的HTTP响应,也搞成了异常,因此我们只需要捕捉异常就好了。同时,我们也可以获取urlopen返回的响应对象,读取它的HTTP状态码。除此之外,我们在urlopen的时候,也需要设置timeout参数,以保证处理好超时。下面是代码示例:

importurllib2

importsocket

try:

f=urllib2.urlopen('http://www.server110.com',timeout=10)

code=f.getcode()

ifcode<200orcode>=300:

#你自己的HTTP错误处理

exceptException,e:

ifisinstance(e,urllib2.HTTPError):

print'http error: {0}'.format(e.code)

elifisinstance(e,urllib2.URLError)andisinstance(e.reason,socket.timeout):

print'url error: socket timeout {0}'.format(e.__str__())

else:

print'misc error: '+e.__str__()

如果是服务器的限制,一般来说我们都可以通过查看真实浏览器的请求来设置对应的HTTP头部,例如针对浏览器的限制我们可以设置User-Agent头部,针对防盗链限制,我们可以设置Referer头部,下面是示例代码:

importurllib2

req=urllib2.Request('http://www.server110.com',

headers={"Referer":"http://www.baidu.com",

"User-Agent":"Mozilla/5.0 (Windows NT 5.1) AppleWebKit/534.24 (KHTML, like Gecko) Chrome/11.0.696.68 Safari/534.24"

})

html=urllib2.urlopen(url=req,timeout=10).read()

有的网站用了Cookie来限制,主要是涉及到登录和限流,这时候没有什么通用的方法,只能看能否做自动登录或者分析Cookie的问题了。

URL与内容处理

URL里奇形怪状的格式只能个别分析,个别处理,逮到一个算一个。例如针对URL里可能有汉字,相对路径,以及回车换行之类的问题,我们可以先用urlparse模块的urljoin函数处理相对路径的问题,然后去掉乱七八糟的回车换行之类的玩意,最后用urllib2的quote函数处理特殊字符编码和转义的问题,生成真正的URL。

当然,在使用的过程中你会发现Python的urljoin和urlopen都有些问题,因此具体的代码我们在后面再给。

对于那些不管三七二十一就把gzip内容丢过来的,我们直接分析它的内容格式,不要管HTTP头部就好了。代码是这样的:

importurllib2

importgzip,cStringIO

html=urllib2.urlopen('http://www.server110.com').read()

ifhtml[:6]=='\x1f\x8b\x08\x00\x00\x00':

html=gzip.GzipFile(fileobj=cStringIO.StringIO(html)).read()

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值