perf: 新增 folder_name 参数检查

This commit is contained in:
Quan
2025-08-01 11:15:16 +08:00
parent 079fe6c1f7
commit 1b48e70d44
3 changed files with 18 additions and 10 deletions

View File

@@ -137,6 +137,7 @@ class XHS:
author_archive,
write_mtime,
_print,
self.CLEANER,
)
self.mapping_data = mapping_data or {}
self.map_recorder = MapRecorder(

View File

@@ -16,6 +16,10 @@ from source.expansion import remove_empty_directories
from ..translation import _
from .static import HEADERS, USERAGENT, WARNING
from .tools import logging
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from ..expansion import Cleaner
__all__ = ["Manager"]
@@ -67,8 +71,10 @@ class Manager:
author_archive: bool,
write_mtime: bool,
_print: bool,
cleaner: "Cleaner",
):
self.root = root
self.cleaner = cleaner
self.temp = root.joinpath("Temp")
self.path = self.__check_path(path)
self.folder = self.__check_folder(folder)
@@ -130,8 +136,8 @@ class Manager:
return r if (r := self.__check_root_again(r)) else self.root
def __check_folder(self, folder: str) -> Path:
# TODO: 待实现
return self.path.joinpath(folder or "Download")
folder = self.cleaner.filter_name(folder, default="Download")
return self.path.joinpath(folder)
@staticmethod
def __check_root_again(root: Path) -> bool | Path: