首页 > 代码库 > python + pycurl + 高效批量get请求

python + pycurl + 高效批量get请求

pycurl包是用C编写的libcurl  python接口,速度高于urllib。Libcurl 是一个支持FTP, FTPS, HTTP, HTTPS, GOPHER, TELNET, DICT, FILE 和 LDAP的客户端URL传输库.libcurl也支持HTTPS认证,HTTP POST,HTTP PUT,FTP上传,代理,Cookies,基本身份验证,FTP文件断点继传,HTTP代理通道等等。

 

 

def doBatchGet(url_list):

"""批量get请求,
返回成功的请求list和失败的请求list
"""

  mgr = pycurl.CurlMulti()
  handles = []
  for url in url_list:
    c = pycurl.Curl()
    c.url = url
    c.body = StringIO.StringIO()
    c.http_code = 1
    c.setopt(c.URL, c.url)
    c.setopt(c.WRITEFUNCTION, c.body.write)
    mgr.add_handle(c)
    handles.append(c)

  handle_num = len(handles)
  while handle_num:
    while 1:
      ret, handle_num = mgr.socket_all()
      if ret != pycurl.E_CALL_MULTI_PERFORM:
        break
    mgr.select(0.1)

  err_list, succ_list = [], []
  for c in handles:
    c.http_code = c.getinfo(c.HTTP_CODE)
    if c.http_code != 200:
      print ‘get error: %s, %s‘ % (c.url, c.http_code)
      err_list.append(c.url)
    else:
      res = c.body.getvalue()
      if res: #添加成功条件
        succ_list.append((c.url, res))
      else:
        err_list.append(c.url)

  for c in handles:
    mgr.remove_handle(c)
    c.body.close()
    c.close()

  mgr.close()
  return err_list, succ_list