diff --git a/media_servers/alist.py b/media_servers/alist.py index 0f2d5b0..741f9bd 100644 --- a/media_servers/alist.py +++ b/media_servers/alist.py @@ -1,11 +1,15 @@ import os -import re import requests class Alist: - default_config = {"url": "", "token": "", "path_prefix": "/quark", "quark_root_dir": "/"} + default_config = { + "url": "", # Alist服务器URL + "token": "", # Alist服务器Token + "quark_root_path": "/quark", # 夸克根目录在Alist中的挂载路径 + "quark_root_dir": "/" # 夸克在Alist中挂载的根目录 + } is_active = False def __init__(self, **kwargs): @@ -15,14 +19,15 @@ class Alist: setattr(self, key, kwargs[key]) else: print(f"{self.__class__.__name__} 模块缺少必要参数: {key}") - if self.url and self.token: - if self.get_info(): + if self.url and self.token and self.get_info(): self.is_active = True def run(self, task): if task.get("savepath") and task.get("savepath").startswith(self.quark_root_dir): - path = self._normalize_path(task["savepath"]) - self.refresh(path) + full_path = os.path.normpath( + os.path.join(self.quark_root_path, task["savepath"].lstrip("/").lstrip(self.quark_root_dir)) + ).replace("\\", "/") + self.refresh(full_path) def get_info(self): url = f"{self.url}/api/admin/setting/list" @@ -54,22 +59,20 @@ class Alist: "per_page": 0, } try: - response = requests.request( - "POST", url, headers=headers, json=payload - ) + response = requests.request("POST", url, headers=headers, json=payload) response.raise_for_status() response = response.json() if response.get("code") == 200: - print(f"📁 刷新Alist目录:{path} 成功✅") + print(f"📁 刷新Alist目录:[{path}] 成功✅") return response.get("data") elif "object not found" in response.get("message", ""): # 如果是根目录就不再往上查找 - if path == "/" or path == self.path_prefix: + if path == "/" or path == self.quark_root_path: print(f"📁 刷新Alist目录:根目录不存在,请检查 Alist 配置") return False # 获取父目录 parent_path = os.path.dirname(path) - print(f"📁 刷新Alist目录:{path} 不存在,转父目录 {parent_path}") + print(f"📁 刷新Alist目录:[{path}] 不存在,转父目录 [{parent_path}]") # 递归刷新父目录 return self.refresh(parent_path) else: @@ -77,9 +80,3 @@ class Alist: except requests.exceptions.RequestException as e: print(f"刷新Alist目录出错: {e}") return False - - def _normalize_path(self, path): - """标准化路径格式""" - if not path.startswith(self.path_prefix): - path = f"/{self.path_prefix}/{path.lstrip(self.quark_root_dir)}" - return re.sub(r"/{2,}", "/", path) diff --git a/media_servers/plex.py b/media_servers/plex.py new file mode 100644 index 0000000..158c14a --- /dev/null +++ b/media_servers/plex.py @@ -0,0 +1,99 @@ +import os +import requests + + +class Plex: + default_config = { + "url": "", # Plex服务器URL + "token": "", # Plex Token + "base_path": "" # Plex媒体库基础路径 + } + is_active = False + + def __init__(self, **kwargs): + if kwargs: + for key, value in self.default_config.items(): + if key in kwargs: + setattr(self, key, kwargs[key]) + else: + print(f"{self.__class__.__name__} 模块缺少必要参数: {key}") + if self.url and self.token and self.base_path: + if self.get_info(): + self.is_active = True + + def run(self, task): + if task.get("savepath"): + # 拼接完整路径 + full_path = os.path.join(self.base_path, task["savepath"].lstrip("/")) + full_path = full_path.replace("\\", "/") + self.refresh(full_path) + return task + + def get_info(self): + """获取Plex服务器信息""" + headers = { + 'Accept': 'application/json', + 'X-Plex-Token': self.token + } + + try: + 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','')}") + return True + else: + print(f"Plex媒体库: 连接失败❌ 状态码:{response.status_code}") + except Exception as e: + print(f"获取Plex媒体库信息出错: {e}") + return False + + def refresh(self, folder_path): + """刷新指定文件夹""" + if not folder_path: + return False + + 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}" + ) + refresh_response = requests.get(refresh_url, headers=headers) + + if refresh_response.status_code == 200: + print(f"🎞️ 刷新Plex媒体库:成功✅") + print(f"📁 扫描路径: {folder_path}") + return True + else: + print(f"🎞️ 刷新Plex媒体库:刷新请求失败❌ 状态码:{refresh_response.status_code}") + return False + + print(f"🎞️ 刷新Plex媒体库:未找到匹配的媒体库❌ {folder_path}") + + except Exception as e: + print(f"刷新Plex媒体库出错: {e}") + + return False \ No newline at end of file