⚡️ 优化Plex路径处理及库信息缓存

media_servers/plex.py

@@ -3,12 +3,14 @@ import requests
 
 
 class Plex:
+
     default_config = {
         "url": "",  # Plex服务器URL
-        "token": "",  # Plex Token
-        "base_path": ""  # Plex媒体库基础路径
+        "token": "",  # Plex Token，可F12在请求中抓取
+        "quark_root_path": "",  # 夸克根目录在Plex中的路径；假设夸克目录/media/tv在plex中对应的路径为/quark/media/tv，则为/quark
     }
     is_active = False
+    _libraries = None  # 缓存库信息
 
     def __init__(self, **kwargs):
         if kwargs:
@@ -17,33 +19,31 @@ class Plex:
                     setattr(self, key, kwargs[key])
                 else:
                     print(f"{self.__class__.__name__} 模块缺少必要参数: {key}")
-            if self.url and self.token and self.base_path:
+            if self.url and self.token and self.quark_root_path:
                 if self.get_info():
                     self.is_active = True
 
     def run(self, task):
         if task.get("savepath"):
+            # 检查是否已缓存库信息
+            if self._libraries is None:
+                self._libraries = self._get_libraries()
             # 拼接完整路径
-            full_path = os.path.join(self.base_path, task["savepath"].lstrip("/"))
-            full_path = full_path.replace("\\", "/")
+            full_path = os.path.normpath(
+                os.path.join(self.quark_root_path, task["savepath"].lstrip("/"))
+            ).replace("\\", "/")
             self.refresh(full_path)
-        return task
 
     def get_info(self):
         """获取Plex服务器信息"""
-        headers = {
-            'Accept': 'application/json',
-            'X-Plex-Token': self.token
-        }
-
+        headers = {"Accept": "application/json", "X-Plex-Token": self.token}
         try:
-            response = requests.get(
-                f"{self.url}/",
-                headers=headers
-            )
+            response = requests.get(f"{self.url}/", headers=headers)
             if response.status_code == 200:
-                info = response.json()['MediaContainer']
-                print(f"Plex媒体库: {info.get('friendlyName','')} v{info.get('version','')}")
+                info = response.json()["MediaContainer"]
+                print(
+                    f"Plex媒体库: {info.get('friendlyName','')} v{info.get('version','')}"
+                )
                 return True
             else:
                 print(f"Plex媒体库: 连接失败❌ 状态码：{response.status_code}")
@@ -55,45 +55,42 @@ class Plex:
         """刷新指定文件夹"""
         if not folder_path:
             return False
-
-        headers = {
-            'Accept': 'application/json',
-            'X-Plex-Token': self.token
-        }
-
+        headers = {"Accept": "application/json", "X-Plex-Token": self.token}
         try:
-            response = requests.get(
-                f"{self.url}/library/sections",
-                headers=headers
-            )
-
-            if response.status_code != 200:
-                print(f"🎞️ 刷新Plex媒体库：获取库信息失败❌ 状态码：{response.status_code}")
-                return False
-
-            libraries = response.json()['MediaContainer']['Directory']
-
-            for library in libraries:
-                for location in library.get('Location', []):
-                    if folder_path.startswith(location['path']):
-                        library_id = library['key']
-                        refresh_url = (
-                            f"{self.url}/library/sections/{library_id}/refresh"
-                            f"?path={folder_path}"
-                        )
+            for library in self._libraries:
+                for location in library.get("Location", []):
+                    if (
+                        os.path.commonpath([folder_path, location["path"]])
+                        == location["path"]
+                    ):
+                        refresh_url = f"{self.url}/library/sections/{library['key']}/refresh?path={folder_path}"
                         refresh_response = requests.get(refresh_url, headers=headers)
-
                         if refresh_response.status_code == 200:
-                            print(f"🎞️ 刷新Plex媒体库：成功✅")
-                            print(f"📁 扫描路径: {folder_path}")
+                            print(
+                                f"🎞️ 刷新Plex媒体库：{library['title']} [{folder_path}] 成功✅"
+                            )
                             return True
                         else:
-                            print(f"🎞️ 刷新Plex媒体库：刷新请求失败❌ 状态码：{refresh_response.status_code}")
-                            return False
-
-            print(f"🎞️ 刷新Plex媒体库：未找到匹配的媒体库❌ {folder_path}")
-
+                            print(
+                                f"🎞️ 刷新Plex媒体库：刷新请求失败❌ 状态码：{refresh_response.status_code}"
+                            )
+            print(f"🎞️ 刷新Plex媒体库：{folder_path} 未找到匹配的媒体库❌")
         except Exception as e:
             print(f"刷新Plex媒体库出错: {e}")
+        return False
 
-        return False
+    def _get_libraries(self):
+        """获取Plex媒体库信息"""
+        url = f"{self.url}/library/sections"
+        headers = {"Accept": "application/json", "X-Plex-Token": self.token}
+        try:
+            response = requests.get(url, headers=headers)
+            if response.status_code == 200:
+                libraries = response.json()["MediaContainer"].get("Directory", [])
+                return libraries
+            else:
+                print(f"🎞️ 获取Plex媒体库信息失败❌ 状态码：{response.status_code}")
+                return None
+        except Exception as e:
+            print(f"获取Plex媒体库信息出错: {e}")
+            return None