Failure if best_elem is root (fix buriy#58)

avalanchy · avalanchy · commit bf9e7404fa68 · 2015-05-06T09:34:55.000+02:00
diff --git a/readability/readability.py b/readability/readability.py
@@ -212,7 +212,9 @@ def get_article(self, candidates, best_candidate, html_partial=False):
         else:
             output = document_fromstring('<div/>')
         best_elem = best_candidate['elem']
-        for sibling in best_elem.getparent().getchildren():
+        parent = best_elem.getparent()
+        siblings = parent.getchildren() if parent else [best_elem]
+        for sibling in siblings:
             # in lxml there no concept of simple text
             # if isinstance(sibling, NavigableString): continue
             append = False
diff --git a/tests/test_article_only.py b/tests/test_article_only.py
@@ -50,3 +50,14 @@ def test_wrong_link_issue_49(self):
         doc = Document(sample)
         res = doc.summary(html_partial=True)
         self.assertEqual('<div><div class="content__article-body ', res[0:39])
+
+    def test_best_elem_is_root_and_passing(self):
+        sample = (
+            '<html class="article" id="body">'
+            '   <body>'
+            '       <p>1234567890123456789012345</p>'
+            '   </body>'
+            '</html>'
+        )
+        doc = Document(sample)
+        doc.summary()
