您的位置:首页 > 博客中心 > 数据库 >

运维学python之爬虫高级篇(七)scrapy爬取知乎关注用户存入mongodb

时间:2022-03-15 06:45

首先,祝大家开工大吉!
本篇将要介绍的是从一个用户开始,通过抓关注列表和粉丝列表,实现用户的详细信息抓取并将抓取到的结果存储到 MongoDB。

1 环境需求

基础环境沿用之前的环境,只是增加了MongoDB(非关系型数据库)和PyMongo(Python 的 MongoDB 连接库),默认我认为大家都已经安装好并启动 了MongoDB 服务。

项目创建、爬虫创建、禁用ROBOTSTXT_OBEY设置略(可以参考上一篇)

2 测试爬虫效果

我这里先写一个简单的爬虫,爬取用户的关注人数和粉丝数,代码如下:

# -*- coding: utf-8 -*-
import scrapy

class ZhuHuSpider(scrapy.Spider):
    """
    知乎爬虫
    """
    name = ‘zhuhu‘
    allowed_domains = [‘zhihu.com‘]
    start_urls = [‘https://www.zhihu.com/people/wo-he-shui-jiu-xing/following‘]

    def parse(self, response):
        # 他关注的人数
        tnum = response.css("strong.NumberBoard-itemValue::text").extract()[0]
        # 粉丝数
        fnum = response.css("strong.NumberBoard-itemValue::text").extract()[1]
        print("他关注的人数为:%s" % tnum)
        print("他粉丝的人数为:%s" % fnum)

pychram中运行的结果如下:
技术分享图片
出现500错误了,我们加上headers再试试,我们直接在settings.py中设置,如下:
技术分享图片
再次执行看看结果:
技术分享图片
这次就正常获取到我们需要的信息了

3 爬取分析

我们就用中本聪的主页作为分析入口吧,主页如下:

分析用户关注列表如下:
技术分享图片
鼠标放到用户图像上,会显示详细信息如下:
技术分享图片

这里要注意我用的是火狐浏览器,选择网络--XHR来获取信息

ajax技术的核心是XMLHttpRequest对象(简称XHR),这是由微软首先引入的一个特性,其他浏览器提供商后来都提供了相同的实现。XHR为向服务器发送请求和解析服务器响应提供了流畅的接口,能够以异步方式从服务器取得更多信息,意味着用户单击后,可以不必刷新页面也能取得新数据。

通过上面的请求我们可以获取的连接如下:

#用户详细信息
https://www.zhihu.com/api/v4/members/li-kang-65?include=allow_message,is_followed,is_following,is_org,is_blocking,employments,answer_count,follower_count,articles_count,gender,badge[?(type=best_answerer)].topics

https://www.zhihu.com/api/v4/members/jin-xiao-94-7?include=allow_message,is_followed,is_following,is_org,is_blocking,employments,answer_count,follower_count,articles_count,gender,badge[?(type=best_answerer)].topics

#关注的人信息
https://www.zhihu.com/api/v4/members/satoshi_nakamoto/followees?include=data[*].answer_count,articles_count,gender,follower_count,is_followed,is_following,badge[?(type=best_answerer)].topics&offset=0&limit=20

通过分析上面的链接可以看出
1.用户详细信息链接组成:}
其中user是用户的url_token,include是allow_message,is_followed,is_following,is_org,is_blocking,employments,answer_count,follower_count,articles_count,gender,badge[?(type=best_answerer)].topics
2.关注人信息链接组成:}
其中include为data[*].answer_count,articles_count,gender,follower_count,is_followed,is_following,badge[?(type=best_answerer)].topics,offset为分页偏移量,limit为每页用户数量,可以通过下图看出:

第一页
技术分享图片
第二页
技术分享图片
第三页
技术分享图片

4 开始爬取

我们还是先写一个简易的爬虫,把功能先实现,代码如下:

# -*- coding: utf-8 -*-
import scrapy

class ZhuHuSpider(scrapy.Spider):
    """
    知乎爬虫
    """
    name = ‘zhuhu‘
    allowed_domains = [‘zhihu.com‘]
    # 用户详细信息地址
    user_detail = ‘https://www.zhihu.com/api/v4/members/{user}?include={include}‘
    # 用户详细信息中的include
    user_include = ‘allow_message,is_followed,‘                    ‘is_following,‘                    ‘is_org,is_blocking,‘                    ‘employments,‘                    ‘answer_count,‘                    ‘follower_count,‘                    ‘articles_count,‘                    ‘gender,‘                    ‘badge[?(type=best_answerer)].topics‘
    # 关注的人地址
    follow_url = ‘https://www.zhihu.com/api/v4/members/{user}/followees?include={include}&offset={offset}&limit={limit}‘
    # 关注的人include
    follow_include = ‘data[*].answer_count,‘                      ‘articles_count,‘                      ‘gender,‘                      ‘follower_count,‘                      ‘is_followed,‘                      ‘is_following,‘                      ‘badge[?(type=best_answerer)].topics‘
    # 初始user
    start_user = ‘satoshi_nakamoto‘

    def start_requests(self):
        # 这里重新定义start_requests方法,注意这里的format用法
        yield scrapy.Request(self.user_detail.format(user=self.start_user, include=self.user_include),
                             callback=self.parse_user)
        yield scrapy.Request(self.follow_url.format(user=self.start_user, include=self.follow_include, offset=20, limit=20),
                             callback=self.parse_follow)

    def parse_user(self, response):
        print(‘user:%s‘ % response.text)

    def parse_follow(self, response):
        print(‘follow:%s‘ % response.text)

输出结果如下:
技术分享图片
这里需要注意的是authorization信息一定要在headers中添加,不然会报错,authorization在headers中的形式如下:
技术分享图片
测试发现authorization值在一段时间内是不会发生变化的,是否永久不变还有待考证。

5 parse_user编写

parse_user方法用来解析用户的详细数据,存储并发现此用户的关注列表,返回给parse_follow方法来处理,用户详细存储字段如下:
技术分享图片
为了省事我把所有字段都添加到items.py中(如果运行spider后报错,提示字段未找到,就将那个字段添加进来即可),如下:

class UserItem(scrapy.Item):
    """
    定义了响应报文中json的字段
    """
    is_followed = scrapy.Field()
    avatar_url_template = scrapy.Field()
    user_type = scrapy.Field()
    answer_count = scrapy.Field()
    is_following = scrapy.Field()
    url = scrapy.Field()
    type = scrapy.Field()
    url_token = scrapy.Field()
    id = scrapy.Field()
    allow_message = scrapy.Field()
    articles_count = scrapy.Field()
    is_blocking = scrapy.Field()
    name = scrapy.Field()
    headline = scrapy.Field()
    gender = scrapy.Field()
    avatar_url = scrapy.Field()
    follower_count = scrapy.Field()
    is_org = scrapy.Field()
    employments = scrapy.Field()
    badge = scrapy.Field()
    is_advertiser = scrapy.Field()

parse_user方法代码如下:

    def parse_user(self, response):
        """
        解析用户详细信息方法
        :param response: 获取的内容,转化为json格式
        """
        # 通过json.loads方式转换为json格式
        results = json.loads(response.text)
        # 引入item类
        item = UserItem()
        # 通过循环判断字段是否存在,存在将结果存入items中
        for field in item.fields:
            if field in results.keys():
                item[field] = results.get(field)
        # 直接返回item
        yield item
        # 将获取的用户通过format方式组合成新的url,调用callback函数交给parse_follow方法解析
        yield scrapy.Request(self.follows_url.format(user=results.get(‘url_token‘),
                                                     include=self.follow_include, offset=0, limit=20),
                             callback=self.parse_follow)

6 parse_follow方法编写

首先也要将获取的response转换为json格式,获取关注的用户,对每一个用户继续爬取,同时也要处理分页。可以看下面两个图:
技术分享图片
技术分享图片

重新编写后的parse_follow方法如下:

    def parse_follow(self, response):
        """
        解析关注的人列表方法
        """
        # 格式化response
        results = json.loads(response.text)
        # 判断data是否存在,如果存在就继续调用parse_user解析用户详细信息
        if ‘data‘ in results.keys():
            for result in results.get(‘data‘):
                yield scrapy.Request(self.user_detail.format(user=result.get(‘url_token‘), include=self.user_include),
                                     callback=self.parse_user)
        # 判断paging是否存在,如果存在并且is_end参数为False,则继续爬取下一页,如果is_end为True,说明为最后一页
        if ‘paging‘ in results.keys() and results.get(‘paging‘).get(‘is_end‘) == False:
            next_page = results.get(‘paging‘).get(‘next‘)
            yield scrapy.Request(next_page, callback=self.parse_follow)

运行爬虫后的结果如下图:
技术分享图片
可以看到一直在获取内容。

7 存入mongodb

7.1 item pipeline

存储使用MongoDB,我们需要修改Item Pipeline,参照修改的代码如下:

class ZhiHuspiderPipeline(object):
    """
    知乎数据存入monogodb数据库类,参考官网示例
    """

    collection_name = ‘user‘

    def __init__(self, mongo_uri, mongo_db):
        """
        初始化参数
        :param mongo_uri:mongo uri
        :param mongo_db: db name
        """
        self.mongo_uri = mongo_uri
        self.mongo_db = mongo_db

    @classmethod
    def from_crawler(cls, crawler):
        return cls(
            mongo_uri=crawler.settings.get(‘MONGO_URI‘),
            mongo_db=crawler.settings.get(‘MONGO_DATABASE‘, ‘items‘)
        )

    def open_spider(self, spider):
        # 打开连接
        self.client = pymongo.MongoClient(self.mongo_uri)
        # db_auth因为我的mongodb设置了认证,所以需要这两步,未设置可以注释
        self.db_auth = self.client.admin
        self.db_auth.authenticate("admin", "password")
        self.db = self.client[self.mongo_db]

    def close_spider(self, spider):
        self.client.close()

    def process_item(self, item, spider):
        # 这里使用update方法
        self.db[self.collection_name].update({‘url_token‘: item[‘url_token‘]},  dict(item), True)
        return item

这里要说一说update方法,update() 方法用于更新已存在的文档。语法格式如下:

db.collection.update(
   <query>,  # update的查询条件,类似sql update查询内where后面的
   <update>, #  update的对象和一些更新的操作符(如$,$inc...)等,也可以理解为sql update查询内set后面的
   {
     upsert: <boolean>, # 可选,这个参数的意思是,如果不存在update的记录,是否插入objNew,true为插入,默认是false,不插入。
     multi: <boolean>, # 可选,mongodb 默认是false,只更新找到的第一条记录,如果这个参数为true,就把按条件查出来多条记录全部更新
     writeConcern: <document> # 可选,抛出异常的级别。
   }
)

使用update方法,如果查询数据存在的话就更新,不存在的话就插入dict(item),这样就可以去重了。

7.2 settings配置

技术分享图片
再次运行spider后结果如下:
技术分享图片
也可以看到mongodb中数据,如下:
技术分享图片

技术分享图片
本篇部分参考:

到此本篇完结。

热门排行

今日推荐

热门手游