mirror of
https://github.com/Cp0204/quark-auto-save.git
synced 2026-01-14 00:10:43 +08:00
70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
import re
|
||
import requests
|
||
"""
|
||
配合 alist-strm 项目，触发特定配置运行
|
||
https://github.com/tefuirZ/alist-strm
|
||
"""
|
||
class Alist_strm:
|
||
|
||
default_config = {"url": "", "cookie": "", "config_id": ""}
|
||
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.cookie and self.config_id:
|
||
if self.get_info(self.config_id):
|
||
self.is_active = True
|
||
|
||
def run(self, task):
|
||
self.run_selected_configs(self.config_id)
|
||
|
||
def get_info(self, config_id):
|
||
url = f"{self.url}/edit/{config_id}"
|
||
headers = {"Cookie": self.cookie}
|
||
try:
|
||
response = requests.request("GET", url, headers=headers)
|
||
response.raise_for_status()
|
||
html_content = response.text
|
||
# 用正则提取 config_name 的值
|
||
match = re.search(r'name="config_name" value="([^"]+)"', html_content)
|
||
if match:
|
||
config_name = match.group(1)
|
||
print(f"alist-strm配置: {config_name}")
|
||
return True
|
||
else:
|
||
print(f"alist-strm配置: 匹配失败❌")
|
||
except requests.exceptions.RequestException as e:
|
||
print(f"获取alist-strm配置信息出错: {e}")
|
||
return False
|
||
|
||
def run_selected_configs(self, selected_configs_str):
|
||
url = f"{self.url}/run_selected_configs"
|
||
headers = {"Cookie": self.cookie}
|
||
try:
|
||
selected_configs = [int(x.strip()) for x in selected_configs_str.split(",")]
|
||
except ValueError:
|
||
print("Error: 运行alist-strm配置错误，id应以,分割")
|
||
return None
|
||
data = [("selected_configs", config_id) for config_id in selected_configs]
|
||
data.append(("action", "run_selected"))
|
||
try:
|
||
response = requests.post(url, headers=headers, data=data)
|
||
response.raise_for_status()
|
||
html_content = response.text
|
||
# 用正则提取 config_name 的值
|
||
match = re.search(r'role="alert">\s*([^<]+)\s*<button', html_content)
|
||
if match:
|
||
alert = match.group(1).strip()
|
||
print(f"🔗 alist-strm配置运行: {alert}✅")
|
||
return True
|
||
else:
|
||
print(f"🔗 alist-strm配置运行: 失败❌")
|
||
except requests.exceptions.RequestException as e:
|
||
print(f"Error: {e}")
|
||
return False
|