chickliu/quark-auto-save
1
0
Fork 0
mirror of https://github.com/Cp0204/quark-auto-save.git synced 2026-01-12 23:30:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge b18fffc4f6 into c5662644da

Browse Source
This commit is contained in:
xiaoQQya 2024-11-16 15:07:07 +00:00 committed by GitHub
commit f7c0d766c3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 191 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
README.md
View File

@ -77,6 +77,7 @@ docker run -d \
-e WEBUI_USERNAME=admin \
-e WEBUI_PASSWORD=admin123 \
-v ./quark-auto-save/config:/app/config \
-v ./media:/media \
-v /etc/localtime:/etc/localtime \
--network bridge \
--restart unless-stopped \
@ -101,6 +102,7 @@ services:
WEBUI_PASSWORD: "admin123"
volumes:
- ./quark-auto-save/config:/app/config
- ./media:/media
- /etc/localtime:/etc/localtime
```
@ -145,6 +147,43 @@ docker run --rm -v /var/run/docker.sock:/var/run/docker.sock containrrr/watchtow
更多正则使用说明已转移到 Wiki ：[正则处理教程](https://github.com/Cp0204/quark-auto-save/wiki/%E6%AD%A3%E5%88%99%E5%A4%84%E7%90%86%E6%95%99%E7%A8%8B)
### 媒体库配置
#### alist
用于自动刷新 Alist 目录，各配置含义如下：
* url: Alist 访问地址，例如：http://127.0.0.1:5244
* token：Alist 访问令牌，Alist 管理后台-设置-其他-令牌-复制令牌
* quark_root_path: Alist 夸克网盘的挂载路径，Alist 管理后台-存储-夸克驱动-挂载路径，默认 `/quark`
* quark_root_dir: Alist 挂载的夸克网盘根目录，Alist 管理后台-存储-夸克驱动-根文件夹ID（Alist 填写的为文件夹 ID，此处需要填写文件夹路径），默认 `/`
#### alist_strm_lite
用于从 Alist 生成 strm 文件，基于 WebDAV 实现的轻量版本，alist_strm_lite 与 alist_strm 二选一即可，各配置含义如下：
* url: Alist 访问地址，例如：http://127.0.0.1:5244
* token: Alist 访问令牌，Alist 管理后台-设置-其他-令牌-复制令牌
* quark_root_path: Alist 夸克网盘的挂载路径，Alist 管理后台-存储-夸克驱动-挂载路径，默认 `/quark`
* quark_root_dir: Alist 挂载的夸克网盘根目录，Alist 管理后台-存储-夸克驱动-根文件夹ID（Alist 填写的为文件夹 ID，此处需要填写文件夹路径），默认 `/`
* strm_save_dir: strm 文件保存路径，如使用 docker 对应 docker 内部路径，默认 `/media`
* strm_url_host: strm 文件内链接使用的主机地址，例如：http://example.host ，配合自定义 host 解析在 strm 文件迁移机器时无需重新生成 strm 文件，修改自定义 host 解析地址即可，默认为空时使用 url 配置
#### alist_strm
用于从 Alist 生成 strm 文件，需配合 [alist-strm](https://github.com/tefuirZ/alist-strm) 项目使用，alist_strm_lite 与 alist_strm 二选一即可，各配置含义如下：
* url: alist-strm 访问地址，例如：http://127.0.0.1:5000
* cookie: alist-strm 的访问 cookie
* config_id: alist-strm 的配置 ID
#### emby
用于自动扫描媒体库文件，各配置含义如下：
* url: Emby 访问地址，例如：http://127.0.0.1:8096
* token: Emby API 密钥，管理 Emby Server-高级-API 密钥-新 API 密钥
### 特殊场景使用技巧
#### 忽略后缀

8
media_servers/alist.py
View File

@ -8,6 +8,7 @@ class Alist:
"url": "", # Alist服务器URL
"token": "", # Alist服务器Token
"quark_root_path": "/quark", # 夸克根目录在Alist中的挂载路径
"quark_root_dir": "/" # 夸克在Alist中挂载的根目录
}
is_active = False
@ -18,14 +19,13 @@ 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"):
if task.get("savepath") and task.get("savepath").startswith(self.quark_root_dir):
full_path = os.path.normpath(
os.path.join(self.quark_root_path, task["savepath"].lstrip("/"))
os.path.join(self.quark_root_path, task["savepath"].lstrip("/").lstrip(self.quark_root_dir))
).replace("\\", "/")
self.refresh(full_path)

104
media_servers/alist_strm_lite.py Normal file
View File

@ -0,0 +1,104 @@
#!/usr/bin/python3
# -*- encoding: utf-8 -*-
"""
@File : alist_strm_lite.py
@Desc : Alist 生成 strm 文件简化版
@Version : v1.0
@Time : 2024/11/16
@Author : xiaoQQya
@Contact : xiaoQQya@126.com
"""
import os
import requests
class Alist_strm_lite:
video_exts = ["mp4", "mkv", "flv", "mov", "m4v", "avi", "webm", "wmv"]
default_config = {
"url": "", # Alist 服务器 URL
"token": "", # Alist 服务器 Token
"quark_root_path": "/quark", # 夸克根目录在 Alist 中的挂载路径
"quark_root_dir": "/", # 夸克在 Alist 中挂载的根目录
"strm_save_dir": "/media", # 生成的 strm 文件保存的路径
"strm_url_host": "" # strm 文件内链接的主机地址
}
is_active = False
def __init__(self, **kwargs):
if kwargs:
for key, _ 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.get_info():
self.is_active = True
def run(self, task):
if task.get("savepath") and task.get("savepath").startswith(self.quark_root_dir):
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"
headers = {"Authorization": self.token}
querystring = {"group": "1"}
try:
response = requests.request("GET", url, headers=headers, params=querystring)
response.raise_for_status()
response = response.json()
if response.get("code") == 200:
print(f"Alist-strm Lite: {response.get('data',[])[1].get('value','')} {response.get('data',[])[0].get('value','')}")
return True
else:
print(f"Alist-strm Lite: 连接失败❌ {response.get('message')}")
except requests.exceptions.RequestException as e:
print(f"Alist-strm Lite: 获取Alist信息出错 {e}")
return False
def refresh(self, path):
try:
response = self.list(path)
if response.get("code") != 200:
print(f"📺 生成 STRM 文件失败❌ {response.get('message')}")
return
else:
files = response.get("data").get("content")
for item in files:
full_path = f"{path}/{item.get('name')}".replace("//", "/")
if item.get("is_dir"):
self.refresh(full_path)
else:
self.generate_strm(full_path)
except Exception as e:
print(f"📺 获取 Alist 文件列表失败❌ {e}")
def list(self, path):
url = f"{self.url}/api/fs/list"
headers = {"Authorization": self.token}
payload = {
"path": path,
"refresh": False,
"password": "",
"page": 1,
"per_page": 0,
}
response = requests.request("POST", url, headers=headers, json=payload)
response.raise_for_status()
return response.json()
def generate_strm(self, file_path):
ext = file_path.split(".")[-1]
if ext.lower() in self.video_exts:
strm_path = f"{self.strm_save_dir}{file_path.rstrip(ext)}strm".replace("//", "/")
if os.path.exists(strm_path):
return
if not os.path.exists(os.path.dirname(strm_path)):
os.makedirs(os.path.dirname(strm_path))
with open(strm_path, "w", encoding="utf-8") as strm_file:
host = self.strm_url_host.rstrip("/") if self.strm_url_host.strip() else self.url.rstrip("/")
strm_file.write(f"{host}/d{file_path}")
print(f"📺 生成STRM文件 {strm_path} 成功✅")

2
media_servers/plex.py
View File

@ -93,4 +93,4 @@ class Plex:
return None
except Exception as e:
print(f"获取Plex媒体库信息出错: {e}")
return None
return None

40
notify.py
View File

@ -101,9 +101,11 @@ push_config = {
'SMTP_SERVER': '', # SMTP 发送邮件服务器，形如 smtp.exmail.qq.com:465
'SMTP_SSL': 'false', # SMTP 发送邮件服务器是否使用 SSL，填写 true 或 false
'SMTP_EMAIL': '', # SMTP 收发件邮箱，通知将会由自己发给自己
'SMTP_EMAIL_FROM': '', # SMTP 发件邮箱
'SMTP_NAME_FROM': '', # SMTP 发件人姓名，可随意填写
'SMTP_PASSWORD': '', # SMTP 登录密码，也可能为特殊口令，视具体邮件服务商说明而定
'SMTP_NAME': '', # SMTP 收发件人姓名，可随意填写
'SMTP_EMAIL_TO': '', # SMTP 收件邮箱，多个收件邮箱逗号分开
'SMTP_NAME_TO': '', # SMTP 收件人姓名，多个收件人逗号分开，顺序与 SMTP_EMAIL_TO 保持一致
'PUSHME_KEY': '', # PushMe 的 PUSHME_KEY
'PUSHME_URL': '', # PushMe 的 PUSHME_URL
@ -662,12 +664,14 @@ def smtp(title: str, content: str) -> None:
if (
not push_config.get("SMTP_SERVER")
or not push_config.get("SMTP_SSL")
or not push_config.get("SMTP_EMAIL")
or not push_config.get("SMTP_EMAIL_FROM")
or not push_config.get("SMTP_EMAIL_TO")
or not push_config.get("SMTP_PASSWORD")
or not push_config.get("SMTP_NAME")
or not push_config.get("SMTP_NAME_FROM")
or not push_config.get("SMTP_NAME_TO")
):
print(
"SMTP 邮件 的 SMTP_SERVER 或者 SMTP_SSL 或者 SMTP_EMAIL 或者 SMTP_PASSWORD 或者 SMTP_NAME 未设置!!\n取消推送"
"SMTP 邮件 的 SMTP_SERVER 或者 SMTP_SSL 或者 SMTP_EMAIL_FROM 或者 SMTP_EMAIL_TO 或者 SMTP_PASSWORD 或者 SMTP_NAME_FROM 或者 SMTP_NAME_TO 未设置!!\n取消推送"
)
return
print("SMTP 邮件 服务启动")
@ -675,16 +679,18 @@ def smtp(title: str, content: str) -> None:
message = MIMEText(content, "plain", "utf-8")
message["From"] = formataddr(
(
Header(push_config.get("SMTP_NAME"), "utf-8").encode(),
push_config.get("SMTP_EMAIL"),
Header(push_config.get("SMTP_NAME_FROM"), "utf-8").encode(),
push_config.get("SMTP_EMAIL_FROM"),
)
)
message["To"] = formataddr(
to_emails = push_config.get("SMTP_EMAIL_TO").split(",")
to_names = push_config.get("SMTP_NAME_TO").split(",")
message["To"] = ",".join([formataddr(
(
Header(push_config.get("SMTP_NAME"), "utf-8").encode(),
push_config.get("SMTP_EMAIL"),
Header(to_names[index] if len(to_names) > index else "", "utf-8").encode(),
to_email,
)
)
) for index, to_email in enumerate(to_emails)])
message["Subject"] = Header(title, "utf-8")
try:
@ -694,11 +700,11 @@ def smtp(title: str, content: str) -> None:
else smtplib.SMTP(push_config.get("SMTP_SERVER"))
)
smtp_server.login(
push_config.get("SMTP_EMAIL"), push_config.get("SMTP_PASSWORD")
push_config.get("SMTP_EMAIL_FROM"), push_config.get("SMTP_PASSWORD")
)
smtp_server.sendmail(
push_config.get("SMTP_EMAIL"),
push_config.get("SMTP_EMAIL"),
push_config.get("SMTP_EMAIL_FROM"),
to_emails,
message.as_bytes(),
)
smtp_server.close()
@ -966,9 +972,11 @@ def add_notify_function():
if (
push_config.get("SMTP_SERVER")
and push_config.get("SMTP_SSL")
and push_config.get("SMTP_EMAIL")
and push_config.get("SMTP_EMAIL_FROM")
and push_config.get("SMTP_PASSWORD")
and push_config.get("SMTP_NAME")
and push_config.get("SMTP_EMAIL_TO")
and push_config.get("SMTP_NAME_FROM")
and push_config.get("SMTP_NAME_TO")
):
notify_function.append(smtp)
if push_config.get("PUSHME_KEY"):

19
quark_config.json
View File

@ -7,6 +7,25 @@
"其他推送渠道//此项可删": "配置方法同青龙"
},
"media_servers": {
"alist": {
"url": "",
"token": "",
"quark_root_path": "/quark",
"quark_root_dir": "/"
},
"alist_strm_lite": {
"url": "",
"token": "",
"quark_root_path": "/quark",
"quark_root_dir": "/",
"strm_save_dir": "/media",
"strm_url_host": ""
},
"alist_strm": {
"url": "",
"cookie": "",
"config_id": ""
},
"emby": {
"url": "",
"token": ""