XHS_Downloader/source/TUI/update.py

from rich.text import Text
from textual import work
from textual.app import ComposeResult
from textual.containers import Grid
from textual.screen import ModalScreen
from textual.widgets import Label
from textual.widgets import LoadingIndicator

from ..application import XHS
from ..module import (
    ERROR,
    WARNING,
    INFO,
    RELEASES,
)
from ..translation import _

__all__ = ["Update"]


class Update(ModalScreen):
    def __init__(self, app: XHS, ):
        super().__init__()
        self.xhs = app

    def compose(self) -> ComposeResult:
        yield Grid(
            Label(_("正在检查新版本，请稍等...")),
            LoadingIndicator(),
            classes="loading",
        )

    @work(exclusive=True)
    async def check_update(self) -> None:
        try:
            url = await self.xhs.html.request_url(RELEASES, False, None, timeout=5, )
            version = url.split("/")[-1]
            match self.compare_versions(f"{XHS.VERSION_MAJOR}.{XHS.VERSION_MINOR}", version, XHS.VERSION_BETA):
                case 4:
                    tip = Text(f"{_("检测到新版本：{0}.{1}").format(
                        XHS.VERSION_MAJOR, XHS.VERSION_MINOR)}\n{RELEASES}", style=WARNING)
                case 3:
                    tip = Text(
                        f"{_("当前版本为开发版, 可更新至正式版")}\n{RELEASES}",
                        style=WARNING)
                case 2:
                    tip = Text(
                        _("当前已是最新开发版"),
                        style=WARNING)
                case 1:
                    tip = Text(
                        _("当前已是最新正式版"),
                        style=INFO)
                case _:
                    raise ValueError
        except ValueError:
            tip = Text(_("检测新版本失败"), style=ERROR)
        self.dismiss(tip)

    def on_mount(self) -> None:
        self.check_update()

    @staticmethod
    def compare_versions(
            current_version: str,
            target_version: str,
            is_development: bool) -> int:
        current_major, current_minor = map(int, current_version.split('.'))
        target_major, target_minor = map(int, target_version.split('.'))

        if target_major > current_major:
            return 4
        if target_major == current_major:
            if target_minor > current_minor:
                return 4
            if target_minor == current_minor:
                return 3 if is_development else 1
        return 2