XHS_Downloader/source/TUI/setting.py

from textual import on
from textual.app import ComposeResult
from textual.binding import Binding
from textual.containers import Container
from textual.containers import ScrollableContainer
from textual.screen import Screen
from textual.widgets import Button
from textual.widgets import Checkbox
from textual.widgets import Footer
from textual.widgets import Header
from textual.widgets import Input
from textual.widgets import Label
from textual.widgets import Select

from ..translation import _

__all__ = ["Setting"]


class Setting(Screen):
    BINDINGS = [
        Binding(key="Q", action="quit", description=_("退出程序")),
        Binding(key="B", action="index", description=_("返回首页")),
    ]

    def __init__(self, data: dict, ):
        super().__init__()
        self.data = data

    def compose(self) -> ComposeResult:
        yield Header()
        yield ScrollableContainer(
            Label(_("作品数据 / 文件保存根路径"), classes="params", ),
            Input(self.data["work_path"], placeholder=_("程序根路径"), valid_empty=True,
                  id="work_path", ),
            Label(_("作品文件储存文件夹名称"), classes="params", ),
            Input(self.data["folder_name"], placeholder="Download", id="folder_name", ),
            Label(_("作品文件名称格式"), classes="params", ),
            Input(self.data["name_format"], placeholder="发布时间 作者昵称 作品标题", valid_empty=True,
                  id="name_format", ),
            Label("User-Agent", classes="params", ),
            Input(self.data["user_agent"], placeholder=_("内置 Chrome User Agent"), valid_empty=True,
                  id="user_agent", ),
            Label(_("小红书网页版 Cookie"), classes="params", ),
            Input(placeholder=self.__check_cookie(), valid_empty=True, id="cookie", ),
            Label(_("网络代理"), classes="params", ),
            Input(self.data["proxy"], placeholder=_("不使用代理"), valid_empty=True, id="proxy", ),
            Label(_("请求数据超时限制，单位：秒"), classes="params", ),
            Input(str(self.data["timeout"]), placeholder="10", type="integer", id="timeout", ),
            Label(_("下载文件时，每次从服务器获取的数据块大小，单位：字节"), classes="params", ),
            Input(str(self.data["chunk"]), placeholder="1048576", type="integer", id="chunk", ),
            Label(_("请求数据失败时，重试的最大次数"), classes="params", ),
            Input(str(self.data["max_retry"]), placeholder="5", type="integer", id="max_retry", ),
            Label(),
            Container(
                Checkbox(_("记录作品详细数据"), id="record_data", value=self.data["record_data"], ),
                Checkbox(_("作品文件夹归档模式"), id="folder_mode", value=self.data["folder_mode"], ),
                Checkbox(_("视频作品下载开关"), id="video_download", value=self.data["video_download"], ),
                Checkbox(_("图文作品下载开关"), id="image_download", value=self.data["image_download"], ),
                classes="horizontal-layout"),
            Label(),
            Container(
                Checkbox(_("动图文件下载开关"), id="live_download", value=self.data["live_download"], ),
                Checkbox(_("作品下载记录开关"), id="download_record", value=self.data["download_record"], ),
                classes="horizontal-layout"),
            Container(
                Label(_("图片下载格式"), classes="params", ),
                Label(_("程序语言"), classes="params", ),
                classes="horizontal-layout",
            ),
            Label(),
            Container(
                Select.from_values(
                    ("PNG", "WEBP"),
                    value=self.data["image_format"].upper(),
                    allow_blank=False,
                    id="image_format"),
                Select.from_values(
                    ["zh_CN", "en_US"],
                    value=self.data["language"],
                    allow_blank=False,
                    id="language", ),
                classes="horizontal-layout"),
            Container(
                Button(_("保存配置"), id="save", ),
                Button(_("放弃更改"), id="abandon", ),
                classes="settings_button", ),
        )
        yield Footer()

    def __check_cookie(self) -> str:
        if self.data["cookie"]:
            return _("小红书网页版 Cookie，无需登录，参数已设置")
        return _("小红书网页版 Cookie，无需登录，参数未设置")

    def on_mount(self) -> None:
        self.title = _("程序设置")

    @on(Button.Pressed, "#save")
    def save_settings(self):
        self.dismiss({
            "work_path": self.query_one("#work_path").value,
            "folder_name": self.query_one("#folder_name").value,
            "name_format": self.query_one("#name_format").value,
            "user_agent": self.query_one("#user_agent").value,
            "cookie": self.query_one("#cookie").value or self.data["cookie"],
            "proxy": self.query_one("#proxy").value or None,
            "timeout": int(self.query_one("#timeout").value),
            "chunk": int(self.query_one("#chunk").value),
            "max_retry": int(self.query_one("#max_retry").value),
            "record_data": self.query_one("#record_data").value,
            "image_format": self.query_one("#image_format").value,
            "folder_mode": self.query_one("#folder_mode").value,
            "language": self.query_one("#language").value,
            "image_download": self.query_one("#image_download").value,
            "video_download": self.query_one("#video_download").value,
            "live_download": self.query_one("#live_download").value,
            "download_record": self.query_one("#download_record").value,
        })

    @on(Button.Pressed, "#abandon")
    def reset(self):
        self.dismiss(self.data)

    async def action_quit(self) -> None:
        await self.app.action_quit()

    async def action_index(self):
        await self.app.push_screen("index")