codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat 下载器监控支持转移合集

Browse Source
This commit is contained in:
jxxghp
2023-08-23 08:47:03 +08:00
parent e5e33d4486
commit 66a1f25465
10 changed files with 155 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/modules/douban/__init__.py
View File

@ -167,7 +167,7 @@ class DoubanModule(_ModuleBase):
if settings.SCRAP_SOURCE != "douban":
return None
# 目录下的所有文件
for file in SystemUtils.list_files_with_extensions(path, settings.RMT_MEDIAEXT):
for file in SystemUtils.list_files(path, settings.RMT_MEDIAEXT):
if not file:
continue
logger.info(f"开始刮削媒体库文件：{file} ...")

6
app/modules/filetransfer/__init__.py
View File

@ -121,7 +121,7 @@ class FileTransferModule(_ModuleBase):
# 比对文件名并转移字幕
org_dir: Path = org_path.parent
file_list: List[Path] = SystemUtils.list_files_with_extensions(org_dir, settings.RMT_SUBEXT)
file_list: List[Path] = SystemUtils.list_files(org_dir, settings.RMT_SUBEXT)
if len(file_list) == 0:
logger.debug(f"{org_dir} 目录下没有找到字幕文件...")
else:
@ -207,7 +207,7 @@ class FileTransferModule(_ModuleBase):
"""
dir_name = org_path.parent
file_name = org_path.name
file_list: List[Path] = SystemUtils.list_files_with_extensions(dir_name, ['.mka'])
file_list: List[Path] = SystemUtils.list_files(dir_name, ['.mka'])
pending_file_list: List[Path] = [file for file in file_list if org_path.stem == file.stem]
if len(pending_file_list) == 0:
logger.debug(f"{dir_name} 目录下没有找到匹配的音轨文件")
@ -409,7 +409,7 @@ class FileTransferModule(_ModuleBase):
file_list_new=[])
else:
# 获取文件清单
transfer_files: List[Path] = SystemUtils.list_files_with_extensions(in_path, settings.RMT_MEDIAEXT)
transfer_files: List[Path] = SystemUtils.list_files(in_path, settings.RMT_MEDIAEXT)
if len(transfer_files) == 0:
return TransferInfo(message=f"{in_path} 目录下没有找到可转移的文件")
if not in_meta:

2
app/modules/qbittorrent/__init__.py
View File

@ -168,7 +168,7 @@ class QbittorrentModule(_ModuleBase):
logger.info(f"移动模式删除种子成功：{hashs} ")
# 删除残留文件
if transinfo.path and transinfo.path.exists():
files = SystemUtils.list_files_with_extensions(transinfo.path, settings.RMT_MEDIAEXT)
files = SystemUtils.list_files(transinfo.path, settings.RMT_MEDIAEXT)
if not files:
logger.warn(f"删除残留文件夹：{transinfo.path}")
shutil.rmtree(transinfo.path, ignore_errors=True)

2
app/modules/subtitle/__init__.py
View File

@ -108,7 +108,7 @@ class SubtitleModule(_ModuleBase):
# 解压文件
shutil.unpack_archive(zip_file, zip_path, format='zip')
# 遍历转移文件
for sub_file in SystemUtils.list_files_with_extensions(zip_path, settings.RMT_SUBEXT):
for sub_file in SystemUtils.list_files(zip_path, settings.RMT_SUBEXT):
target_sub_file = download_dir / sub_file.name
if target_sub_file.exists():
logger.info(f"字幕文件已存在：{target_sub_file}")

2
app/modules/themoviedb/__init__.py
View File

@ -190,7 +190,7 @@ class TheMovieDbModule(_ModuleBase):
if settings.SCRAP_SOURCE != "themoviedb":
return None
# 目录下的所有文件
for file in SystemUtils.list_files_with_extensions(path, settings.RMT_MEDIAEXT):
for file in SystemUtils.list_files(path, settings.RMT_MEDIAEXT):
if not file:
continue
logger.info(f"开始刮削媒体库文件：{file} ...")

2
app/modules/transmission/__init__.py
View File

@ -152,7 +152,7 @@ class TransmissionModule(_ModuleBase):
logger.info(f"移动模式删除种子成功：{hashs} ")
# 删除残留文件
if transinfo.path and transinfo.path.exists():
files = SystemUtils.list_files_with_extensions(transinfo.path, settings.RMT_MEDIAEXT)
files = SystemUtils.list_files(transinfo.path, settings.RMT_MEDIAEXT)
if not files:
logger.warn(f"删除残留文件夹：{transinfo.path}")
shutil.rmtree(transinfo.path, ignore_errors=True)