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

🔨 调整任务调度器代码

Browse Source
This commit is contained in:
Cp0204 2024-05-24 11:42:40 +08:00
parent 07c6c02d5b
commit f22ab1c9db
1 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
app/run.py
View File

@ -44,6 +44,7 @@ app.json.sort_keys = False
app.jinja_env.variable_start_string = "[[" app.jinja_env.variable_start_string = "[["
app.jinja_env.variable_end_string = "]]" app.jinja_env.variable_end_string = "]]"
scheduler = BackgroundScheduler()
def gen_md5(string): def gen_md5(string):
md5 = hashlib.md5() md5 = hashlib.md5()
@ -173,7 +174,7 @@ def run_script_now():
# 定时任务执行的函数 # 定时任务执行的函数
def run_script(args): def run_python(args):
os.system(f"{python_path} {args}") os.system(f"{python_path} {args}")
@ -184,18 +185,25 @@ def reload_tasks():
# 添加新任务 # 添加新任务
crontab = data.get("crontab") crontab = data.get("crontab")
if crontab: if crontab:
print(">>> 重载调度器")
print(f"调度状态: {scheduler.state}")
print(f"定时规则: {crontab}")
if scheduler.state == 1: if scheduler.state == 1:
scheduler.pause() # 暂停调度器 scheduler.pause() # 暂停调度器
trigger = CronTrigger.from_crontab(crontab) trigger = CronTrigger.from_crontab(crontab)
scheduler.remove_all_jobs() scheduler.remove_all_jobs()
scheduler.add_job( scheduler.add_job(
run_script, trigger=trigger, args=[f"{script_path} {config_path}"] run_python,
trigger=trigger,
args=[f"{script_path} {config_path}"],
id=script_path,
) )
if scheduler.state == 2: if scheduler.state == 0:
scheduler.resume() # 恢复调度器 scheduler.start()
elif scheduler.state == 2:
scheduler.resume()
scheduler_state_map = {0: "停止", 1: "运行", 2: "暂停"}
print(">>> 重载调度器")
print(f"调度状态: {scheduler_state_map[scheduler.state]}")
print(f"定时规则: {crontab}")
print(f"现有任务: {scheduler.get_jobs()}")
else: else:
print(">>> no crontab") print(">>> no crontab")
@ -227,8 +235,5 @@ def init():
if __name__ == "__main__": if __name__ == "__main__":
init() init()
# 创建后台调度器
scheduler = BackgroundScheduler()
reload_tasks() reload_tasks()
scheduler.start() app.run(debug=os.environ.get("DEBUG", False), host="0.0.0.0", port=5005)
app.run(debug=os.environ.get("DEBUG", True), host="0.0.0.0", port=5005)