码迷,mamicode.com
首页 > 其他好文 > 详细

爬虫小案例:多协程工作

时间:2020-01-26 23:51:48      阅读:174      评论:0      收藏:0      [点我收藏+]

标签:text   www   爬取   cin   orm   get   form   一起   mon   

为提高工作效率,让多个爬虫一起工作

需要使用的库是:gevent

方式一

1.使用队列的形式:

from gevent import monkey
# 从gevent库里导入monkey模块。
monkey.patch_all()
# monkey.patch_all()能把程序变成协作式运行,就是可以帮助程序实现异步。
import gevent,time,requests
# 导入gevent、time、requests
from gevent.queue import Queue
# 从gevent库里导入queue模块

start = time.time()
# 记录开始时间

url_list = [
    https://www.baidu.com/,
    https://www.sina.com.cn/,
    http://www.sohu.com/,
    https://www.qq.com/,
    https://www.163.com/,
    http://www.iqiyi.com/,
    https://www.tmall.com/,
    http://www.ifeng.com/
]
# 要爬取的网站

work = Queue()
# 创建队列对象,并赋值给work。
for url in url_list:
    work.put_nowait(url)
    # 用put_nowait()函数可以把网址都放进队列里

def crawler():
    while not work.empty():
        # 当队列不是空的时候,就执行下面的程序。
        url = work.get_nowait()
        # 用get_nowait()函数可以把队列里的网址都取出。
        r = requests.get(url)
        # 抓取网站内容
        print(url, work.qsize(), r.status_code)
        # 打印网址、队列长度、抓取请求的状态码

task_list = []
# 创建空的任务列表

for i in range(2):
    # 相当于创建了2个爬虫
    task = gevent.spawn(crawler)
    # 用gevent.spawn()函数创建执行crawler()函数的任务。
    task_list.append(task)
    # 往任务列表添加任务

gevent.joinall(task_list)
# 用gevent.joinall方法,执行任务列表里的所有任务,就是让爬虫开始爬取网站。

end = time.time()
# 结束时间

print(end-start)
# 总用时

以上是创建了一个队列存储 url,然后让 2 个爬虫一起工作,哪个爬虫执行完了,就可以从队列中获取下一个 url 去执行

技术图片

 

方式二

2.不使用队列,有多少个任务,就创建多少个爬虫去执行:

from gevent import monkey
# 从gevent库里导入monkey模块。
monkey.patch_all()
# monkey.patch_all()能把程序变成协作式运行,就是可以帮助程序实现异步。
import gevent,time,requests
#导入gevent、time、requests。

start = time.time()
# 记录程序开始时间。

url_list = [
    https://www.baidu.com/,
    https://www.sina.com.cn/,
    http://www.sohu.com/,
    https://www.qq.com/,
    https://www.163.com/,
    http://www.iqiyi.com/,
    https://www.tmall.com/,
    http://www.ifeng.com/
]
#把8个网站封装成列表。

def crawler(url):
#定义一个crawler()函数。
    r = requests.get(url)
    #用requests.get()函数爬取网站。
    print(url,time.time()-start,r.status_code)
    #打印网址、请求运行时间、状态码。

tasks_list = []
#创建空的任务列表。

for url in url_list:
    # 遍历url_list。
    task = gevent.spawn(crawler, url)
    # 用gevent.spawn()函数创建任务。
    tasks_list.append(task)
    # 往任务列表添加任务。

gevent.joinall(tasks_list)
# 执行任务列表里的所有任务,就是让爬虫开始爬取网站。
end = time.time()
# 记录程序结束时间。
print(end-start)
#打印程序最终所需时间。

以上只有 8 个网站,数量不多,可以这样操作,但是,如果是 1000 个网站,就是一下子发起1000 次请求,这样的恶意请求,会拖垮网站的服务器。所以,这样做是不可取的,要使用队列的形式来操作。

小练习:多协程爬取豆瓣 Top250电影

from gevent import monkey
# 程序异步执行
monkey.patch_all()
import gevent,requests,csv
from bs4 import BeautifulSoup
from gevent.queue import Queue

# 电影列表
movies = []

# 队列对象
work = Queue()
for x in range(10):
    url = https://movie.douban.com/top250?start={}&filter=.format(x * 25)
    work.put_nowait(url)

def moviesort(movie):
    return int(movie[0])

def crawler():
    while not work.empty():
        url = work.get_nowait()
        # 为躲避反爬机制,伪装成浏览器的请求头
        headers = {
            User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_3) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.108 Safari/537.36 OPR/65.0.3467.78 (Edition Baidu)}
        print(正请求:{}.format(url))
        res = requests.get(url, headers=headers)
        if res.status_code == 200:
            htmltext = res.text
            soup = BeautifulSoup(htmltext, html.parser)
            ol = soup.find(ol, class_=grid_view)
            for li in ol.find_all(li):
                # 排名
                num = li.find(div, class_=pic).find(em).text

                info = li.find(div, class_=info)

                # 标题
                title = info.find(div, class_=hd).find(span, class_=title).text.strip()

                # 链接
                link = info.find(div, class_=hd).find(a)[href]

                # 评分
                rating_num = info.find(span, class_=rating_num).text

                # 推荐语
                inq = info.find(span, class_=inq).text

                # 上映时间、地区、类型
                bd = info.find(div, class_=bd).find(p).contents[2]
                bd = bd.split(/)

                movies.append([num,title,link,rating_num,inq,bd[0].strip(),bd[1].strip(),bd[2].strip()])
        else:
            print({}:请求失败!.format(url))

task_list = []

# 创建 2 个爬虫
for i in range(2):
    # 创建任务
    task = gevent.spawn(crawler)
    # 把任务添加到任务列表
    task_list.append(task)

# 执行所有任务
gevent.joinall(task_list)

# 排序
movies.sort(key=moviesort)

# 打印电影数据
print(movies)

# 保存数据
with open(./豆瓣Top250电影.csv, w, newline=‘‘, encoding=utf-8-sig) as file:
    # 编码utf-8-sig:支持python3,不支持python2
    writer = csv.writer(file)
    writer.writerow([排名, 电影名称, 链接, 评分, 推荐语, 上映时间, 地区, 类型])
    for movie in movies:
        writer.writerow(movie)

print(完毕!)

 

爬虫小案例:多协程工作

标签:text   www   爬取   cin   orm   get   form   一起   mon   

原文地址:https://www.cnblogs.com/KeenLeung/p/12234965.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!