Improved the Nifty extractor a bit.

DreamCobbler · Jun 5, 2021 · 3c42894 · 3c42894
1 parent fc05d64
commit 3c42894
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/fiction_dl/Extractors/ExtractorNifty.py b/fiction_dl/Extractors/ExtractorNifty.py
@@ -113,7 +113,7 @@ def _InternallyScanStory(
             logging.error("Failed to download story page when scanning.")
             return False
 
-        isHTMLCode = (-1 != pageCode.find("<html>"))
+        isHTMLCode = (-1 != pageCode.find("<html>") or -1 != pageCode.find("<HTML>"))
 
         # Process a multi-chapter story.
 
@@ -125,7 +125,12 @@ def _InternallyScanStory(
 
             # Create a list of chapters.
 
-            chapterElements = soup.select("table.table.table-xtra-condensed > tr")
+            chapterElements = (
+                soup.select("table.table.table-xtra-condensed > tr")
+                or
+                soup.select("table > tr")
+            )
+
             if not chapterElements:
                 logging.error("List of chapters not found.")
                 return False