diff --git a/scrapy/linkextractors/lxmlhtml.py b/scrapy/linkextractors/lxmlhtml.py index e9fa521f3..7064e886d 100644 --- a/scrapy/linkextractors/lxmlhtml.py +++ b/scrapy/linkextractors/lxmlhtml.py @@ -9,7 +9,7 @@ import lxml.etree as etree from scrapy.selector import Selector from scrapy.link import Link -from scrapy.utils.misc import arg_to_iter +from scrapy.utils.misc import arg_to_iter, rel_has_nofollow from scrapy.utils.python import unique as unique_list from scrapy.linkextractors import FilteringLinkExtractor from scrapy.utils.response import get_base_url @@ -62,7 +62,7 @@ class LxmlParserLinkExtractor(object): # to fix relative links after process_value url = urljoin(response_url, url) link = Link(url, _collect_string_content(el) or u'', - nofollow=True if el.get('rel') == 'nofollow' else False) + nofollow=rel_has_nofollow(el.get('rel'))) links.append(link) return unique_list(links, key=lambda link: link.url) \ diff --git a/scrapy/linkextractors/sgml.py b/scrapy/linkextractors/sgml.py index acecd9c0d..7084d0180 100644 --- a/scrapy/linkextractors/sgml.py +++ b/scrapy/linkextractors/sgml.py @@ -9,7 +9,7 @@ from w3lib.url import safe_url_string from scrapy.selector import Selector from scrapy.link import Link from scrapy.linkextractors import FilteringLinkExtractor -from scrapy.utils.misc import arg_to_iter +from scrapy.utils.misc import arg_to_iter, rel_has_nofollow from scrapy.utils.python import unique as unique_list, to_unicode from scrapy.utils.response import get_base_url from scrapy.exceptions import ScrapyDeprecationWarning @@ -80,7 +80,7 @@ class BaseSgmlLinkExtractor(SGMLParser): if self.scan_attr(attr): url = self.process_value(value) if url is not None: - link = Link(url=url, nofollow=True if dict(attrs).get('rel') == 'nofollow' else False) + link = Link(url=url, nofollow=rel_has_nofollow(dict(attrs).get('rel'))) self.links.append(link) self.current_link = link diff --git a/scrapy/utils/misc.py b/scrapy/utils/misc.py index 4215e41d2..303a413d8 100644 --- a/scrapy/utils/misc.py +++ b/scrapy/utils/misc.py @@ -112,3 +112,8 @@ def md5sum(file): break m.update(d) return m.hexdigest() + +def rel_has_nofollow(rel): + """Return True if link rel attribute has nofollow type""" + return True if rel is not None and 'nofollow' in rel.split() else False + diff --git a/tests/test_linkextractors.py b/tests/test_linkextractors.py index d78b25f25..d32ff2d55 100644 --- a/tests/test_linkextractors.py +++ b/tests/test_linkextractors.py @@ -96,12 +96,14 @@ class LinkExtractorTestCase(unittest.TestCase): html = """ Printer-friendly page About us + Something """ response = HtmlResponse("http://example.org/page.html", body=html) lx = SgmlLinkExtractor() self.assertEqual([link for link in lx.extract_links(response)], [ Link(url='http://example.org/page.html?action=print', text=u'Printer-friendly page', nofollow=True), Link(url='http://example.org/about.html', text=u'About us', nofollow=False), + Link(url='http://google.com/something', text=u'Something', nofollow=True), ]) @@ -205,6 +207,9 @@ class SgmlLinkExtractorTestCase(unittest.TestCase):
+