novel-downloader 1.1.0__py3-none-any.whl

novel_downloader/core/parsers/qidian_parser/browser/chapter_encrypted.py

@@ -0,0 +1,498 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+novel_downloader.core.parsers.qidian_parser.browser.chapter_encrypted
+---------------------------------------------------------------------
+
+Support for parsing encrypted chapters from Qidian using font OCR mapping,
+CSS rules, and custom rendering logic.
+
+Includes:
+- Font downloading and caching
+- Encrypted paragraph extraction
+- Custom CSS parsing and layout restoration
+- Font-based OCR decryption and mapping
+"""
+
+from __future__ import annotations
+
+import json
+import logging
+from pathlib import Path
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Set, Tuple, Union
+
+import tinycss2
+from bs4 import BeautifulSoup, Tag
+
+from novel_downloader.utils.network import download_font_file
+from novel_downloader.utils.text_utils import apply_font_mapping
+
+from ..shared import (
+    extract_chapter_info,
+    find_ssr_page_context,
+)
+
+if TYPE_CHECKING:
+    from .main_parser import QidianBrowserParser
+
+logger = logging.getLogger(__name__)
+IGNORED_CLASS_LISTS = {"title", "review"}
+
+
+def parse_encrypted_chapter(
+    parser: QidianBrowserParser,
+    soup: BeautifulSoup,
+    chapter_id: str,
+) -> Dict[str, Any]:
+    """
+    Extract and return the formatted textual content of an encrypted chapter.
+
+    Steps:
+    1. Load SSR JSON context for CSS, fonts, and metadata.
+    3. Decode and save randomFont bytes; download fixedFont via download_font().
+    4. Extract paragraph structures and save debug JSON.
+    5. Parse CSS rules and save debug JSON.
+    6. Determine paragraph name prefixes and ending number; save debug text.
+    7. Render encrypted paragraphs, then run OCR font-mapping.
+    8. Extracts paragraph texts and formats them.
+
+    :param html_str: Raw HTML content of the chapter page.
+    :return: Formatted chapter text or empty string if not parsable.
+    """
+    try:
+        if not (parser._decode_font and parser._font_ocr):
+            return {}
+        ssr_data = find_ssr_page_context(soup)
+        chapter_info = extract_chapter_info(ssr_data)
+        if not chapter_info:
+            logger.warning(
+                "[Parser] ssr_chapterInfo not found for chapter '%s'", chapter_id
+            )
+            return {}
+        debug_base_dir: Optional[Path] = None
+        if parser._font_debug_dir:
+            debug_base_dir = parser._font_debug_dir / chapter_id
+            debug_base_dir.mkdir(parents=True, exist_ok=True)
+
+        css_str = chapter_info["css"]
+        randomFont_str = chapter_info["randomFont"]
+        fixedFontWoff2_url = chapter_info["fixedFontWoff2"]
+
+        title = chapter_info.get("chapterName", "Untitled")
+        chapter_id = chapter_info.get("chapterId", "")
+        author_say = chapter_info.get("authorSay", "")
+        update_time = chapter_info.get("updateTime", "")
+        update_timestamp = chapter_info.get("updateTimestamp", 0)
+        modify_time = chapter_info.get("modifyTime", 0)
+        word_count = chapter_info.get("wordsCount", 0)
+        vip = bool(chapter_info.get("vipStatus", 0))
+        is_buy = bool(chapter_info.get("isBuy", 0))
+        seq = chapter_info.get("seq", None)
+        order = chapter_info.get("chapterOrder", None)
+        volume = chapter_info.get("extra", {}).get("volumeName", "")
+
+        # extract + save font
+        rf = json.loads(randomFont_str)
+        rand_path = parser._base_cache_dir / "randomFont.ttf"
+        rand_path.parent.mkdir(parents=True, exist_ok=True)
+        rand_path.write_bytes(bytes(rf["data"]))
+
+        fixed_path = download_font_file(
+            url=fixedFontWoff2_url, target_folder=parser._fixed_font_dir
+        )
+        if fixed_path is None:
+            raise ValueError("fixed_path is None: failed to download font")
+
+        # Extract and render paragraphs from HTML with CSS rules
+        main_paragraphs = extract_paragraphs_recursively(soup, chapter_id)
+        if debug_base_dir:
+            main_paragraphs_path = debug_base_dir / "main_paragraphs_debug.json"
+            main_paragraphs_path.write_text(
+                json.dumps(main_paragraphs, ensure_ascii=False, indent=2),
+                encoding="utf-8",
+            )
+
+        paragraphs_rules = parse_rule(css_str)
+        if debug_base_dir:
+            paragraphs_rules_path = debug_base_dir / "paragraphs_rules_debug.json"
+            paragraphs_rules_path.write_text(
+                json.dumps(paragraphs_rules, ensure_ascii=False, indent=2),
+                encoding="utf-8",
+            )
+
+        paragraph_names = parse_paragraph_names(paragraphs_rules)
+        end_number = parse_end_number(main_paragraphs, paragraph_names)
+        if debug_base_dir:
+            paragraphs_rules_path = debug_base_dir / "paragraph_names_debug.txt"
+            temp = f"names:\n{paragraph_names}\n\nend_number: {end_number}"
+            paragraphs_rules_path.write_text(
+                temp,
+                encoding="utf-8",
+            )
+        if not end_number:
+            logger.warning(
+                f"[Parser] No end_number found after parsing chapter '{chapter_id}'"
+            )
+            return {}
+
+        paragraphs_str, refl_list = render_paragraphs(
+            main_paragraphs, paragraphs_rules, end_number
+        )
+        if debug_base_dir:
+            paragraphs_str_path = debug_base_dir / f"{chapter_id}_debug.txt"
+            paragraphs_str_path.write_text(paragraphs_str, encoding="utf-8")
+
+        # Run OCR + fallback mapping
+        char_set = set(c for c in paragraphs_str if c not in {" ", "\n", "\u3000"})
+        refl_set = set(refl_list)
+        char_set = char_set - refl_set
+        if debug_base_dir:
+            char_sets_path = debug_base_dir / "char_set_debug.txt"
+            temp = f"char_set:\n{char_set}\n\nrefl_set:\n{refl_set}"
+            char_sets_path.write_text(
+                temp,
+                encoding="utf-8",
+            )
+
+        mapping_result = parser._font_ocr.generate_font_map(
+            fixed_font_path=fixed_path,
+            random_font_path=rand_path,
+            char_set=char_set,
+            refl_set=refl_set,
+            chapter_id=chapter_id,
+        )
+        if debug_base_dir:
+            mapping_json_path = debug_base_dir / "font_mapping.json"
+            mapping_json_path.write_text(
+                json.dumps(mapping_result, ensure_ascii=False, indent=2),
+                encoding="utf-8",
+            )
+
+        # Reconstruct final readable text
+        original_text = apply_font_mapping(paragraphs_str, mapping_result)
+
+        final_paragraphs_str = "\n\n".join(
+            line.strip() for line in original_text.splitlines() if line.strip()
+        )
+        chapter_info = {
+            "id": str(chapter_id),
+            "title": title,
+            "content": final_paragraphs_str,
+            "author_say": author_say.strip() if author_say else "",
+            "updated_at": update_time,
+            "update_timestamp": update_timestamp,
+            "modify_time": modify_time,
+            "word_count": word_count,
+            "vip": vip,
+            "purchased": is_buy,
+            "order": order,
+            "seq": seq,
+            "volume": volume,
+        }
+        return chapter_info
+
+    except Exception as e:
+        logger.warning(
+            "[Parser] parse error for encrypted chapter '%s': %s", chapter_id, e
+        )
+    return {}
+
+
+def extract_paragraphs_recursively(
+    soup: BeautifulSoup, chapter_id: str = ""
+) -> List[Dict[str, Any]]:
+    """
+    Extracts paragraph elements under <main id="c-{chapter_id}"> from HTML
+    and converts them to a nested data structure for further processing.
+
+    :param html_str: Full HTML content.
+    :param chapter_id: ID used to locate <main id="c-{chapter_id}">.
+
+    :return list: List of parsed <p> paragraph data.
+    """
+
+    def parse_element(elem: Any) -> Union[Dict[str, Any], None]:
+        if not isinstance(elem, Tag):
+            return None
+        result = {"tag": elem.name, "attrs": dict(elem.attrs), "data": []}
+        for child in elem.contents:
+            if isinstance(child, Tag):
+                parsed = parse_element(child)
+                if parsed:
+                    result["data"].append(parsed)
+            else:
+                text = child
+                if text:
+                    result["data"].append(text)
+        return result
+
+    if chapter_id:
+        main_id = f"c-{chapter_id}"
+        main_tag = soup.find("main", id=main_id)
+        if not main_tag:
+            return []
+    else:
+        main_tag = soup
+
+    result = []
+    for p in main_tag.find_all("p"):
+        parsed_p = parse_element(p)
+        if parsed_p:
+            result.append(parsed_p)
+
+    return result
+
+
+def parse_rule(css_str: str) -> Dict[str, Any]:
+    """
+    Parse a CSS string and extract style rules for rendering.
+
+    Handles:
+    - font-size:0 (mark for deletion)
+    - scaleX(-1) (mark as mirrored)
+    - ::before / ::after with content or attr()
+    - class + tag selector mapping
+    - custom rendering order via 'order'
+
+    :param css_str: Raw CSS stylesheet string.
+    :return: Dict with "rules" and "orders" for rendering.
+    """
+
+    rules: Dict[str, Any] = {}
+    orders = []
+
+    stylesheet = tinycss2.parse_stylesheet(
+        css_str, skip_comments=True, skip_whitespace=True
+    )
+
+    for rule in stylesheet:
+        if rule.type != "qualified-rule":
+            continue
+
+        selector = tinycss2.serialize(rule.prelude).strip()
+        declarations = tinycss2.parse_declaration_list(rule.content)
+
+        parsed = {}
+        order_val = None
+
+        for decl in declarations:
+            if decl.type != "declaration":
+                continue
+            name = decl.lower_name
+            value = tinycss2.serialize(decl.value).strip()
+
+            if name == "font-size" and value == "0":
+                if "::first-letter" in selector:
+                    parsed["delete-first"] = True
+                else:
+                    parsed["delete-all"] = True
+            elif name == "transform" and value.lower() == "scalex(-1)":
+                parsed["transform-x_-1"] = True
+            elif name == "order":
+                order_val = value
+            elif name == "content":
+                if "::after" in selector:
+                    if "attr(" in value:
+                        parsed["append-end-attr"] = value.split("attr(")[1].split(")")[
+                            0
+                        ]
+                    else:
+                        parsed["append-end-char"] = value.strip("\"'")
+                elif "::before" in selector:
+                    if "attr(" in value:
+                        parsed["append-start-attr"] = value.split("attr(")[1].split(
+                            ")"
+                        )[0]
+                    else:
+                        parsed["append-start-char"] = value.strip("\"'")
+
+        # Store in structure
+        if selector.startswith(".sy-"):
+            rules.setdefault("sy", {})[selector[1:]] = parsed
+        elif selector.startswith(".p") and " " in selector:
+            class_str, tag_part = selector.split(" ", 1)
+            class_str = class_str.lstrip(".")
+            tag_part = tag_part.split("::")[0]
+            rules.setdefault(class_str, {}).setdefault(tag_part, {}).update(parsed)
+
+        if order_val:
+            orders.append((selector, order_val))
+
+    orders.sort(key=lambda x: int(x[1]))
+    return {"rules": rules, "orders": orders}
+
+
+def parse_paragraph_names(rules: Dict[str, Any]) -> Set[str]:
+    """
+    Extract all paragraph selector names from parsed rules, excluding "sy".
+    """
+    paragraph_names = set()
+    for group, group_rules in rules.get("rules", {}).items():
+        if group == "sy":
+            continue
+        paragraph_names.update(group_rules.keys())
+    return paragraph_names
+
+
+def parse_end_number(
+    main_paragraphs: List[Dict[str, Any]], paragraph_names: Set[str]
+) -> Optional[int]:
+    """
+    Find the most frequent numeric suffix from tag names
+    matched by given paragraph prefixes.
+    """
+    end_numbers: Dict[int, int] = {}
+    sorted_names = sorted(paragraph_names, key=len, reverse=True)
+
+    def rec_parse(item: Union[List[Any], Dict[str, Any]]) -> None:
+        if isinstance(item, list):
+            for element in item:
+                rec_parse(element)
+        elif isinstance(item, dict):
+            tag = item.get("tag")
+            if isinstance(tag, str):
+                for prefix in sorted_names:
+                    if tag.startswith(prefix):
+                        remain = tag[len(prefix) :]
+                        if remain.isdigit():
+                            num = int(remain)
+                            end_numbers[num] = end_numbers.get(num, 0) + 1
+                        break
+            for val in item.values():
+                if isinstance(val, (list, dict)):
+                    rec_parse(val)
+
+    rec_parse(main_paragraphs)
+
+    if not end_numbers:
+        logger.warning("[Parser] No valid ending numbers found")
+        return None
+
+    sorted_numbers = sorted(
+        end_numbers.items(), key=lambda x: (x[1], x[0]), reverse=True
+    )
+
+    logger.debug(
+        "[Parser] Top 3 end numbers:\n%s",
+        "\n".join(f"{n}: {c}" for n, c in sorted_numbers[:3]),
+    )
+
+    return sorted_numbers[0][0]
+
+
+def render_paragraphs(
+    main_paragraphs: List[Dict[str, Any]],
+    rules: Dict[str, Any],
+    end_number: int,
+) -> Tuple[str, List[str]]:
+    """
+    Applies the parsed CSS rules to the paragraph structure and
+    reconstructs the visible text.
+
+    Handles special class styles like .sy-*, text order control,
+    mirrored characters, etc.
+
+    :param main_paragraphs: A list of paragraph dictionaries, each with 'attrs'
+                            and 'data' fields representing structured content.
+    :param rules: A dictionary with keys 'orders' and 'rules', parsed from CSS.
+                  - rules['orders']: List of (selector, id) tuples.
+                  - rules['rules']: Nested dict containing transformation rules.
+    :param end_number: HTML tag suffix (e.g. span123 -> 123).
+
+    :return:
+        - A reconstructed paragraph string with line breaks.
+        - A list of mirrored (reflected) characters for later OCR processing.
+    """
+    orders: List[Tuple[str, str]] = rules.get("orders", [])
+    rules = rules.get("rules", {})
+    refl_list: List[str] = []
+
+    def apply_rule(data: Dict[str, Any], rule: Dict[str, Any]) -> str:
+        if rule.get("delete-all", False):
+            return ""
+
+        curr_str = ""
+        if isinstance(data.get("data"), list) and data["data"]:
+            first_data = data["data"][0]
+            if isinstance(first_data, str):
+                curr_str += first_data
+
+        if rule.get("delete-first", False):
+            if len(curr_str) <= 1:
+                curr_str = ""
+            else:
+                curr_str = curr_str[1:]
+
+        curr_str += rule.get("append-end-char", "")
+
+        attr_name = rule.get("append-end-attr", "")
+        if attr_name:
+            curr_str += data.get("attrs", {}).get(f"{attr_name}{end_number}", "")
+
+        curr_str = rule.get("append-start-char", "") + curr_str
+
+        attr_name = rule.get("append-start-attr", "")
+        if attr_name:
+            curr_str = (
+                data.get("attrs", {}).get(f"{attr_name}{end_number}", "") + curr_str
+            )
+
+        if rule.get("transform-x_-1", False):
+            refl_list.append(curr_str)
+        return curr_str
+
+    paragraphs_str = ""
+    for paragraph in main_paragraphs:
+        class_list = paragraph.get("attrs", {}).get("class", [])
+        p_class_str = next((c for c in class_list if c.startswith("p")), None)
+        curr_datas = paragraph.get("data", [])
+
+        ordered_cache = {}
+        for data in curr_datas:
+            # 文本节点直接加
+            if isinstance(data, str):
+                paragraphs_str += data
+                continue
+
+            if isinstance(data, dict):
+                tag = data.get("tag", "")
+                attrs = data.get("attrs", {})
+
+                # 跳过 span.review
+                if tag == "span" and "class" in attrs and "review" in attrs["class"]:
+                    continue
+
+                # sy 类型标签处理
+                if tag == "y":
+                    tag_class_list = attrs.get("class", [])
+                    tag_class = next(
+                        (c for c in tag_class_list if c.startswith("sy-")), None
+                    )
+
+                    if tag_class in rules.get("sy", {}):
+                        curr_rule = rules["sy"][tag_class]
+                        paragraphs_str += apply_rule(data, curr_rule)
+                    continue
+
+                if not p_class_str:
+                    if any(cls in IGNORED_CLASS_LISTS for cls in class_list):
+                        continue
+                    logger.debug(f"[parser] not find p_class_str: {class_list}")
+                    continue
+                # 普通标签处理，根据 orders 顺序匹配
+                for ord_selector, ord_id in orders:
+                    tag_name = f"{ord_selector}{end_number}"
+                    if data.get("tag") != tag_name:
+                        continue
+                    curr_rule = rules.get(p_class_str, {}).get(ord_selector)
+                    curr_rule = curr_rule if curr_rule else {}
+                    ordered_cache[ord_selector] = apply_rule(data, curr_rule)
+                    break
+        # 最后按 orders 顺序拼接
+        for ord_selector, ord_id in orders:
+            if ord_selector in ordered_cache:
+                paragraphs_str += ordered_cache[ord_selector]
+
+        paragraphs_str += "\n\n"
+
+    return paragraphs_str, refl_list

novel_downloader/core/parsers/qidian_parser/browser/chapter_normal.py

@@ -0,0 +1,97 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+novel_downloader.core.parsers.qidian_parser.browser.chapter_normal
+------------------------------------------------------------------
+
+Parser logic for extracting readable text from Qidian chapters
+that use plain (non-encrypted) browser-rendered HTML.
+"""
+
+import logging
+from typing import Any, Dict
+
+from bs4 import BeautifulSoup
+
+from ..shared import (
+    extract_chapter_info,
+    find_ssr_page_context,
+)
+
+logger = logging.getLogger(__name__)
+
+
+def parse_normal_chapter(
+    soup: BeautifulSoup,
+    chapter_id: str,
+) -> Dict[str, Any]:
+    """
+    Extract and format the chapter text from a normal Qidian page.
+    Returns empty string if VIP/encrypted.
+
+    This method performs the following steps:
+      1. Parses HTML into soup.
+      2. Skips parsing if VIP or encrypted.
+      3. Locates main content container.
+      4. Extracts SSR-rendered chapter info (title, author note).
+      5. Removes review spans.
+      6. Extracts paragraph texts and formats them.
+
+    :param html_str: Raw HTML content of the chapter page.
+    :return: Formatted chapter text or empty string if not parsable.
+    """
+    try:
+        main = soup.select_one("div#app div#reader-content main")
+        if not main:
+            logger.warning("[Parser] Main content not found for chapter")
+            return {}
+
+        ssr_data = find_ssr_page_context(soup)
+        chapter_info = extract_chapter_info(ssr_data)
+        if not chapter_info:
+            logger.warning(
+                "[Parser] ssr_chapterInfo not found for chapter '%s'", chapter_id
+            )
+            return {}
+
+        title = chapter_info.get("chapterName", "Untitled")
+        chapter_id = chapter_info.get("chapterId", "")
+        author_say = chapter_info.get("authorSay", "")
+        update_time = chapter_info.get("updateTime", "")
+        update_timestamp = chapter_info.get("updateTimestamp", 0)
+        modify_time = chapter_info.get("modifyTime", 0)
+        word_count = chapter_info.get("wordsCount", 0)
+        vip = bool(chapter_info.get("vipStatus", 0))
+        is_buy = bool(chapter_info.get("isBuy", 0))
+        seq = chapter_info.get("seq", None)
+        order = chapter_info.get("chapterOrder", None)
+        volume = chapter_info.get("extra", {}).get("volumeName", "")
+
+        # remove review spans
+        for span in main.select("span.review"):
+            span.decompose()
+
+        paras = [p.get_text(strip=True) for p in main.find_all("p")]
+        chapter_text = "\n\n".join(paras)
+
+        return {
+            "id": str(chapter_id),
+            "title": title,
+            "content": chapter_text,
+            "author_say": author_say.strip() if author_say else "",
+            "updated_at": update_time,
+            "update_timestamp": update_timestamp,
+            "modify_time": modify_time,
+            "word_count": word_count,
+            "vip": vip,
+            "purchased": is_buy,
+            "order": order,
+            "seq": seq,
+            "volume": volume,
+        }
+
+    except Exception as e:
+        logger.warning(
+            "[Parser] parse error for normal chapter '%s': %s", chapter_id, e
+        )
+        return {}

novel_downloader/core/parsers/qidian_parser/browser/chapter_router.py

@@ -0,0 +1,70 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+novel_downloader.core.parsers.qidian_parser.browser.chapter_router
+------------------------------------------------------------------
+
+Routing logic for selecting the correct chapter parser for Qidian browser pages.
+
+This module acts as a dispatcher that analyzes a chapter's HTML content and
+routes the parsing task to either the encrypted or normal chapter parser.
+"""
+
+from __future__ import annotations
+
+import logging
+from typing import TYPE_CHECKING, Any, Dict
+
+from ..shared import (
+    can_view_chapter,
+    html_to_soup,
+    is_encrypted,
+)
+from .chapter_normal import parse_normal_chapter
+
+if TYPE_CHECKING:
+    from .main_parser import QidianBrowserParser
+
+logger = logging.getLogger(__name__)
+
+
+def parse_chapter(
+    parser: QidianBrowserParser,
+    html_str: str,
+    chapter_id: str,
+) -> Dict[str, Any]:
+    """
+    Extract and return the formatted textual content of chapter.
+
+    :param parser: Instance of QidianBrowserParser.
+    :param html_str: Raw HTML content of the chapter page.
+    :param chapter_id: Identifier of the chapter being parsed.
+    :return: Formatted chapter text or empty string if not parsable.
+    """
+    try:
+        soup = html_to_soup(html_str)
+
+        if not can_view_chapter(soup):
+            logger.warning(
+                "[Parser] Chapter '%s' is not purchased or inaccessible.", chapter_id
+            )
+            return {}
+
+        if is_encrypted(soup):
+            if not parser._decode_font:
+                return {}
+            try:
+                from .chapter_encrypted import parse_encrypted_chapter
+
+                return parse_encrypted_chapter(parser, soup, chapter_id)
+            except ImportError:
+                logger.warning(
+                    "[Parser] Encrypted chapter '%s' requires extra dependencies.",
+                    chapter_id,
+                )
+                return {}
+
+        return parse_normal_chapter(soup, chapter_id)
+    except Exception as e:
+        logger.warning("[Parser] parse error for chapter '%s': %s", chapter_id, e)
+        return {}