Item Pipelines

当Item在Spider中被收集之后,它将会被传递到Item Pipeline

每个Item Pipeline组件接收到Item,定义一些操作行为,比如决定此Item是丢弃而存储。

以下是item pipeline的一些典型应用:

  • 验证爬取的数据(检查item包含某些字段,比如说name字段)
  • 查重(并丢弃)
  • 将爬取结果保存到文件或者数据库中

编写item pipeline

编写item pipeline很简单,item pipiline组件是一个独立的Python类,必须实现process_item方法:

process_item(self, item, spider)

当Item在Spider中被收集之后,都需要调用该方法

参数:

  1. item - 爬取的结构化数据
  2. spider 爬取该itemspider

open_spider(self, spider)

当spider被开启时,这个方法被调用。

参数:

  1. spider 被开启的spider

close_spider(spider)

当spider被关闭时,这个方法被调用

参数:

  1. spider 被关闭的spider

将item写入JSON文件

以下pipeline将所有爬取到的item,存储到一个独立地items.json 文件,每行包含一个序列化为'JSON'格式的'item':

  1. import json
  2. class JsonWriterPipeline(object):
  3. def __init__(self):
  4. self.file = open('items.json', 'wb')
  5. def process_item(self, item, spider):
  6. line = json.dumps(dict(item),ensure_ascii=False) + "\n"
  7. self.file.write(line)
  8. return item

启用一个Item Pipeline组件

为了启用Item Pipeline组件,必须将它的类添加到 settings.py文件ITEM_PIPELINES 配置,就像下面这个例子:

  1. ITEM_PIPELINES = {
  2. #'tutorial.pipelines.PricePipeline': 300,
  3. 'tutorial.pipelines.JsonWriterPipeline': 800,
  4. }

分配给每个类的整型值,确定了他们运行的顺序,item按数字从低到高的顺序,通过pipeline,通常将这些数字定义在0-1000范围内。

在这里优化:

以下pipeline将所有爬取到的item,存储到一个独立地items.json 文件,每行包含一个序列化为'JSON'格式的'item':

  1. import json
  2. import codecs
  3. class JsonWriterPipeline(object):
  4. def __init__(self):
  5. self.file = codecs.open('items.json', 'w', encoding='utf-8')
  6. def process_item(self, item, spider):
  7. line = json.dumps(dict(item), ensure_ascii=False) + "\n"
  8. self.file.write(line)
  9. return item
  10. def spider_closed(self, spider):
  11. self.file.close()

针对spider里面的utf-8编码格式去掉.encode('utf-8')

  1. item = RecruitItem()
  2. item['name']=name.encode('utf-8')
  3. item['detailLink']=detailLink.encode('utf-8')
  4. item['catalog']=catalog.encode('utf-8')
  5. item['recruitNumber']=recruitNumber.encode('utf-8')
  6. item['workLocation']=workLocation.encode('utf-8')
  7. item['publishTime']=publishTime.encode('utf-8')

将item写入MongoDB

from_crawler(cls, crawler)

如果使用,这类方法被调用创建爬虫管道实例。必须返回管道的一个新实例。crawler提供存取所有Scrapy核心组件配置和信号管理器;对于pipelines这是一种访问配置和信号管理器 的方式。

  1. 参数: crawler (Crawler object) crawler that uses this pipeline

在这个例子中,我们将使用pymongo将Item写到MongoDB。MongoDB的地址和数据库名称在Scrapy setttings.py配置文件中;

这个例子主要是说明如何使用from_crawler()方法

  1. import pymongo
  2. class MongoPipeline(object):
  3. collection_name = 'scrapy_items'
  4. def __init__(self, mongo_uri, mongo_db):
  5. self.mongo_uri = mongo_uri
  6. self.mongo_db = mongo_db
  7. @classmethod
  8. def from_crawler(cls, crawler):
  9. return cls(
  10. mongo_uri=crawler.settings.get('MONGO_URI'),
  11. mongo_db=crawler.settings.get('MONGO_DATABASE', 'items')
  12. )
  13. def open_spider(self, spider):
  14. self.client = pymongo.MongoClient(self.mongo_uri)
  15. self.db = self.client[self.mongo_db]
  16. def close_spider(self, spider):
  17. self.client.close()
  18. def process_item(self, item, spider):
  19. self.db[self.collection_name].insert(dict(item))
  20. return item