mirror of
https://github.com/6dylan6/jdpro.git
synced 2025-12-17 15:25:03 +08:00
1
This commit is contained in:
29
jd_taskop.py
29
jd_taskop.py
@@ -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__":
|
||||||
|
|||||||
Reference in New Issue
Block a user