fix: enhance error handling and logging for subtitle download process

This commit is contained in:
jxxghp
2026-06-10 18:40:32 +08:00
parent cba52c57e6
commit 486c5294ba

View File

@@ -72,6 +72,29 @@ class DownloadChain(ChainBase):
"""
return Path(file_name).suffix.lower() in settings.RMT_SUBEXT
@classmethod
def _get_subtitle_working_dir(
cls,
storage_chain: StorageChain,
storage: str,
target_path: Path,
) -> Tuple[Optional[schemas.FileItem], str]:
"""
获取字幕保存目录,返回失败原因供前端展示。
"""
try:
working_dir_item = storage_chain.get_folder(storage, target_path)
except Exception as err:
message = f"下载目录获取失败,无法保存字幕:{target_path} - {str(err)}"
logger.error(message)
return None, message
if not working_dir_item:
message = f"下载目录不存在,无法保存字幕:{target_path}"
logger.error(message)
return None, message
return working_dir_item, ""
@staticmethod
def _detect_subtitle_fallback_name(subtitle: SubtitleInfo, content: bytes) -> str:
"""
@@ -115,26 +138,49 @@ class DownloadChain(ChainBase):
storage: str,
working_dir_item: schemas.FileItem,
subtitle_file: Path,
) -> Optional[str]:
) -> Tuple[Optional[str], str]:
"""
上传单个字幕文件到目标目录。
"""
target_sub_file = Path(working_dir_item.path) / subtitle_file.name
if storage_chain.get_file_item(storage, target_sub_file):
logger.info(f"字幕文件已存在:{target_sub_file}")
return target_sub_file.as_posix()
return target_sub_file.as_posix(), ""
logger.info(f"转移字幕 {subtitle_file}{target_sub_file} ...")
uploaded = storage_chain.upload_file(working_dir_item, subtitle_file)
if uploaded:
return uploaded.path
return None
return uploaded.path, ""
message = f"保存字幕文件失败:{target_sub_file}"
logger.error(message)
return None, message
@staticmethod
def _build_subtitle_download_error(response) -> str:
"""
从字幕下载响应中提取前端可展示的失败原因。
"""
status_code = getattr(response, "status_code", None)
reason = getattr(response, "reason", "") or ""
message = "下载字幕文件失败"
if status_code:
message = f"{message},状态码:{status_code}"
if reason:
message = f"{message} {reason}"
try:
response_text = (getattr(response, "text", "") or "").strip()
response_text = re.sub(r"\s+", " ", response_text)
if response_text:
message = f"{message}{response_text[:200]}"
except Exception as err:
logger.debug(f"读取字幕下载失败响应内容失败:{str(err)}")
return message
def _save_subtitle_response(
self,
subtitle: SubtitleInfo,
response,
target_dir: Path,
) -> List[str]:
) -> Tuple[bool, str, List[str]]:
"""
保存字幕下载响应到目标目录。
"""
@@ -147,23 +193,27 @@ class DownloadChain(ChainBase):
fallback_name=fallback_name,
)
if not self._is_subtitle_archive(file_name) and not self._is_subtitle_file(file_name):
logger.warn(f"下载链接不是支持的字幕文件:{subtitle.enclosure} - {file_name}")
return []
message = f"下载链接不是支持的字幕文件:{file_name}"
logger.warn(f"{message},链接:{subtitle.enclosure}")
return False, message, []
file_uri = FileURI.from_uri(target_dir.as_posix())
storage = file_uri.storage
target_path = Path(file_uri.path)
storage_chain = StorageChain()
working_dir_item = storage_chain.get_folder(storage, target_path)
working_dir_item, message = self._get_subtitle_working_dir(
storage_chain=storage_chain,
storage=storage,
target_path=target_path,
)
if not working_dir_item:
logger.error(f"下载目录不存在,无法保存字幕:{target_path}")
return []
return False, message, []
saved_files = []
settings.TEMP_PATH.mkdir(parents=True, exist_ok=True)
temp_file = settings.TEMP_PATH / file_name
temp_extract_dir = temp_file.with_name(temp_file.stem)
try:
settings.TEMP_PATH.mkdir(parents=True, exist_ok=True)
temp_file.write_bytes(response.content)
if self._is_subtitle_archive(file_name):
try:
@@ -173,10 +223,11 @@ class DownloadChain(ChainBase):
archive_format=self._subtitle_archive_format(file_name),
)
except Exception as err:
logger.error(f"字幕压缩包解压失败:{temp_file} - {str(err)}")
return []
message = f"字幕压缩包解压失败:{str(err)}"
logger.error(f"{message},文件:{temp_file}")
return False, message, []
for sub_file in SystemUtils.list_files(temp_extract_dir, settings.RMT_SUBEXT):
uploaded_path = self._upload_subtitle_file(
uploaded_path, message = self._upload_subtitle_file(
storage_chain=storage_chain,
storage=storage,
working_dir_item=working_dir_item,
@@ -184,8 +235,10 @@ class DownloadChain(ChainBase):
)
if uploaded_path:
saved_files.append(uploaded_path)
elif message:
logger.error(message)
else:
uploaded_path = self._upload_subtitle_file(
uploaded_path, message = self._upload_subtitle_file(
storage_chain=storage_chain,
storage=storage,
working_dir_item=working_dir_item,
@@ -193,7 +246,17 @@ class DownloadChain(ChainBase):
)
if uploaded_path:
saved_files.append(uploaded_path)
return saved_files
elif message:
return False, message, []
if not saved_files:
message = "未保存任何字幕文件"
logger.error(message)
return False, message, []
return True, "字幕文件保存成功", saved_files
except Exception as err:
message = f"保存字幕文件失败:{str(err)}"
logger.error(message)
return False, message, []
finally:
try:
if temp_extract_dir.exists():
@@ -245,17 +308,26 @@ class DownloadChain(ChainBase):
ua=subtitle.site_ua or settings.USER_AGENT,
proxies=settings.PROXY if subtitle.site_proxy else None,
)
response = request.get_res(subtitle.enclosure)
if not response or response.status_code != 200:
return False, "下载字幕文件失败", []
try:
response = request.get_res(subtitle.enclosure, raise_exception=True)
except Exception as err:
message = f"下载字幕文件失败:{str(err)}"
logger.error(message)
return False, message, []
if response is None:
return False, "下载字幕文件失败:未收到站点响应", []
if response.status_code != 200:
message = self._build_subtitle_download_error(response)
logger.error(message)
return False, message, []
saved_files = self._save_subtitle_response(
success, message, saved_files = self._save_subtitle_response(
subtitle=subtitle,
response=response,
target_dir=target_dir,
)
if not saved_files:
return False, "未保存任何字幕文件", []
if not success:
return False, message, []
logger.info(
f"{mediainfo.title_year} 字幕下载完成:{subtitle.site_name} - {subtitle.title},用户:{username}"