mirror of
https://github.com/Cp0204/quark-auto-save.git
synced 2026-01-18 19:00:44 +08:00
Compare commits
4 Commits
179856adca
...
c5662644da
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c5662644da | ||
|
|
6fba6eefa9 | ||
|
|
22e764e234 | ||
|
|
d42e2f56d6 |
@ -6,7 +6,7 @@
|
|||||||
|
|
||||||
* 模块位于 `media_servers` 目录下.
|
* 模块位于 `media_servers` 目录下.
|
||||||
* 每个模块是一个 `.py` 文件 (例如 `emby.py`, `plex.py`),文件名小写。
|
* 每个模块是一个 `.py` 文件 (例如 `emby.py`, `plex.py`),文件名小写。
|
||||||
* 每个模块文件包含一个与文件名对应的大驼峰命名法类(例如 `emby.py` 中的 `Emby` 类)。
|
* 每个模块文件包含一个与文件名对应的首字母大写命名类(例如 `emby.py` 中的 `Emby` 类)。
|
||||||
|
|
||||||
## 模块要求
|
## 模块要求
|
||||||
|
|
||||||
@ -27,7 +27,7 @@
|
|||||||
|
|
||||||
* **`run(self, task)`**:整个模块入口函数,处理模块逻辑。
|
* **`run(self, task)`**:整个模块入口函数,处理模块逻辑。
|
||||||
* `task` 是一个字典,包含任务信息。如果需要修改任务参数,返回修改后的 `task` 字典;
|
* `task` 是一个字典,包含任务信息。如果需要修改任务参数,返回修改后的 `task` 字典;
|
||||||
* 无修改则不返回或返回 `False`。
|
* 无修改则不返回或返回 `None`。
|
||||||
|
|
||||||
## 模块示例
|
## 模块示例
|
||||||
|
|
||||||
@ -84,4 +84,10 @@ docker run -d \
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
当模块代码正确赋值 `default_config` 时,首次运行会自动补充缺失的键。
|
当模块代码正确赋值 `default_config` 时,首次运行会自动补充缺失的键。
|
||||||
|
|
||||||
|
## 🤝 贡献者
|
||||||
|
|
||||||
|
| 模块 | 说明 | 贡献者 |
|
||||||
|
| ------- | -------------------- | --------------------------------------- |
|
||||||
|
| plex.py | 自动刷新 Plex 媒体库 | [zhazhayu](https://github.com/zhazhayu) |
|
||||||
@ -1,11 +1,14 @@
|
|||||||
import os
|
import os
|
||||||
import re
|
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
|
|
||||||
class Alist:
|
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
|
is_active = False
|
||||||
|
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
@ -21,8 +24,10 @@ class Alist:
|
|||||||
|
|
||||||
def run(self, task):
|
def run(self, task):
|
||||||
if task.get("savepath"):
|
if task.get("savepath"):
|
||||||
path = self._normalize_path(task["savepath"])
|
full_path = os.path.normpath(
|
||||||
self.refresh(path)
|
os.path.join(self.quark_root_path, task["savepath"].lstrip("/"))
|
||||||
|
).replace("\\", "/")
|
||||||
|
self.refresh(full_path)
|
||||||
|
|
||||||
def get_info(self):
|
def get_info(self):
|
||||||
url = f"{self.url}/api/admin/setting/list"
|
url = f"{self.url}/api/admin/setting/list"
|
||||||
@ -54,22 +59,20 @@ class Alist:
|
|||||||
"per_page": 0,
|
"per_page": 0,
|
||||||
}
|
}
|
||||||
try:
|
try:
|
||||||
response = requests.request(
|
response = requests.request("POST", url, headers=headers, json=payload)
|
||||||
"POST", url, headers=headers, json=payload
|
|
||||||
)
|
|
||||||
response.raise_for_status()
|
response.raise_for_status()
|
||||||
response = response.json()
|
response = response.json()
|
||||||
if response.get("code") == 200:
|
if response.get("code") == 200:
|
||||||
print(f"📁 刷新Alist目录:{path} 成功✅")
|
print(f"📁 刷新Alist目录:[{path}] 成功✅")
|
||||||
return response.get("data")
|
return response.get("data")
|
||||||
elif "object not found" in response.get("message", ""):
|
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 配置")
|
print(f"📁 刷新Alist目录:根目录不存在,请检查 Alist 配置")
|
||||||
return False
|
return False
|
||||||
# 获取父目录
|
# 获取父目录
|
||||||
parent_path = os.path.dirname(path)
|
parent_path = os.path.dirname(path)
|
||||||
print(f"📁 刷新Alist目录:{path} 不存在,转父目录 {parent_path}")
|
print(f"📁 刷新Alist目录:[{path}] 不存在,转父目录 [{parent_path}]")
|
||||||
# 递归刷新父目录
|
# 递归刷新父目录
|
||||||
return self.refresh(parent_path)
|
return self.refresh(parent_path)
|
||||||
else:
|
else:
|
||||||
@ -77,9 +80,3 @@ class Alist:
|
|||||||
except requests.exceptions.RequestException as e:
|
except requests.exceptions.RequestException as e:
|
||||||
print(f"刷新Alist目录出错: {e}")
|
print(f"刷新Alist目录出错: {e}")
|
||||||
return False
|
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)
|
|
||||||
|
|||||||
96
media_servers/plex.py
Normal file
96
media_servers/plex.py
Normal file
@ -0,0 +1,96 @@
|
|||||||
|
import os
|
||||||
|
import requests
|
||||||
|
|
||||||
|
|
||||||
|
class Plex:
|
||||||
|
|
||||||
|
default_config = {
|
||||||
|
"url": "", # Plex服务器URL
|
||||||
|
"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:
|
||||||
|
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.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.normpath(
|
||||||
|
os.path.join(self.quark_root_path, task["savepath"].lstrip("/"))
|
||||||
|
).replace("\\", "/")
|
||||||
|
self.refresh(full_path)
|
||||||
|
|
||||||
|
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:
|
||||||
|
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媒体库:{library['title']} [{folder_path}] 成功✅"
|
||||||
|
)
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
print(
|
||||||
|
f"🎞️ 刷新Plex媒体库:刷新请求失败❌ 状态码:{refresh_response.status_code}"
|
||||||
|
)
|
||||||
|
print(f"🎞️ 刷新Plex媒体库:{folder_path} 未找到匹配的媒体库❌")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"刷新Plex媒体库出错: {e}")
|
||||||
|
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
|
||||||
Loading…
Reference in New Issue
Block a user