🚨 style: fix lint errors

Author: SigureMo

src/yutto/cli/settings.py

@@ -4,7 +4,7 @@
 import platform
 import sys
 from pathlib import Path
-from typing import Annotated, Any, Literal, Optional
+from typing import Annotated, Any, Literal
 
 from pydantic import BaseModel, Field
 
@@ -36,7 +36,7 @@ class YuttoBasicSettings(BaseModel):
     audio_quality: Annotated[AudioQuality, Field(30251)]
     vcodec: Annotated[str, Field("avc:copy")]
     acodec: Annotated[str, Field("mp4a:copy")]
-    download_vcodec_priority: Annotated[Optional[list[str]], Field(None)]  # noqa: UP007
+    download_vcodec_priority: Annotated[list[str] | None, Field(None)]
     output_format: Annotated[Literal["infer", "mp4", "mkv", "mov"], Field("infer")]
     output_format_audio_only: Annotated[
         Literal["infer", "m4a", "aac", "mp3", "flac", "mp4", "mkv", "mov"], Field("infer")
@@ -46,13 +46,13 @@ class YuttoBasicSettings(BaseModel):
     overwrite: Annotated[bool, Field(False)]
     proxy: Annotated[str, Field("auto")]
     dir: Annotated[str, Field("./")]
-    tmp_dir: Annotated[Optional[str], Field(None)]  # noqa: UP007
+    tmp_dir: Annotated[str | None, Field(None)]
     sessdata: Annotated[str, Field("")]
     subpath_template: Annotated[str, Field("{auto}")]
     aliases: Annotated[dict[str, str], Field(dict[str, str]())]
     metadata_format_premiered: Annotated[str, Field(TIME_DATE_FMT)]
     download_interval: Annotated[int, Field(0)]
-    banned_mirrors_pattern: Annotated[Optional[str], Field(None)]  # noqa: UP007
+    banned_mirrors_pattern: Annotated[str | None, Field(None)]
     vip_strict: Annotated[bool, Field(False)]
     login_strict: Annotated[bool, Field(False)]
     no_color: Annotated[bool, Field(False)]
@@ -72,7 +72,7 @@ class YuttoResourceSettings(BaseModel):
 
 
 class YuttoDanmakuSettings(BaseModel):
-    font_size: Annotated[Optional[int], Field(None)]  # noqa: UP007
+    font_size: Annotated[int | None, Field(None)]
     font: Annotated[str, Field("SimHei")]
     opacity: Annotated[float, Field(0.8)]
     display_region_ratio: Annotated[float, Field(1.0)]
@@ -89,8 +89,8 @@ class YuttoDanmakuSettings(BaseModel):
 
 class YuttoBatchSettings(BaseModel):
     with_section: Annotated[bool, Field(False)]
-    batch_filter_start_time: Annotated[Optional[str], Field(None)]  # noqa: UP007
-    batch_filter_end_time: Annotated[Optional[str], Field(None)]  # noqa: UP007
+    batch_filter_start_time: Annotated[str | None, Field(None)]
+    batch_filter_end_time: Annotated[str | None, Field(None)]
 
 
 class YuttoSettings(BaseModel):

src/yutto/mcp.py

@@ -3,7 +3,7 @@
 from collections.abc import AsyncIterator
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, cast
+from typing import TYPE_CHECKING, Any
 
 from fastmcp import Context, FastMCP
 from pydantic import Field
@@ -13,6 +13,7 @@
 
 if TYPE_CHECKING:
     from mcp.server.session import ServerSession
+    from mcp.shared.context import RequestContext
 
 
 @dataclass
@@ -47,15 +48,15 @@ def parse_args(url: str, dir: str):
 
 @mcp.tool()
 async def add_task(
-    ctx: Context,  # pyright: ignore[reportMissingTypeArgument, reportUnknownParameterType]
+    ctx: Context,
     url: str = Field(description="The URL to download, you can also use a short link like 'BV1CrfKYLEeP'"),
     dir: str = Field(description="The directory to save the downloaded file"),
 ) -> str:
     """
     Use this tool to download a video from Bilibili using the given URL or short link.
     """
-    ctx_typed = cast("Context[ServerSession, AppContext]", ctx)
-    download_manager: DownloadManager = ctx_typed.request_context.lifespan_context.download_manager
+    request_context: RequestContext[ServerSession, AppContext, Any] = ctx.request_context  # pyright: ignore[reportUnknownMemberType, reportUnknownVariableType]
+    download_manager: DownloadManager = request_context.lifespan_context.download_manager
     await download_manager.add_task(DownloadTask(args=parse_args(url, dir)))
     return "Task added"
 

src/yutto/utils/fetcher.py

@@ -145,7 +145,7 @@ async def fetch_text(
             Logger.debug(f"Fetch text: {url}")
             Logger.status.next_tick()
             resp = await client.get(url, params=params)
-            if resp.status_code != httpx.codes.OK:
+            if not resp.is_success:
                 return None
             return resp.text
 
@@ -162,7 +162,7 @@ async def fetch_bin(
             Logger.debug(f"Fetch bin: {url}")
             Logger.status.next_tick()
             resp = await client.get(url, params=params)
-            if resp.status_code != httpx.codes.OK:
+            if not resp.is_success:
                 return None
             return resp.read()
 
@@ -179,7 +179,7 @@ async def fetch_json(
             Logger.debug(f"Fetch json: {url}")
             Logger.status.next_tick()
             resp = await client.get(url, params=params)
-            if resp.status_code != httpx.codes.OK:
+            if not resp.is_success:
                 return None
             return resp.json()