diff --git a/README.md b/README.md index c65cf14..5a6927c 100644 --- a/README.md +++ b/README.md @@ -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 密钥 + ### 特殊场景使用技巧 #### 忽略后缀 diff --git a/media_servers/alist.py b/media_servers/alist.py index 2a33c6f..741f9bd 100644 --- a/media_servers/alist.py +++ b/media_servers/alist.py @@ -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) diff --git a/media_servers/alist_strm_lite.py b/media_servers/alist_strm_lite.py new file mode 100644 index 0000000..13e61bb --- /dev/null +++ b/media_servers/alist_strm_lite.py @@ -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} 成功✅") diff --git a/media_servers/plex.py b/media_servers/plex.py index a0da69f..16e9fdc 100644 --- a/media_servers/plex.py +++ b/media_servers/plex.py @@ -93,4 +93,4 @@ class Plex: return None except Exception as e: print(f"获取Plex媒体库信息出错: {e}") - return None + return None \ No newline at end of file diff --git a/notify.py b/notify.py index aa37e86..fd92abf 100644 --- a/notify.py +++ b/notify.py @@ -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"): diff --git a/quark_config.json b/quark_config.json index 85222d5..11194a8 100644 --- a/quark_config.json +++ b/quark_config.json @@ -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": ""