From d42e2f56d664a3476bf8c5fc908caab588884305 Mon Sep 17 00:00:00 2001 From: Cp0204 Date: Sat, 16 Nov 2024 01:27:49 +0800 Subject: [PATCH 1/2] =?UTF-8?q?=E2=9A=A1=EF=B8=8F=20=E4=BC=98=E5=8C=96Alis?= =?UTF-8?q?t=E9=85=8D=E7=BD=AE=E5=8F=8A=E8=B7=AF=E5=BE=84=E5=A4=84?= =?UTF-8?q?=E7=90=86=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- media_servers/alist.py | 29 +++++++++++++---------------- 1 file changed, 13 insertions(+), 16 deletions(-) diff --git a/media_servers/alist.py b/media_servers/alist.py index d34c2ab..2a33c6f 100644 --- a/media_servers/alist.py +++ b/media_servers/alist.py @@ -1,11 +1,14 @@ import os -import re import requests class Alist: - default_config = {"url": "", "token": "", "path_prefix": "/quark"} + default_config = { + "url": "", # Alist服务器URL + "token": "", # Alist服务器Token + "quark_root_path": "/quark", # 夸克根目录在Alist中的挂载路径 + } is_active = False def __init__(self, **kwargs): @@ -21,8 +24,10 @@ class Alist: def run(self, task): if task.get("savepath"): - path = self._normalize_path(task["savepath"]) - self.refresh(path) + full_path = os.path.normpath( + os.path.join(self.quark_root_path, task["savepath"].lstrip("/")) + ).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}" - return re.sub(r"/{2,}", "/", path) From 25b4802ab8929186ce20b1f641f371baea574bb9 Mon Sep 17 00:00:00 2001 From: zhazhayu Date: Sat, 16 Nov 2024 14:22:30 +0800 Subject: [PATCH 2/2] =?UTF-8?q?=20=E6=B7=BB=E5=8A=A0Plex=E5=AA=92=E4=BD=93?= =?UTF-8?q?=E5=BA=93=E6=94=AF=E6=8C=81=E6=A8=A1=E5=9D=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- media_servers/plex.py | 99 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 99 insertions(+) create mode 100644 media_servers/plex.py 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