novel-downloader 1.5.0__py3-none-any.whl → 2.0.1__py3-none-any.whl

novel_downloader/web/main.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python3
+"""
+novel_downloader.web.main
+-------------------------
+
+Novel Downloader web UI (NiceGUI).
+
+This entry point starts the local server and registers the app's pages.
+"""
+
+import argparse
+from pathlib import Path
+
+from nicegui import app, ui
+
+import novel_downloader.web.pages  # noqa: F401
+from novel_downloader.config import get_config_value
+from novel_downloader.utils.logger import setup_logging
+
+
+def mount_exports() -> None:
+    output_dir = get_config_value(["general", "output_dir"], "./downloads")
+    out = Path(output_dir).expanduser().resolve()
+    out.mkdir(parents=True, exist_ok=True)
+    # serves /download/<filename> from the export dir
+    app.add_static_files("/download", local_directory=out)
+
+
+def web_main() -> None:
+    p = argparse.ArgumentParser(
+        description="Novel Downloader web UI.",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+    p.add_argument(
+        "--listen",
+        choices=["local", "public"],
+        default="local",
+        help=(
+            "Bind address mode: 'local' binds to 127.0.0.1; "
+            "'public' binds to 0.0.0.0."
+        ),
+    )
+    p.add_argument(
+        "--port",
+        type=int,
+        default=8080,
+        help="TCP port to serve the app on.",
+    )
+    p.add_argument(
+        "--reload",
+        action="store_true",
+        help="Enable autoreload on code changes (development).",
+    )
+    args = p.parse_args()
+
+    host = "127.0.0.1" if args.listen == "local" else "0.0.0.0"
+
+    log_level = get_config_value(["general", "debug", "log_level"], "INFO")
+    setup_logging(console_level=log_level)
+
+    app.on_startup(mount_exports)
+    ui.run(host=host, port=args.port, reload=args.reload)
+
+
+if __name__ in {"__main__", "__mp_main__"}:
+    web_main()

novel_downloader/web/pages/__init__.py

@@ -0,0 +1,17 @@
+#!/usr/bin/env python3
+"""
+novel_downloader.web.pages
+--------------------------
+
+NiceGUI page registrations; importing this package exposes and registers all routes.
+"""
+
+__all__ = [
+    "page_download",  # /download
+    "page_progress",  # /progress
+    "page_search",  # /
+]
+
+from .download import page_download
+from .progress import page_progress
+from .search import page_search

novel_downloader/web/pages/download.py

@@ -0,0 +1,78 @@
+#!/usr/bin/env python3
+"""
+novel_downloader.web.pages.download
+-----------------------------------
+
+"""
+
+from nicegui import ui
+
+from novel_downloader.web.components import navbar
+from novel_downloader.web.services import manager, setup_dialog
+
+_SUPPORT_SITES = {
+    "aaatxt": "3A电子书 (aaatxt)",
+    "biquge": "笔趣阁 (biquge)",
+    "biquyuedu": "精彩小说 (biquyuedu)",
+    "dxmwx": "大熊猫文学网 (dxmwx)",
+    "eightnovel": "无限轻小说 (8novel)",
+    "esjzone": "ESJ Zone (esjzone)",
+    "guidaye": "名著阅读 (guidaye)",
+    "hetushu": "和图书 (hetushu)",
+    "i25zw": "25中文网 (i25zw)",
+    "ixdzs8": "爱下电子书 (ixdzs8)",
+    "jpxs123": "精品小说网 (jpxs123)",
+    "lewenn": "乐文小说网 (lewenn)",
+    "linovelib": "哔哩轻小说 (linovelib)",
+    "piaotia": "飘天文学网 (piaotia)",
+    "qbtr": "全本同人小说 (qbtr)",
+    "qianbi": "铅笔小说 (qianbi)",
+    "qidian": "起点中文网 (qidian)",
+    "quanben5": "全本小说网 (quanben5)",
+    "sfacg": "SF轻小说 (sfacg)",
+    "shencou": "神凑轻小说 (shencou)",
+    "shuhaige": "书海阁小说网 (shuhaige)",
+    "tongrenquan": "同人圈 (tongrenquan)",
+    "ttkan": "天天看小说 (ttkan)",
+    "wanbengo": "完本神站 (wanbengo)",
+    "xiaoshuowu": "小说屋 (xiaoshuowu)",
+    "xiguashuwu": "西瓜书屋 (xiguashuwu)",
+    "xs63b": "小说路上 (xs63b)",
+    "xshbook": "小说虎 (xshbook)",
+    "yamibo": "百合会 (yamibo)",
+    "yibige": "一笔阁 (yibige)",
+}
+_DEFAULT_SITE = "qidian"
+
+
+@ui.page("/download")  # type: ignore[misc]
+def page_download() -> None:
+    navbar("download")
+    ui.label("下载界面").classes("text-lg")
+    setup_dialog()
+
+    with ui.card().classes("max-w-[600px]"):
+        site = ui.select(
+            _SUPPORT_SITES,
+            value=_DEFAULT_SITE,
+            label="站点",
+            with_input=True,
+        ).classes("w-full")
+
+        book_id = ui.input("书籍ID").props("outlined dense").classes("w-full")
+
+        async def add_task() -> None:
+            bid = (book_id.value or "").strip()
+            if not bid:
+                ui.notify("请输入书籍ID", type="warning")
+                return
+            title = f"{site.value} (id = {bid})"
+            ui.notify(f"已添加任务: {title}")
+            await manager.add_task(title=title, site=str(site.value), book_id=bid)
+
+        with ui.row().classes("justify-end w-full"):
+            ui.button(
+                "添加到下载队列",
+                on_click=add_task,
+                color="primary",
+            ).props("unelevated")

novel_downloader/web/pages/progress.py

@@ -0,0 +1,147 @@
+#!/usr/bin/env python3
+"""
+novel_downloader.web.pages.progress
+-----------------------------------
+
+Layout for active/history tasks with compact cards and status chips.
+"""
+
+
+from nicegui import ui
+
+from novel_downloader.web.components import navbar
+from novel_downloader.web.services import DownloadTask, Status, manager, setup_dialog
+
+
+def _status_chip(status: Status) -> None:
+    label_map = {
+        "queued": "已排队",
+        "running": "运行中",
+        "completed": "完成",
+        "cancelled": "已取消",
+        "failed": "失败",
+    }
+    color_map = {
+        "queued": "warning",
+        "running": "primary",
+        "completed": "positive",
+        "cancelled": "grey-6",
+        "failed": "negative",
+    }
+    ui.chip(label_map[status]).props(
+        f"outline color={color_map[status]} dense"
+    ).classes("q-ml-sm")
+
+
+def _meta_row(label: str, value: str) -> None:
+    with ui.row().classes("items-center justify-between text-xs text-grey-7 w-full"):
+        ui.label(label)
+        ui.label(value)
+
+
+def _progress_block(t: DownloadTask) -> None:
+    # progress or summary depending on state
+    if t.status == "running":
+        if t.chapters_total <= 0:
+            label_text = f"{t.chapters_done}/? · 正在获取总章节..."
+            ui.linear_progress().props("indeterminate striped").classes("w-full")
+            ui.label(label_text).classes("text-xs text-grey-7")
+        else:
+            ui.linear_progress(value=t.progress()).props("instant-feedback").classes(
+                "w-full"
+            )
+            ui.label(f"{t.chapters_done}/{t.chapters_total} · running").classes(
+                "text-xs text-grey-7"
+            )
+    else:
+        suffix = {"completed": "完成", "cancelled": "已取消", "failed": "失败"}.get(
+            t.status, ""
+        )
+        if t.chapters_total > 0:
+            ui.label(f"{t.chapters_done}/{t.chapters_total} · {suffix}").classes(
+                "text-xs text-grey-7"
+            )
+        else:
+            ui.label(f"{t.chapters_done}/? · {suffix}").classes("text-xs text-grey-7")
+
+        if t.status == "completed" and t.exported_paths:
+            with ui.row().classes("w-full gap-2 mt-1"):
+                for key, p in t.exported_paths.items():
+                    url = f"/download/{p.name}?v={t.task_id}"
+                    ui.button(key, on_click=lambda e, url=url: ui.download(url)).props(
+                        "outline size=sm"
+                    )
+
+
+def _task_card(t: DownloadTask, *, active: bool) -> None:
+    with ui.card().classes("w-full"):
+        # header
+        with ui.row().classes("items-center justify-between w-full"):
+            with ui.row().classes("items-center gap-2"):
+                ui.label(t.title).classes("text-sm font-medium")
+                _status_chip(t.status)
+            if active and t.status in ("running", "queued"):
+
+                async def cancel_this(tid: str = t.task_id) -> None:
+                    ok = await manager.cancel_task(tid)
+                    ui.notify(
+                        f"任务 {tid[:8]} {'已取消' if ok else '取消失败'}",
+                        color=("primary" if ok else "negative"),
+                    )
+
+                ui.button("取消", on_click=cancel_this)
+            else:
+                ui.button(
+                    "取消",
+                    on_click=lambda: ui.notify("任务已结束，无法取消"),
+                ).props("disable")
+
+        # meta grid
+        with ui.column().classes("w-full gap-1 mt-2"):
+            _meta_row("站点", t.site)
+            _meta_row("书号", t.book_id)
+            if t.status == "failed" and t.error:
+                with ui.row().classes("items-start justify-between w-full"):
+                    ui.label("错误").classes("text-xs text-grey-7")
+                    ui.label(t.error).classes("text-xs text-negative q-ml-md")
+
+        # progress / summary
+        with ui.column().classes("w-full mt-2"):
+            _progress_block(t)
+
+
+@ui.page("/progress")  # type: ignore[misc]
+def page_progress() -> None:
+    navbar("progress")
+    ui.label("正在下载 / 历史记录").classes("text-lg")
+    setup_dialog()
+
+    @ui.refreshable  # type: ignore[misc]
+    def section() -> None:
+        s = manager.snapshot()
+
+        # Active first
+        ui.label("运行中 / 等待中").classes("text-base mt-2")
+        with ui.card().classes("w-full"):
+            running = s["running"]
+            pending = s["pending"]
+            if not running and not pending:
+                ui.label("暂无").classes("text-sm text-grey-6")
+            else:
+                if running:
+                    _task_card(running, active=True)
+                for t in pending:
+                    _task_card(t, active=True)
+
+        # History next
+        ui.label("已完成 / 已取消 / 失败").classes("text-base mt-4")
+        with ui.card().classes("w-full"):
+            if not s["completed"]:
+                ui.label("暂无").classes("text-sm text-grey-6")
+            else:
+                for t in s["completed"]:
+                    _task_card(t, active=False)
+
+    # periodic refresh
+    ui.timer(0.5, section.refresh)
+    section()

novel_downloader/web/pages/search.py

@@ -0,0 +1,329 @@
+#!/usr/bin/env python3
+"""
+novel_downloader.web.pages.search
+---------------------------------
+
+Search UI with a settings dropdown, persistent state, and paginated results.
+"""
+
+from __future__ import annotations
+
+import contextlib
+from collections.abc import Callable
+from math import ceil
+from typing import Any
+
+from nicegui import ui
+from nicegui.elements.number import Number
+from nicegui.events import ValueChangeEventArguments
+
+from novel_downloader.core import search
+from novel_downloader.models import SearchResult
+from novel_downloader.web.components import navbar
+from novel_downloader.web.services import manager, setup_dialog
+
+_SUPPORT_SITES = {
+    "aaatxt": "3A电子书",
+    "biquge": "笔趣阁",
+    "dxmwx": "大熊猫文学网",
+    "eightnovel": "无限轻小说",
+    "esjzone": "ESJ Zone",
+    "hetushu": "和图书",
+    "i25zw": "25中文网",
+    "ixdzs8": "爱下电子书",
+    "jpxs123": "精品小说网",
+    "piaotia": "飘天文学网",
+    "qbtr": "全本同人小说",
+    "qianbi": "铅笔小说",
+    "quanben5": "全本小说网",
+    "shuhaige": "书海阁小说网",
+    "tongrenquan": "同人圈",
+    "ttkan": "天天看小说",
+    # "wanbengo": "完本神站",
+    "xiaoshuowu": "小说屋",
+    "xiguashuwu": "西瓜书屋",
+    "xs63b": "小说路上",
+    # "xshbook": "小说虎",
+}
+
+_DEFAULT_TIMEOUT = 10.0
+_DEFAULT_SITE_LIMIT = 30
+_PAGE_SIZE = 20
+_PAGER_WIDTH = 9
+
+_STATE: dict[str, dict[str, Any]] = {}
+
+
+def _get_state() -> dict[str, Any]:
+    cid = ui.context.client.id
+    if cid not in _STATE:
+        _STATE[cid] = {
+            "query": "",
+            "sites": None,  # list[str] | None (None => search all)
+            "per_site_limit": _DEFAULT_SITE_LIMIT,
+            "timeout": _DEFAULT_TIMEOUT,
+            "results": [],  # list[SearchResult]
+            "page": 1,
+            "page_size": _PAGE_SIZE,
+        }
+    return _STATE[cid]
+
+
+def _cleanup_state() -> None:
+    cid = ui.context.client.id
+    _STATE.pop(cid, None)
+
+
+def _coerce_timeout(inp: Number) -> float:
+    v = inp.value
+    try:
+        v = float(v)
+        if v <= 0:
+            raise ValueError
+    except (TypeError, ValueError):
+        ui.notify("超时需 > 0 秒，已重置为 10.0", type="warning")
+        v = _DEFAULT_TIMEOUT
+    inp.set_value(v)
+    inp.sanitize()
+    return float(v)
+
+
+def _coerce_psl(inp: Number) -> int:
+    v = inp.value
+    try:
+        v = int(v)
+        if v <= 0:
+            raise ValueError
+    except (TypeError, ValueError):
+        ui.notify("单站条数上限需为正整数，已重置为 5", type="warning")
+        v = _DEFAULT_SITE_LIMIT
+    inp.set_value(v)
+    inp.sanitize()
+    return int(v)
+
+
+def _render_placeholder_cover() -> None:
+    with ui.element("div").classes(
+        "w-[72px] h-[96px] bg-grey-3 rounded-md flex items-center " "justify-center"
+    ):
+        ui.icon("book").classes("text-grey-6 text-3xl")
+
+
+def _render_result_row(r: SearchResult) -> None:
+    with (
+        ui.card().classes("w-full"),
+        ui.row().classes("items-start justify-between w-full gap-3"),
+    ):
+        cover = (r.get("cover_url") or "").strip()
+        if cover.startswith(("http://", "https://")):
+            ui.image(cover).classes("w-[72px] h-[96px] object-cover rounded-md")
+        else:
+            _render_placeholder_cover()
+
+        with ui.column().classes("gap-1 grow"):
+            ui.link(r["title"], r["book_url"], new_tab=True).classes(
+                "text-base font-medium"
+            )
+            ui.label(
+                f"{r['author']} · {r['word_count']} · 更新于 {r['update_date']}"
+            ).classes("text-xs text-grey-6")
+            ui.label(r["latest_chapter"]).classes("text-sm text-grey-7")
+            ui.label(f"{r['site']} · ID: {r['book_id']}").classes("text-xs text-grey-5")
+
+        async def _add_task() -> None:
+            title = r["title"]
+            ui.notify(f"已添加任务：{title}")
+            await manager.add_task(title=title, site=r["site"], book_id=r["book_id"])
+
+        ui.button("下载", color="primary", on_click=_add_task).props("unelevated")
+
+
+def _build_settings_dropdown(
+    state: dict[str, Any],
+) -> tuple[Callable[[], list[str] | None], Callable[[], int], Callable[[], float]]:
+    """
+    Create settings button + anchored menu with initial values from state.
+
+    Returns a tuple of getter functions:
+      * get_sites(): list of site keys, or None if none selected
+      * get_psl(): per-site limit (int)
+      * get_timeout(): timeout (float)
+    """
+    site_cbs: dict[str, Any] = {}
+
+    settings_btn = ui.button("设置").props("outline icon=settings")
+    with settings_btn:
+        menu = ui.menu().props("no-parent-event")
+        with menu:
+            ui.label("站点选择").classes("text-sm text-grey-7 q-mb-xs")
+
+            with ui.row().classes("gap-2"):
+
+                def _select_all() -> None:
+                    for cb in site_cbs.values():
+                        cb.set_value(True)
+
+                def _clear_all() -> None:
+                    for cb in site_cbs.values():
+                        cb.set_value(False)
+
+                ui.button("全选", on_click=_select_all).props("dense")
+                ui.button("清空", on_click=_clear_all).props("dense")
+
+            ui.separator()
+
+            with (
+                ui.scroll_area().classes("w-[300px] max-h-[260px] q-mt-xs"),
+                ui.column().classes("gap-1"),
+            ):
+                selected = set(state.get("sites") or [])
+                for key, label in _SUPPORT_SITES.items():
+                    site_cbs[key] = ui.checkbox(label, value=(key in selected))
+
+            ui.separator()
+            ui.label("高级设置").classes("text-sm text-grey-7 q-mt-sm")
+
+            psl_in = (
+                ui.number(
+                    "单站条数上限",
+                    value=state["per_site_limit"],
+                    min=1,
+                    step=1,
+                )
+                .without_auto_validation()
+                .classes("w-[180px]")
+            )
+            timeout_in = (
+                ui.number(
+                    "超时(秒)",
+                    value=state["timeout"],
+                    format="%.1f",
+                    min=0.1,
+                    step=0.1,
+                )
+                .without_auto_validation()
+                .classes("w-[180px]")
+            )
+
+    settings_btn.on("click", lambda: menu.open())
+
+    def _get_sites() -> list[str] | None:
+        chosen = [k for k, cb in site_cbs.items() if bool(cb.value)]
+        return chosen or None
+
+    def _get_psl() -> int:
+        val = _coerce_psl(psl_in)
+        state["per_site_limit"] = val
+        return val
+
+    def _get_timeout() -> float:
+        val = _coerce_timeout(timeout_in)
+        state["timeout"] = val
+        return val
+
+    return _get_sites, _get_psl, _get_timeout
+
+
+@ui.page("/")  # type: ignore[misc]
+def page_search() -> None:
+    navbar("search")
+    ui.label("搜索页面").classes("text-lg")
+    setup_dialog()
+
+    state = _get_state()
+
+    # settings (left) + query (middle) + search (right)
+    with ui.row().classes("items-center gap-2 my-2 w-full"):
+        get_sites, get_psl, get_timeout = _build_settings_dropdown(state)
+
+        query_in = (
+            ui.input("输入关键字", value=state["query"])
+            .props("outlined dense clearable")
+            .classes("min-w-[320px] grow")
+        )
+
+        search_btn = ui.button("搜索", color="primary").props("unelevated")
+
+    # results & pagination container
+    list_area = ui.column().classes("w-full")
+    pager_area = ui.row().classes("items-center justify-center w-full q-mt-md")
+
+    @ui.refreshable  # type: ignore[misc]
+    def render_results() -> None:
+        list_area.clear()
+        pager_area.clear()
+
+        results: list[SearchResult] = state["results"]
+        total = len(results)
+        page_size = int(state["page_size"])
+        total_pages = max(1, ceil(total / page_size))
+        page = max(1, min(int(state["page"]), total_pages))
+        state["page"] = page
+
+        start = (page - 1) * page_size
+        end = min(total, start + page_size)
+        current = results[start:end]
+
+        tip = (
+            f"共 {total} 条结果（第 {page}/{total_pages} 页）"
+            if state["sites"]
+            else f"共 {total} 条结果（第 {page}/{total_pages} 页，已搜索全部站点）"
+        )
+
+        with list_area:
+            ui.label(tip).classes("text-sm text-grey-7")
+            with ui.column().classes("w-full gap-2"):
+                for r in current:
+                    _render_result_row(r)
+
+        # pagination (only show if more than 1 page)
+        if total_pages > 1:
+
+            def _on_page_change(e: ValueChangeEventArguments) -> None:
+                try:
+                    state["page"] = int(e.value or 1)
+                except Exception:
+                    state["page"] = 1
+                render_results.refresh()
+
+            with pager_area:
+                ui.pagination(
+                    1,  # min
+                    total_pages,  # max
+                    direction_links=True,
+                    value=page,
+                    on_change=_on_page_change,
+                ).props(f"max-pages={_PAGER_WIDTH} boundary-numbers ellipses")
+
+    async def do_search() -> None:
+        q = (query_in.value or "").strip()
+        if not q:
+            ui.notify("请输入关键词", type="warning")
+            return
+
+        state["query"] = q
+        state["sites"] = get_sites()
+        per_site_limit = get_psl()
+        timeout_val = get_timeout()
+
+        # perform search
+        results = await search(
+            keyword=q,
+            sites=state["sites"],
+            limit=None,  # show all
+            per_site_limit=per_site_limit,
+            timeout=timeout_val,
+        )
+        state["results"] = results
+        state["page"] = 1
+        render_results.refresh()
+
+    search_btn.on("click", do_search)
+    query_in.on("keydown.enter", do_search)
+
+    # initial render
+    render_results()
+
+    # clean up state on disconnect to avoid leaks
+    with contextlib.suppress(Exception):
+        ui.context.client.on_disconnect(_cleanup_state)

novel_downloader/web/services/__init__.py

@@ -0,0 +1,17 @@
+#!/usr/bin/env python3
+"""
+novel_downloader.web.services
+-----------------------------
+
+Convenience re-exports for web UI services
+"""
+
+__all__ = [
+    "setup_dialog",
+    "manager",
+    "DownloadTask",
+    "Status",
+]
+
+from .client_dialog import setup_dialog
+from .task_manager import DownloadTask, Status, manager