多线程-豆瓣爬取

#coding:utf-8

import time

 

# Python2、Python3兼容

try:

    from Queue import Queue  # Python2

except:

    from queue import Queue  # Python3

import threading

 

import requests

from lxml import etree

 

 

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):

        # for url in self.url_list:

        #     response = self.send_request(url)

        #     self.parse_page(response)

 

        thread_list = []

        for url in self.url_list:

            thread = threading.Thread(target=self.send_request, args=[url])

            thread.start()

            thread_list.append(thread)

 

        # 让主线程阻塞，等待所有子线程执行结束，主线程再继续执行

        for thread in thread_list:

            thread.join()

 

 

if __name__ == '__main__':

    spider = DoubanSpider()

    start = time.time()

    spider.main()

    end = time.time()

    # 0.3 sec

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