From dad980fa14293b221ac4eefa1c55d003cf7dcdbc Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Wed, 23 Oct 2024 17:05:20 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E9=99=84=E5=B1=9E=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E6=95=B4=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/modules/filemanager/__init__.py | 29 ++++++++++++++++++++---------
 1 file changed, 20 insertions(+), 9 deletions(-)

diff --git a/app/modules/filemanager/__init__.py b/app/modules/filemanager/__init__.py
index dd111445..3c833d9c 100644
--- a/app/modules/filemanager/__init__.py
+++ b/app/modules/filemanager/__init__.py
@@ -560,13 +560,20 @@ class FileManagerModule(_ModuleBase):
 
         # 比对文件名并整理字幕
         org_path = Path(fileitem.path)
-        org_dir: Path = org_path.parent
         # 列出所有字幕文件
         storage_oper = self.__get_storage_oper(fileitem.storage)
         if not storage_oper:
             logger.error(f"不支持 {fileitem.storage} 的文件整理")
             return False, f"不支持的文件存储：{fileitem.storage}"
-        file_list: List[FileItem] = storage_oper.list(fileitem)
+
+        # 上级目录
+        org_dir: Path = org_path.parent
+        # 查找上级文件项
+        parent_item: FileItem = storage_oper.get_folder(org_dir)
+        if not parent_item:
+            return False, f"{org_dir} 目录获取失败"
+        # 字幕文件列表
+        file_list: List[FileItem] = storage_oper.list(parent_item)
         if len(file_list) == 0:
             logger.debug(f"{org_dir} 目录下没有找到字幕文件...")
         else:
@@ -650,19 +657,23 @@ class FileManagerModule(_ModuleBase):
         :param transfer_type: 整理方式
         """
         org_path = Path(fileitem.path)
-        dir_name = org_path.parent
-        file_name = org_path.name
         # 列出所有音轨文件
         storage_oper = self.__get_storage_oper(fileitem.storage)
         if not storage_oper:
             logger.error(f"不支持 {fileitem.storage} 的文件整理")
             return False, f"不支持的文件存储：{fileitem.storage}"
-        file_list: List[FileItem] = storage_oper.list(fileitem)
+        # 上级目录
+        org_dir = org_path.parent
+        # 查找上级文件项
+        parent_item: FileItem = storage_oper.get_folder(org_dir)
+        if not parent_item:
+            return False, f"{org_dir} 目录获取失败"
+        file_list: List[FileItem] = storage_oper.list(parent_item)
         # 匹配音轨文件
         pending_file_list: List[FileItem] = [file for file in file_list if Path(file.name).stem == org_path.name
                                              and f".{file.extension.lower()}" in settings.RMT_AUDIOEXT]
         if len(pending_file_list) == 0:
-            return True, f"{dir_name} 目录下没有找到匹配的音轨文件"
+            return True, f"{org_dir} 目录下没有找到匹配的音轨文件"
         logger.debug("音轨文件清单：" + str(pending_file_list))
         for track_file in pending_file_list:
             track_ext = f".{track_file.extension}"
@@ -674,11 +685,11 @@ class FileManagerModule(_ModuleBase):
                                                            target_file=new_track_file,
                                                            transfer_type=transfer_type)
                 if new_item:
-                    logger.info(f"音轨文件 {file_name} 整理完成")
+                    logger.info(f"音轨文件 {org_path.name} 整理完成")
                 else:
-                    logger.error(f"音轨文件 {file_name} 整理失败：{errmsg}")
+                    logger.error(f"音轨文件 {org_path.name} 整理失败：{errmsg}")
             except Exception as error:
-                logger.error(f"音轨文件 {file_name} 整理失败：{str(error)}")
+                logger.error(f"音轨文件 {org_path.name} 整理失败：{str(error)}")
         return True, ""
 
     def __transfer_dir(self, fileitem: FileItem, transfer_type: str,