kmoe-manga-downloader 1.1.1__tar.gz → 1.2.0__tar.gz

{kmoe_manga_downloader-1.1.1 → kmoe_manga_downloader-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kmoe-manga-downloader
-Version: 1.1.1
+Version: 1.2.0
 Summary: A CLI-downloader for site @kox.moe.
 Author-email: Chris Zheng <chrisis58@outlook.com>
 License: MIT License
@@ -35,25 +35,31 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: beautifulsoup4<5.0.0,>=4.11.1
-Requires-Dist: requests<3.0.0,>=2.32.3
-Requires-Dist: tqdm<5.0.0,>=4.67.1
-Requires-Dist: cloudscraper<2.0.0,>=1.2.71
+Requires-Dist: deprecation~=2.1.0
+Requires-Dist: beautifulsoup4~=4.13.4
+Requires-Dist: aiohttp~=3.12.15
+Requires-Dist: aiofiles~=24.1.0
+Requires-Dist: async-lru~=2.0.5
+Requires-Dist: rich~=13.9.4
 Dynamic: license-file
 
 # Kmoe Manga Downloader
 
 [![PyPI Downloads](https://static.pepy.tech/badge/kmoe-manga-downloader)](https://pepy.tech/projects/kmoe-manga-downloader) [![PyPI version](https://img.shields.io/pypi/v/kmoe-manga-downloader.svg)](https://pypi.org/project/kmoe-manga-downloader/) [![Unit Tests](https://github.com/chrisis58/kmdr/actions/workflows/unit-test.yml/badge.svg)](https://github.com/chrisis58/kmdr/actions/workflows/unit-test.yml) [![Interpretor](https://img.shields.io/badge/python-3.9+-blue)](https://www.python.org/) [![License](https://img.shields.io/badge/License-MIT-green)](https://github.com/chrisis58/kmdr/blob/main/LICENSE)
 
-`kmdr (Kmoe Manga Downloader)` 是一个 Python 应用，用于从 [Kmoe](https://kox.moe/) 网站下载漫画。它支持在终端环境下的登录、下载指定书籍及其卷，并支持回调脚本执行。
+`kmdr (Kmoe Manga Downloader)` 是一个 Python 应用，用于从 [Kmoe](https://kox.moe/) 网站下载漫画。它支持在终端环境下的登录、下载指定漫画及其卷，并支持回调脚本执行。
+
+<p align="center">
+  <img src="assets/kmdr-demo.gif" alt="kmdr 使用演示" width="720">
+</p>
 
 ## ✨功能特性
 
-- **凭证管理**: 命令行登录并持久化会话
-- **多种下载方式**: 支持通过书籍 URL 或从收藏列表下载
-- **高效下载**: 支持多线程、失败重试及断点续传
-- **配置持久化**: 保存常用下载目录、代理等设置
-- **回调支持**: 下载完成后自动执行自定义脚本
+- **现代化终端界面**: 使用 [rich](https://github.com/Textualize/rich) 构建的终端用户界面（TUI），提供进度条和菜单等现代化、美观的交互式终端界面。
+- **凭证和配置管理**: 应用自动维护登录凭证和下载设置，实现一次配置、持久有效，提升使用效率。
+- **高效下载的性能**:  采用 `asyncio` 并发分片下载技术，充分利用网络带宽，极大加速单个大文件的下载速度。
+- **强大的高可用性**: 内置强大的自动重试与断点续传机制，无惧网络中断，确保下载任务最终成功。
+- **灵活的自动化接口**: 支持下载完成后自动执行自定义回调脚本，轻松集成到您的自动化流程。
 
 ## 🛠️安装应用
 

{kmoe_manga_downloader-1.1.1 → kmoe_manga_downloader-1.2.0}/README.md

@@ -2,15 +2,19 @@
 
 [![PyPI Downloads](https://static.pepy.tech/badge/kmoe-manga-downloader)](https://pepy.tech/projects/kmoe-manga-downloader) [![PyPI version](https://img.shields.io/pypi/v/kmoe-manga-downloader.svg)](https://pypi.org/project/kmoe-manga-downloader/) [![Unit Tests](https://github.com/chrisis58/kmdr/actions/workflows/unit-test.yml/badge.svg)](https://github.com/chrisis58/kmdr/actions/workflows/unit-test.yml) [![Interpretor](https://img.shields.io/badge/python-3.9+-blue)](https://www.python.org/) [![License](https://img.shields.io/badge/License-MIT-green)](https://github.com/chrisis58/kmdr/blob/main/LICENSE)
 
-`kmdr (Kmoe Manga Downloader)` 是一个 Python 应用，用于从 [Kmoe](https://kox.moe/) 网站下载漫画。它支持在终端环境下的登录、下载指定书籍及其卷，并支持回调脚本执行。
+`kmdr (Kmoe Manga Downloader)` 是一个 Python 应用，用于从 [Kmoe](https://kox.moe/) 网站下载漫画。它支持在终端环境下的登录、下载指定漫画及其卷，并支持回调脚本执行。
+
+<p align="center">
+  <img src="assets/kmdr-demo.gif" alt="kmdr 使用演示" width="720">
+</p>
 
 ## ✨功能特性
 
-- **凭证管理**: 命令行登录并持久化会话
-- **多种下载方式**: 支持通过书籍 URL 或从收藏列表下载
-- **高效下载**: 支持多线程、失败重试及断点续传
-- **配置持久化**: 保存常用下载目录、代理等设置
-- **回调支持**: 下载完成后自动执行自定义脚本
+- **现代化终端界面**: 使用 [rich](https://github.com/Textualize/rich) 构建的终端用户界面（TUI），提供进度条和菜单等现代化、美观的交互式终端界面。
+- **凭证和配置管理**: 应用自动维护登录凭证和下载设置，实现一次配置、持久有效，提升使用效率。
+- **高效下载的性能**:  采用 `asyncio` 并发分片下载技术，充分利用网络带宽，极大加速单个大文件的下载速度。
+- **强大的高可用性**: 内置强大的自动重试与断点续传机制，无惧网络中断，确保下载任务最终成功。
+- **灵活的自动化接口**: 支持下载完成后自动执行自定义回调脚本，轻松集成到您的自动化流程。
 
 ## 🛠️安装应用
 

{kmoe_manga_downloader-1.1.1 → kmoe_manga_downloader-1.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kmoe-manga-downloader"
-version = "1.1.1"
+version = "1.2.0"
 authors = [
     { name="Chris Zheng", email="chrisis58@outlook.com" },
 ]
@@ -17,10 +17,12 @@ classifiers = [
 ]
 
 dependencies = [
-    "beautifulsoup4>=4.11.1,<5.0.0",
-    "requests>=2.32.3,<3.0.0",
-    "tqdm>=4.67.1,<5.0.0",
-    "cloudscraper>=1.2.71,<2.0.0",
+    "deprecation~=2.1.0",
+    "beautifulsoup4~=4.13.4",
+    "aiohttp~=3.12.15",
+    "aiofiles~=24.1.0",
+    "async-lru~=2.0.5",
+    "rich~=13.9.4"
 ]
 
 [project.urls]

{kmoe_manga_downloader-1.1.1 → kmoe_manga_downloader-1.2.0}/src/kmdr/core/__init__.py

@@ -1,7 +1,9 @@
 from .bases import Authenticator, Lister, Picker, Downloader, Configurer
 from .structure import VolInfo, BookInfo, VolumeType
-from .bases import AUTHENTICATOR, LISTERS, PICKERS, DOWNLOADER, CONFIGURER
+from .bases import AUTHENTICATOR, LISTERS, PICKERS, DOWNLOADER, CONFIGURER, KMDR_SESSION
 
-from .defaults import argument_parser
+from .defaults import argument_parser, session_var
 
-from .error import KmdrError, LoginError
+from .error import KmdrError, LoginError
+
+from .session import KmdrSession

kmoe_manga_downloader-1.2.0/src/kmdr/core/bases.py

@@ -0,0 +1,97 @@
+from typing import Callable, Optional
+from abc import abstractmethod
+
+import asyncio
+from aiohttp import ClientSession
+
+from .error import LoginError
+from .registry import Registry
+from .structure import VolInfo, BookInfo
+from .utils import construct_callback
+
+from .context import TerminalContext, SessionContext, UserProfileContext, ConfigContext
+
+class Configurer(ConfigContext, TerminalContext):
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+    
+    @abstractmethod
+    def operate(self) -> None: ...
+
+class Authenticator(SessionContext, ConfigContext, UserProfileContext, TerminalContext):
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    # 在使用代理登录时，可能会出现问题，但是现在还不清楚是不是代理的问题。
+    # 主站正常情况下不使用代理也能登录成功。但是不排除特殊的网络环境下需要代理。
+    # 所以暂时保留代理登录的功能，如果后续确认是代理的问题，可以考虑启用 @no_proxy 装饰器。
+    # @no_proxy
+    async def authenticate(self) -> None:
+        with self._console.status("认证中..."):
+            try:
+                assert await self._authenticate()
+            except LoginError as e:
+                self._console.print("[red]认证失败。请检查您的登录凭据或会话 cookie。[/red]")
+                self._console.print(f"[yellow]详细信息：{e}[/yellow]")
+                exit(1)
+
+    @abstractmethod
+    async def _authenticate(self) -> bool: ...
+
+class Lister(SessionContext, TerminalContext):
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    @abstractmethod
+    async def list(self) -> tuple[BookInfo, list[VolInfo]]: ...
+
+class Picker(TerminalContext):
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    @abstractmethod
+    def pick(self, volumes: list[VolInfo]) -> list[VolInfo]: ...
+
+class Downloader(SessionContext, UserProfileContext, TerminalContext):
+
+    def __init__(self,
+                 dest: str = '.',
+                 callback: Optional[str] = None,
+                 retry: int = 3,
+                 num_workers: int = 8,
+                 *args, **kwargs
+    ):
+        super().__init__(*args, **kwargs)
+
+        self._dest: str = dest
+        self._callback: Optional[Callable] = construct_callback(callback)
+        self._retry: int = retry
+        self._semaphore = asyncio.Semaphore(num_workers)
+
+    async def download(self, book: BookInfo, volumes: list[VolInfo]):
+        if not volumes:
+            self._console.print("No volumes to download.")
+            exit(0)
+
+        try:
+            with self._progress:
+                tasks = [self._download(book, volume) for volume in volumes]
+                await asyncio.gather(*tasks, return_exceptions=True)
+
+        except KeyboardInterrupt:
+            self._console.print("\n操作已取消（KeyboardInterrupt）")
+            exit(130)
+
+    @abstractmethod
+    async def _download(self, book: BookInfo, volume: VolInfo): ...
+
+KMDR_SESSION = Registry[ClientSession]('KmdrSession', True)
+AUTHENTICATOR = Registry[Authenticator]('Authenticator')
+LISTERS = Registry[Lister]('Lister')
+PICKERS = Registry[Picker]('Picker')
+DOWNLOADER = Registry[Downloader]('Downloader', True)
+CONFIGURER = Registry[Configurer]('Configurer')

kmoe_manga_downloader-1.2.0/src/kmdr/core/context.py

@@ -0,0 +1,28 @@
+from aiohttp import ClientSession
+
+from .defaults import Configurer as InnerConfigurer, UserProfile, session_var, progress, console
+
+class TerminalContext:
+
+    def __init__(self, *args, **kwargs):
+        super().__init__()
+        self._progress = progress
+        self._console = console
+
+class SessionContext:
+
+    def __init__(self, *args, **kwargs):
+        super().__init__()
+        self._session: ClientSession = session_var.get()
+
+class UserProfileContext:
+
+    def __init__(self, *args, **kwargs):
+        super().__init__()
+        self._profile = UserProfile()
+
+class ConfigContext:
+
+    def __init__(self, *args, **kwargs):
+        super().__init__()
+        self._configurer = InnerConfigurer()

{kmoe_manga_downloader-1.1.1 → kmoe_manga_downloader-1.2.0}/src/kmdr/core/defaults.py

@@ -2,10 +2,46 @@ import os
 import json
 from typing import Optional
 import argparse
+from contextvars import ContextVar
+from rich.console import Console
+from rich.progress import (
+    Progress,
+    BarColumn,
+    DownloadColumn,
+    TextColumn,
+    TransferSpeedColumn,
+    TimeRemainingColumn,
+)
 
 from .utils import singleton
 from .structure import Config
 
+HEADERS = {
+    'User-Agent': 'kmdr/1.0 (https://github.com/chrisis58/kmoe-manga-downloader)'
+}
+
+console = Console()
+
+def console_print(*args, **kwargs):
+    console.print(*args, **kwargs)
+
+progress = Progress(
+    TextColumn("[blue]{task.fields[filename]}", justify="left"),
+    TextColumn("{task.fields[status]}", justify="right"),
+    TextColumn("{task.percentage:>3.1f}%"),
+    BarColumn(bar_width=None),
+    "[progress.percentage]",
+    DownloadColumn(),
+    "[",
+    TransferSpeedColumn(),
+    ",",
+    TimeRemainingColumn(),
+    "]",
+    console=console,
+)
+
+session_var = ContextVar('session')
+
 parser: Optional[argparse.ArgumentParser] = None
 args: Optional[argparse.Namespace] = None
 
@@ -14,33 +50,33 @@ def argument_parser():
     if parser is not None:
         return parser
 
-    parser = argparse.ArgumentParser(description='Kox Downloader')
-    subparsers = parser.add_subparsers(title='subcommands', dest='command')
-
-    download_parser = subparsers.add_parser('download', help='Download books')
-    download_parser.add_argument('-d', '--dest', type=str, help='Download destination, default to current directory', required=False)
-    download_parser.add_argument('-l', '--book-url', type=str, help='Book page\'s url', required=False)
-    download_parser.add_argument('-v', '--volume', type=str, help='Volume(s), split using commas, `all` for all', required=False)
-    download_parser.add_argument('-t', '--vol-type', type=str, help='Volume type, `vol` for volume, `extra` for extras, `seri` for serialized', required=False, choices=['vol', 'extra', 'seri', 'all'], default='vol')
-    download_parser.add_argument('--max-size', type=float, help='Max size of volume in MB', required=False)
-    download_parser.add_argument('--limit', type=int, help='Limit number of volumes to download', required=False)
-    download_parser.add_argument('--num-workers', type=int, help='Number of workers to use for downloading', required=False)
-    download_parser.add_argument('-p', '--proxy', type=str, help='Proxy server', required=False)
-    download_parser.add_argument('-r', '--retry', type=int, help='Retry times', required=False)
-    download_parser.add_argument('-c', '--callback', type=str, help='Callback script, use as `echo {v.name} downloaded!`', required=False)
-
-    login_parser = subparsers.add_parser('login', help='Login to kox.moe')
-    login_parser.add_argument('-u', '--username', type=str, help='Your username', required=True)
-    login_parser.add_argument('-p', '--password', type=str, help='Your password', required=False)
-
-    status_parser = subparsers.add_parser('status', help='Show status of account and script')
-    status_parser.add_argument('-p', '--proxy', type=str, help='Proxy server', required=False)
-
-    config_parser = subparsers.add_parser('config', help='Configure the downloader')
-    config_parser.add_argument('-l', '--list-option', action='store_true', help='List all configurations')
-    config_parser.add_argument('-s', '--set', nargs='+', type=str, help='Configuration options to set, e.g. num_workers=3 dest=.')
-    config_parser.add_argument('-c', '--clear', type=str, help='Clear configurations, `all`, `cookie`, `option` are available')
-    config_parser.add_argument('-d', '--delete', '--unset', dest='unset', type=str, help='Delete a specific configuration option')
+    parser = argparse.ArgumentParser(description='Kmoe 漫画下载器')
+    subparsers = parser.add_subparsers(title='可用的子命令', dest='command')
+
+    download_parser = subparsers.add_parser('download', help='下载指定的漫画')
+    download_parser.add_argument('-d', '--dest', type=str, help='指定下载文件的保存路径，默认为当前目录', required=False)
+    download_parser.add_argument('-l', '--book-url', type=str, help='漫画详情页面的 URL', required=False)
+    download_parser.add_argument('-v', '--volume', type=str, help='指定下载的卷，多个用逗号分隔，例如 `1,2,3` 或 `1-5,8`，`all` 表示全部', required=False)
+    download_parser.add_argument('-t', '--vol-type', type=str, help='指定下载的卷类型，`vol` 为单行本, `extra` 为番外, `seri` 为连载', required=False, choices=['vol', 'extra', 'seri', 'all'], default='vol')
+    download_parser.add_argument('--max-size', type=float, help='限制下载卷的最大体积 (单位: MB)', required=False)
+    download_parser.add_argument('--limit', type=int, help='限制下载卷的总数量', required=False)
+    download_parser.add_argument('--num-workers', type=int, help='下载时使用的并发任务数', required=False)
+    download_parser.add_argument('-p', '--proxy', type=str, help='设置下载使用的代理服务器', required=False)
+    download_parser.add_argument('-r', '--retry', type=int, help='网络请求失败时的重试次数', required=False)
+    download_parser.add_argument('-c', '--callback', type=str, help='每个卷下载完成后执行的回调脚本，例如: `echo {v.name} downloaded!`', required=False)
+
+    login_parser = subparsers.add_parser('login', help='登录到 Kmoe')
+    login_parser.add_argument('-u', '--username', type=str, help='用户名', required=True)
+    login_parser.add_argument('-p', '--password', type=str, help='密码 (如果留空，应用将提示您输入)', required=False)
+
+    status_parser = subparsers.add_parser('status', help='显示账户信息以及配额')
+    status_parser.add_argument('-p', '--proxy', type=str, help='代理服务器', required=False)
+
+    config_parser = subparsers.add_parser('config', help='配置下载器')
+    config_parser.add_argument('-l', '--list-option', action='store_true', help='列出所有配置')
+    config_parser.add_argument('-s', '--set', nargs='+', type=str, help='设置一个或多个配置项，格式为 `key=value`，例如: `num_workers=8`')
+    config_parser.add_argument('-c', '--clear', type=str, help='清除指定配置，可选值为 `all`, `cookie`, `option`')
+    config_parser.add_argument('-d', '--delete', '--unset', dest='unset', type=str, help='删除特定的配置选项')
 
     return parser
 
@@ -144,7 +180,7 @@ class Configurer:
         elif key == 'option':
             self._config.option = None
         else:
-            raise ValueError(f"Unsupported clear option: {key}")
+            raise KeyError(f"[red]对应配置不存在: {key}。可用配置项：all, cookie, option[/red]")
 
         self.update()
     

{kmoe_manga_downloader-1.1.1 → kmoe_manga_downloader-1.2.0}/src/kmdr/core/error.py

@@ -5,7 +5,7 @@ class KmdrError(RuntimeError):
         super().__init__(message, *args, **kwargs)
         self.message = message
 
-        self._solution = "" if solution is None else "\nSuggested Solution: \n" + "\n".join(f">>> {sol}" for sol in solution)
+        self._solution = "" if solution is None else "\n[bold cyan]推荐解决方法:[/bold cyan] \n" + "\n".join(f"[cyan]>>> {sol}[/cyan]" for sol in solution)
 
 class LoginError(KmdrError):
     def __init__(self, message, solution: Optional[list[str]] = None):

kmoe_manga_downloader-1.2.0/src/kmdr/core/session.py

@@ -0,0 +1,16 @@
+from typing import Optional
+
+from aiohttp import ClientSession
+
+from .bases import KMDR_SESSION
+from .defaults import session_var, HEADERS
+
+@KMDR_SESSION.register()
+class KmdrSession(ClientSession):
+    """
+    Kmdr 的 HTTP 会话管理类，支持从参数中初始化。简化 ClientSession 的使用。
+    """
+
+    def __init__(self, proxy: Optional[str] = None, *args, **kwargs):
+        ClientSession.__init__(self, proxy=proxy, trust_env=True, headers=HEADERS)
+        session_var.set(self)

kmoe_manga_downloader-1.2.0/src/kmdr/core/utils.py

@@ -0,0 +1,70 @@
+import functools
+from typing import Optional, Callable
+import asyncio
+
+import aiohttp
+
+from deprecation import deprecated
+import subprocess
+
+from .structure import BookInfo, VolInfo
+
+
+def singleton(cls):
+    """
+    **非线程安全**的单例装饰器
+    """
+
+    instances = {}
+
+    def get_instance(*args, **kwargs):
+        if cls not in instances:
+            instances[cls] = cls(*args, **kwargs)
+        return instances[cls]
+
+    return get_instance
+
+def construct_callback(callback: Optional[str]) -> Optional[Callable]:
+    if callback is None or not isinstance(callback, str) or not callback.strip():
+        return None
+
+    def _callback(book: BookInfo, volume: VolInfo) -> int:
+        nonlocal callback
+
+        assert callback, "Callback script cannot be empty"
+        formatted_callback = callback.strip().format(b=book, v=volume)
+
+        return subprocess.run(formatted_callback, shell=True, check=True).returncode
+
+    return _callback
+
+
+def async_retry(
+    attempts: int = 3,
+    delay: float = 1.0,
+    backoff: float = 2.0,
+    retry_on_status: set[int] = {500, 502, 503, 504, 429, 408}
+):
+    def decorator(func):
+        @functools.wraps(func)
+        async def wrapper(*args, **kwargs):
+            current_delay = delay
+            for attempt in range(attempts):
+                try:
+                    return await func(*args, **kwargs)
+                except aiohttp.ClientResponseError as e:
+                    if e.status in retry_on_status:
+                        if attempt == attempts - 1:
+                            raise
+                    else:
+                        raise
+                except (aiohttp.ClientError, asyncio.TimeoutError) as e:
+                    # 对于所有其他 aiohttp 客户端异常和超时，进行重试
+                    if attempt == attempts - 1:
+                        raise
+                
+                await asyncio.sleep(current_delay)
+
+                current_delay *= backoff
+        return wrapper
+    return decorator

kmoe_manga_downloader-1.2.0/src/kmdr/main.py

@@ -0,0 +1,49 @@
+from typing import Callable
+from argparse import Namespace
+import asyncio
+
+from kmdr.core import *
+from kmdr.module import *
+
+async def main(args: Namespace, fallback: Callable[[], None] = lambda: print('NOT IMPLEMENTED!')) -> None:
+
+    if args.command == 'config':
+        CONFIGURER.get(args).operate()
+        return
+
+    async with KMDR_SESSION.get(args):
+
+        if args.command == 'login':
+            await AUTHENTICATOR.get(args).authenticate()
+
+        elif args.command == 'status':
+            await AUTHENTICATOR.get(args).authenticate()
+
+        elif args.command == 'download':
+            await AUTHENTICATOR.get(args).authenticate()
+
+            book, volumes = await LISTERS.get(args).list()
+
+            volumes = PICKERS.get(args).pick(volumes)
+
+            await DOWNLOADER.get(args).download(book, volumes)
+
+        else:
+            fallback()
+
+def main_sync(args: Namespace, fallback: Callable[[], None] = lambda: print('NOT IMPLEMENTED!')) -> None:
+    asyncio.run(main(args, fallback))
+
+def entry_point():
+    try:
+        parser = argument_parser()
+        args = parser.parse_args()
+
+        main_coro = main(args, lambda: parser.print_help())
+        asyncio.run(main_coro)
+    except KeyboardInterrupt:
+        print("\n操作已取消（KeyboardInterrupt）")
+        exit(130)
+
+if __name__ == '__main__':
+    entry_point()

{kmoe_manga_downloader-1.1.1 → kmoe_manga_downloader-1.2.0}/src/kmdr/module/authenticator/CookieAuthenticator.py

@@ -1,8 +1,11 @@
 from typing import Optional
 
+from yarl import URL
+
 from kmdr.core import Authenticator, AUTHENTICATOR, LoginError
 
 from .utils import check_status
+from .utils import PROFILE_URL
 
 @AUTHENTICATOR.register()
 class CookieAuthenticator(Authenticator):
@@ -14,15 +17,16 @@ class CookieAuthenticator(Authenticator):
         else:
             self._show_quota = False
 
-    def _authenticate(self) -> bool:
+    async def _authenticate(self) -> bool:
         cookie = self._configurer.cookie
         
         if not cookie:
-            raise LoginError("No cookie found, please login first.", ['kmdr login -u <username>'])
+            raise LoginError("无法找到 Cookie，请先完成登录。", ['kmdr login -u <username>'])
         
-        self._session.cookies.update(cookie)
-        return check_status(
+        self._session.cookie_jar.update_cookies(cookie, response_url=URL(PROFILE_URL))
+        return await check_status(
             self._session,
+            self._console,
             show_quota=self._show_quota,
             is_vip_setter=lambda value: setattr(self._profile, 'is_vip', value),
             level_setter=lambda value: setattr(self._profile, 'user_level', value),

{kmoe_manga_downloader-1.1.1 → kmoe_manga_downloader-1.2.0}/src/kmdr/module/authenticator/LoginAuthenticator.py

@@ -1,6 +1,7 @@
 from typing import Optional
 import re
-from getpass import getpass
+
+from rich.prompt import Prompt
 
 from kmdr.core import Authenticator, AUTHENTICATOR, LoginError
 
@@ -25,32 +26,35 @@ class LoginAuthenticator(Authenticator):
         self._show_quota = show_quota
 
         if password is None:
-            password = getpass("please input your password: ")
+            password = Prompt.ask("请输入密码", password=True, console=self._console)
 
         self._password = password
 
-    def _authenticate(self) -> bool:
-        
-        response = self._session.post(
+    async def _authenticate(self) -> bool:
+
+        async with self._session.post(
             url = 'https://kox.moe/login_do.php', 
             data = {
                 'email': self._username,
                 'passwd': self._password,
                 'keepalive': 'on'
             },
-        )
-        response.raise_for_status()
-        match = re.search(r'"\w+"', response.text)
-
-        if not match:
-            raise LoginError("Failed to extract authentication code from response.")
-        
-        code = match.group(0).split('"')[1]
-        if code != CODE_OK:
-            raise LoginError(f"Authentication failed with error code: {code} " + CODE_MAPPING.get(code, "Unknown error."))
-
-        if check_status(self._session, show_quota=self._show_quota):
-            self._configurer.cookie = self._session.cookies.get_dict()
-            return True
-        
-        return False
+        ) as response:
+
+            response.raise_for_status()
+            match = re.search(r'"\w+"', await response.text())
+
+            if not match:
+                raise LoginError("无法解析登录响应。")
+            
+            code = match.group(0).split('"')[1]
+            if code != CODE_OK:
+                raise LoginError(f"认证失败，错误代码：{code} " + CODE_MAPPING.get(code, "未知错误。"))
+
+            if await check_status(self._session, self._console, show_quota=self._show_quota):
+                cookie = self._session.cookie_jar.filter_cookies('https://kox.moe')
+                self._configurer.cookie = {key: morsel.value for key, morsel in cookie.items()}
+
+                return True
+            
+            return False