Merge b18fffc4f6 into 42212dc615

* ✨ 新增 alist-strm-lite 模块，基于 WebDAV 生成 strm

(cherry picked from commit 9c859b212e)

* ♻️ 重构 alist-strm-lite 模块，使用 Alist API

- 由 WebDAV 更改为 Alist API
- 避免页面暴露 Alist 明文密码

(cherry picked from commit 4d6a465057)

* ✨ 优化 alist-strm-lite 模块，从Alist API读取存储信息

* ✨ 优化 strm 链接主机替换参数处理

* 🎨 增加驱动不支持提示

* 🔧 重命名 alist_strm_lite 为 alist_strm_gen

- 更易识别功能

---------

Co-authored-by: xiaoQQya <xiaoQQya@126.com>

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 密钥
+
 ### 特殊场景使用技巧
 
 #### 忽略后缀

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)
 

media_servers/alist_strm_gen.py

@@ -0,0 +1,173 @@
+#!/usr/bin/python3
+# -*- encoding: utf-8 -*-
+"""
+@File    :   alist_strm_gen.py
+@Desc    :   Alist 生成 strm 文件简化版
+@Version :   v1.1
+@Time    :   2024/11/16
+@Author  :   xiaoQQya
+@Contact :   xiaoQQya@126.com
+"""
+import os
+import json
+import requests
+
+
+class Alist_strm_gen:
+
+    video_exts = ["mp4", "mkv", "flv", "mov", "m4v", "avi", "webm", "wmv"]
+    default_config = {
+        "url": "",  # Alist 服务器 URL
+        "token": "",  # Alist 服务器 Token
+        "storage_id": "",  # Alist 服务器夸克存储 ID
+        "strm_save_dir": "/media",  # 生成的 strm 文件保存的路径
+        "strm_replace_host": "",  # strm 文件内链接的主机地址 （可选，缺省时=url）
+    }
+    is_active = False
+    # 缓存参数
+    storage_mount_path = None
+    quark_root_dir = None
+    strm_server = None
+
+    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.storage_id:
+                storage_info = self.get_storage_info(self.storage_id)
+                if storage_info:
+                    if storage_info["driver"] == "Quark":
+                        addition = json.loads(storage_info["addition"])
+                        # 存储挂载路径
+                        self.storage_mount_path = storage_info["mount_path"]
+                        # 夸克根文件夹
+                        self.quark_root_dir = self.get_root_folder_full_path(
+                            addition["cookie"], addition["root_folder_id"]
+                        )
+                        if self.storage_mount_path and self.quark_root_dir:
+                            self.is_active = True
+                    else:
+                        print(
+                            f"Alist-Strm生成: 不支持[{storage_info['driver']}]驱动 ❌"
+                        )
+                    # 替换strm文件内链接的主机地址
+                    self.strm_replace_host = self.strm_replace_host.strip()
+                    if self.strm_replace_host:
+                        if self.strm_replace_host.startswith("http"):
+                            self.strm_server = f"{self.strm_replace_host}/d"
+                        else:
+                            self.strm_server = f"http://{self.strm_replace_host}/d"
+                    else:
+                        self.strm_server = f"{self.url.strip()}/d"
+
+    def run(self, task):
+        if task.get("savepath") and task.get("savepath").startswith(
+            self.quark_root_dir
+        ):
+            alist_path = os.path.normpath(
+                os.path.join(
+                    self.storage_mount_path,
+                    task["savepath"].replace(self.quark_root_dir, "", 1).lstrip("/"),
+                )
+            ).replace("\\", "/")
+            self.refresh(alist_path)
+
+    def get_storage_info(self, storage_id):
+        url = f"{self.url}/api/admin/storage/get"
+        headers = {"Authorization": self.token}
+        querystring = {"id": storage_id}
+        try:
+            response = requests.request("GET", url, headers=headers, params=querystring)
+            response.raise_for_status()
+            data = response.json()
+            if data.get("code") == 200:
+                print(f"Alist-Strm生成: {data['data']['driver']}[{data['data']['mount_path']}]")
+                return data.get("data", [])
+            else:
+                print(f"Alist-Strm生成: 连接失败❌ {response.get('message')}")
+        except requests.exceptions.RequestException as e:
+            print(f"Alist-Strm生成: 获取Alist存储出错 {e}")
+        return False
+
+    def refresh(self, path):
+        try:
+            response = self.get_file_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:
+                    item_path = f"{path}/{item.get('name')}".replace("//", "/")
+                    if item.get("is_dir"):
+                        self.refresh(item_path)
+                    else:
+                        self.generate_strm(item_path)
+        except Exception as e:
+            print(f"📺 获取 Alist 文件列表失败❌ {e}")
+
+    def get_file_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}{os.path.splitext(file_path)[0]}.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:
+                strm_file.write(f"{self.strm_server}{file_path}")
+            print(f"📺 生成STRM文件 {strm_path} 成功✅")
+
+    def get_root_folder_full_path(self, cookie, pdir_fid):
+        if pdir_fid == "0":
+            return "/"
+        url = "https://drive-h.quark.cn/1/clouddrive/file/sort"
+        headers = {
+            "cookie": cookie,
+            "content-type": "application/json",
+        }
+        querystring = {
+            "pr": "ucpro",
+            "fr": "pc",
+            "uc_param_str": "",
+            "pdir_fid": pdir_fid,
+            "_page": 1,
+            "_size": "50",
+            "_fetch_total": "1",
+            "_fetch_sub_dirs": "0",
+            "_sort": "file_type:asc,updated_at:desc",
+            "_fetch_full_path": 1,
+        }
+        try:
+            response = requests.request(
+                "GET", url, headers=headers, params=querystring
+            ).json()
+            if response["code"] == 0:
+                file_names = [
+                    item["file_name"] for item in response["data"]["full_path"]
+                ]
+                return "/".join(file_names)
+        except requests.exceptions.RequestException as e:
+            print(f"Alist-Strm生成: 获取Quark路径出错 {e}")
+        return False

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} 成功✅")

media_servers/plex.py

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

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"):

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": ""