协程池

#coding:utf-8

import time

try:

    from Queue import Queue

except:

    from queue import Queue

 

import gevent

from gevent import monkey

from gevent.pool import Pool

 

import requests

from lxml import etree

 

monkey.patch_all()

 

 

class DoubanSpider(object):

    def __init__(self):

        self.url_list = ["https://movie.douban.com/top250?start=" + str(page) for page in range(0, 226, 25)]

        self.headers = {"User-Agent" : "Mozilla/5.0 (Windows NT 10.0; WOW64; Trident/7.0; rv:11.0) like Gecko"}

        self.queue = Queue()

 

    def send_request(self, url):

        print("[INFO]: send requests {}".format(url))

        response = requests.get(url, headers=self.headers)

        time.sleep(1)

        return response

 

    def parse_page(self, response):

        html = response.content

        html_obj = etree.HTML(html)

        node_list = html_obj.xpath("//div[@class='info']")

 

        for node in node_list:

            title = node.xpath("./div[1]/a/span[1]/text()")[0]

            score = node.xpath(".//span[@class='rating_num']/text()")[0]

 

            self.queue.put(title + "\t" + score)

 

    def main(self):

        # 协程池

        pool = Pool()

        for url in self.url_list:

            pool.apply_async(self.send_request, args=[url], callback=self.parse_page)

 

        # 让主线程

        pool.join()

 

 

if __name__ == '__main__':

    spider = DoubanSpider()

    start = time.time()

    spider.main()

    end = time.time()

    # 3 sec

    print("[INFO]: using time {}".format(end - start))