Fix issue with trying to drop root node

jcharum · jcharum · commit cba19f209bad · 2011-07-05T17:17:38.000-07:00
remove_unlikely_candidates would try to drop_tree the root node if it deemed it
an unlikely candidate.  This prevents that from happening.
diff --git a/readability/readability.py b/readability/readability.py
@@ -295,7 +295,11 @@ def remove_unlikely_candidates(self):
         for elem in self.html.iter():
             s = "%s %s" % (elem.get('class', ''), elem.get('id', ''))
             #self.debug(s)
-            if REGEXES['unlikelyCandidatesRe'].search(s) and (not REGEXES['okMaybeItsACandidateRe'].search(s)) and elem.tag != 'body':
+            if (REGEXES['unlikelyCandidatesRe'].search(s) and
+                    (not REGEXES['okMaybeItsACandidateRe'].search(s)) and
+                    elem.tag != 'body' and
+                    elem.getparent() is not None
+                    ):
                 self.debug("Removing unlikely candidate - %s" % describe(elem))
                 elem.drop_tree()
 
