mirror of
https://github.com/Cp0204/quark-auto-save.git
synced 2026-01-14 00:10:43 +08:00
Merge branch 'Cp0204/quark-auto-save:main'
This commit is contained in:
xiaoQQya
commit
eabb376347
@ -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)
|
||||
|
||||
99
media_servers/plex.py
Normal file
99
media_servers/plex.py
Normal file
@ -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
|
||||
Loading…
Reference in New Issue
Block a user