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

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

Browse Source
This commit is contained in:
Cp0204 2024-11-18 16:03:22 +08:00
parent 64b144b613
commit 465eba0709
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
media_servers/alist_strm_lite.py
View File

@ -21,11 +21,13 @@ class Alist_strm_lite:
"token": "", # Alist 服务器 Token "token": "", # Alist 服务器 Token
"storage_id": "", # Alist 服务器夸克存储 ID "storage_id": "", # Alist 服务器夸克存储 ID
"strm_save_dir": "/media", # 生成的 strm 文件保存的路径 "strm_save_dir": "/media", # 生成的 strm 文件保存的路径
"strm_url_host": "", # strm 文件内链接的主机地址 （可选，缺省时=url） "strm_replace_host": "", # strm 文件内链接的主机地址 （可选，缺省时=url）
} }
is_active = False is_active = False
# 缓存参数
storage_mount_path = None storage_mount_path = None
quark_root_dir = None quark_root_dir = None
strm_server = None
def __init__(self, **kwargs): def __init__(self, **kwargs):
if kwargs: if kwargs:
@ -46,6 +48,15 @@ class Alist_strm_lite:
) )
if self.storage_mount_path and self.quark_root_dir: if self.storage_mount_path and self.quark_root_dir:
self.is_active = True self.is_active = True
# 替换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): def run(self, task):
if task.get("savepath") and task.get("savepath").startswith( if task.get("savepath") and task.get("savepath").startswith(
@ -120,10 +131,7 @@ class Alist_strm_lite:
if not os.path.exists(os.path.dirname(strm_path)): if not os.path.exists(os.path.dirname(strm_path)):
os.makedirs(os.path.dirname(strm_path)) os.makedirs(os.path.dirname(strm_path))
with open(strm_path, "w", encoding="utf-8") as strm_file: with open(strm_path, "w", encoding="utf-8") as strm_file:
host = ( strm_file.write(f"{self.strm_server}{file_path}")
self.strm_url_host if self.strm_url_host.strip() else self.url
).rstrip("/")
strm_file.write(f"{host}/d{file_path}")
print(f"📺 生成STRM文件 {strm_path} 成功✅") print(f"📺 生成STRM文件 {strm_path} 成功✅")
def get_root_folder_full_path(self, cookie, pdir_fid): def get_root_folder_full_path(self, cookie, pdir_fid):