GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-07-09 22:43:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix transfer

Browse Source
This commit is contained in:
jxxghp
2024-09-19 20:34:26 +08:00
parent f3b2bbfb6f
commit 4016295696
3 changed files with 36 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/chain/storage.py
View File

@@ -28,11 +28,11 @@ class StorageChain(ChainBase):
"""
return self.run_module("check_login", storage=storage, **kwargs)
def list_files(self, fileitem: schemas.FileItem) -> Optional[List[schemas.FileItem]]:
def list_files(self, fileitem: schemas.FileItem, recursion: bool = False) -> Optional[List[schemas.FileItem]]:
"""
查询当前目录下所有目录和文件
"""
return self.run_module("list_files", fileitem=fileitem)
return self.run_module("list_files", fileitem=fileitem, recursion=recursion)
def create_folder(self, fileitem: schemas.FileItem, name: str) -> Optional[schemas.FileItem]:
"""

34
app/chain/transfer.py
View File

@@ -191,27 +191,14 @@ class TransferChain(ChainBase):
# 跳过数量
skip_num = 0
# 目录所有文件清单
transfer_files = self.storagechain.list_files(fileitem=fileitem)
if transfer_files:
# 过滤后缀和大小
transfer_files = [f for f in transfer_files
if (f".{f.extension.lower()}" in self.all_exts
and (not min_filesize or f.size > min_filesize * 1024 * 1024))]
if formaterHandler:
# 有集自定义格式，过滤文件
transfer_files = [f for f in transfer_files if formaterHandler.match(f.name)]
else:
return False, f"{fileitem.name} 没有找到可转移的媒体文件"
# 总文件数
total_num = len(transfer_files)
self.progress.update(value=0,
text=f"开始转移 {fileitem.path}，共 {total_num} 个文件 ...",
key=ProgressKey.FileTransfer)
# 获取待转移路径清单
trans_items = self.__get_trans_fileitems(fileitem)
# 总文件数
total_num = len(trans_items)
self.progress.update(value=0,
text=f"开始转移 {fileitem.path}，共 {total_num} 个文件或子目录 ...",
key=ProgressKey.FileTransfer)
if not trans_items:
logger.warn(f"{fileitem.path} 没有找到可转移的媒体文件")
return False, f"{fileitem.name} 没有找到可转移的媒体文件"
@@ -231,15 +218,16 @@ class TransferChain(ChainBase):
# 如果是目录且不是⼀蓝光原盘，获取所有文件并转移
if (trans_item.type == "dir"
and not (trans_item.storage == "local" and not SystemUtils.is_bluray_dir(item_path))):
# 遍历获取下载目录所有文件
file_items = self.storagechain.list_files(trans_item)
# 遍历获取下载目录所有文件（递归）
file_items = self.storagechain.list_files(trans_item, recursion=True)
if not file_items:
continue
# 过滤后缀和大小
file_items = [f for f in file_items
if (f".{f.extension.lower()}" in self.all_exts
and (not min_filesize or f.size > min_filesize * 1024 * 1024))]
if f.extension and (f".{f.extension.lower()}" in self.all_exts
and (not min_filesize or f.size > min_filesize * 1024 * 1024))]
else:
# 文件或蓝光目录
file_items = [trans_item]
if formaterHandler: