Scrapy 设置深度限制 allowed_domains

Scrapy set depth limit per allowed_domains

我正在抓取 6 个不同的 allowed_domains 并想限制 1 个域的深度。我将如何限制 scrapy 中那个 1 域的深度? 或者是否可以只抓取异地域的 1 个深度?

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

我们能做什么? Read the code,喝咖啡解决(顺序很重要)。

想法是禁用 Scrapy 的内置 DepthMiddleware and 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模块(用于从url中提取域名):

>>> import tldextract
>>> url = '
>>> tldextract.extract(url).registered_domain
'whosebug.com'

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

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