This commit is contained in:
2024
2025-01-25 23:01:49 +08:00
parent 987eeab917
commit 8e5bfb1a6d

View File

@@ -7,7 +7,7 @@ new Env('重复任务优化');
import json import json
import logging import logging
import os import os
import sys import sys,re
import time import time
import traceback import traceback
@@ -157,20 +157,29 @@ def disable_duplicate_tasks(ids: list) -> None:
def get_token() -> str or None: def get_token() -> str or None:
path = '/ql/config/auth.json' # 设置青龙 auth文件地址 if os.path.exists("/ql/data/db/keyv.sqlite"):
global flag1 path="/ql/data/db/keyv.sqlite"
flag1 = True elif os.path.exists("/ql/config/auth.json"):
if not os.path.isfile(path): path="/ql/config/auth.json"
path = '/ql/data/config/auth.json' # 尝试设置青龙 auth 新版文件地址 elif os.path.exists("/ql/data/config/auth.json"):
flag1 = False path="/ql/data/config/auth.json"
try: try:
with open(path, "r", encoding="utf-8") as f: if 'keyv' in path:
data = json.load(f) with open(path, "r", encoding="latin1") as file:
auth = file.read()
matches = re.search(r'token":"([^"]+)"', auth)
token = matches.group(1)
else:
with open(path, "r") as file:
auth = file.read()
auth = json.loads(auth)
token = auth["token"]
except Exception: except Exception:
logger.info(f"❌无法获取 token!!!\n{traceback.format_exc()}") logger.info(f"❌无法获取 token!!!\n{traceback.format_exc()}")
send("禁用重复任务失败", "无法获取 token!!!") send("禁用重复任务失败", "无法获取 token!!!")
exit(1) exit(1)
return data.get("token") return token
if __name__ == "__main__": if __name__ == "__main__":