【发布时间】:2014-07-15 03:33:26
【问题描述】:
我想为我在蜘蛛的 start_urls 中设置的每个 url 创建单独的输出文件,或者想以某种方式拆分输出文件 start url。
下面是我的蜘蛛的 start_urls
start_urls = ['http://www.dmoz.org/Arts/', 'http://www.dmoz.org/Business/', 'http://www.dmoz.org/Computers/']
我想创建单独的输出文件,如
Arts.xml
业务.xml
计算机.xml
我不知道该怎么做。我正在考虑通过在 item 管道类的 spider_opened 方法中实现一些类似的东西来实现这一点,
import re
from scrapy import signals
from scrapy.contrib.exporter import XmlItemExporter
class CleanDataPipeline(object):
def __init__(self):
self.cnt = 0
self.filename = ''
@classmethod
def from_crawler(cls, crawler):
pipeline = cls()
crawler.signals.connect(pipeline.spider_opened, signals.spider_opened)
crawler.signals.connect(pipeline.spider_closed, signals.spider_closed)
return pipeline
def spider_opened(self, spider):
referer_url = response.request.headers.get('referer', None)
if referer_url in spider.start_urls:
catname = re.search(r'/(.*)$', referer_url, re.I)
self.filename = catname.group(1)
file = open('output/' + str(self.cnt) + '_' + self.filename + '.xml', 'w+b')
self.exporter = XmlItemExporter(file)
self.exporter.start_exporting()
def spider_closed(self, spider):
self.exporter.finish_exporting()
#file.close()
def process_item(self, item, spider):
self.cnt = self.cnt + 1
self.spider_closed(spider)
self.spider_opened(spider)
self.exporter.export_item(item)
return item
我试图在 start_urls 列表中找到每个抓取项目的引用 URL。如果在 start_urls 中找到引用 URL,则将使用该引用 URL 创建文件名。 但问题是如何在 spider_opened() 方法中访问响应对象。如果我可以在那里访问它,我可以基于它创建文件。
对找到执行此操作的方法有什么帮助吗?提前致谢!
[编辑]
通过更改我的管道代码解决了我的问题。
import re
from scrapy import signals
from scrapy.contrib.exporter import XmlItemExporter
class CleanDataPipeline(object):
def __init__(self):
self.filename = ''
self.exporters = {}
@classmethod
def from_crawler(cls, crawler):
pipeline = cls()
crawler.signals.connect(pipeline.spider_opened, signals.spider_opened)
crawler.signals.connect(pipeline.spider_closed, signals.spider_closed)
return pipeline
def spider_opened(self, spider, fileName = 'default.xml'):
self.filename = fileName
file = open('output/' + self.filename, 'w+b')
exporter = XmlItemExporter(file)
exporter.start_exporting()
self.exporters[fileName] = exporter
def spider_closed(self, spider):
for exporter in self.exporters.itervalues():
exporter.finish_exporting()
def process_item(self, item, spider):
fname = 'default'
catname = re.search(r'http://www.dmoz.org/(.*?)/', str(item['start_url']), re.I)
if catname:
fname = catname.group(1)
self.curFileName = fname + '.xml'
if self.filename == 'default.xml':
if os.path.isfile('output/' + self.filename):
os.rename('output/' + self.filename, 'output/' + self.curFileName)
exporter = self.exporters['default.xml']
del self.exporters['default.xml']
self.exporters[self.curFileName] = exporter
self.filename = self.curFileName
if self.filename != self.curFileName and not self.exporters.get(self.curFileName):
self.spider_opened(spider, self.curFileName)
self.exporters[self.curFileName].export_item(item)
return item
还在spider中实现make_requests_from_url为每个项目设置start_url。
def make_requests_from_url(self, url):
request = Request(url, dont_filter=True)
request.meta['start_url'] = url
return request
【问题讨论】:
-
我面临同样的情况,但是当我尝试应用您的解决方案时,我从 process_item() 方法中得到了“exceptions.KeyError: 'start_url'”错误。你能建议我应该怎么做吗?我是这里的新手。 :'(
-
@NatthawuteSae-Lim 您是否尝试过在蜘蛛中添加一个可以在 request.meta 中设置启动 url 的方法?检查我答案的最后一部分。
标签: python web-scraping scrapy scrapy-spider