diff --git a/app/core/task_manager.py b/app/core/task_manager.py index 3b6d81a..a2e05e2 100644 --- a/app/core/task_manager.py +++ b/app/core/task_manager.py @@ -180,9 +180,7 @@ class _TaskManager: task["status"] = "运行" await Config.send_json( WebSocketMessage( - id=str(task_id), - type="Update", - data={"task_list": task_list}, + id=str(task_id), type="Update", data={"task_list": task_list} ).model_dump() ) logger.info(f"任务开始: {script_id}") @@ -270,5 +268,19 @@ class _TaskManager: ).model_dump() ) + if mode == "自动代理" and task_id in Config.QueueConfig: + + await Config.send_json( + WebSocketMessage( + id=str(task_id), + type="Signal", + data={ + "power": Config.QueueConfig[task_id].get( + "Info", "AfterAccomplish" + ) + }, + ).model_dump() + ) + TaskManager = _TaskManager()