Question

我正在抓取6个不同的allowed_domains，并希望限制1个域的深度。我如何在scrapy中限制1个域的深度？或者是否可以仅抓取1个深度的异地域名？

Answer 1

Scrapy不提供这样的东西。您可以set the DEPTH_LIMIT per-spider，但不能按域名。

我们能做什么？ Read the code，喝咖啡并解决（顺序很重要）。

我们的想法是禁用Scrapy的内置DepthMiddleware和provide our custom one。

首先，让我们定义设置：

DOMAIN_DEPTHS将是一个每个域都有深度限制的字典
DEPTH_LIMIT设置如果未配置域，我们将保留默认值

示例设置：

DOMAIN_DEPTHS = {'amazon.com': 1, 'homedepot.com': 4}
DEPTH_LIMIT = 3

好的，现在是自定义中间件（基于DepthMiddleware）：

from scrapy import log
from scrapy.http import Request
import tldextract


class DomainDepthMiddleware(object):
    def __init__(self, domain_depths, default_depth):
        self.domain_depths = domain_depths
        self.default_depth = default_depth

    @classmethod
    def from_crawler(cls, crawler):
        settings = crawler.settings
        domain_depths = settings.getdict('DOMAIN_DEPTHS', default={})
        default_depth = settings.getint('DEPTH_LIMIT', 1)

        return cls(domain_depths, default_depth)

    def process_spider_output(self, response, result, spider):
        def _filter(request):
            if isinstance(request, Request):
                # get max depth per domain
                domain = tldextract.extract(request.url).registered_domain
                maxdepth = self.domain_depths.get(domain, self.default_depth)

                depth = response.meta.get('depth', 0) + 1
                request.meta['depth'] = depth

                if maxdepth and depth > maxdepth:
                    log.msg(format="Ignoring link (depth > %(maxdepth)d): %(requrl)s ",
                            level=log.DEBUG, spider=spider,
                            maxdepth=maxdepth, requrl=request.url)
                    return False
            return True

        return (r for r in result or () if _filter(r))

请注意，它需要安装tldextract模块（用于从网址中提取域名）：

>>> import tldextract
>>> url = 'http://stackoverflow.com/questions/27805952/scrapy-set-depth-limit-per-allowed-domains'
>>> tldextract.extract(url).registered_domain
'stackoverflow.com'

现在我们需要关闭默认的中间件并使用我们实现的中间件：

SPIDER_MIDDLEWARES = {
    'myproject.middlewares.DomainDepthMiddleware': 900,
    'scrapy.contrib.spidermiddleware.depth.DepthMiddleware': None
}

Scrapy设置每个allowed_domains的深度限制

1 个答案: