代码优化

This commit is contained in:
Natalie Johnson
2024-08-22 19:10:47 +08:00
parent 2fe825f67d
commit 7c163142c4
9 changed files with 61 additions and 27 deletions
+7 -7
View File
@@ -87,15 +87,14 @@ async def _handle_setting(
)
if extra_data.tasks:
for task in extra_data.tasks:
task_list.append(
task_list.append((task.create_status,
TaskInfo(
module=task.module,
name=task.name,
status=task.status,
run_time=task.run_time,
default_status=task.default_status,
)
)
))
@driver.on_startup
@@ -164,15 +163,16 @@ async def _():
}
create_list = []
update_list = []
for task in task_list:
for status, task in task_list:
if task.module not in module_dict:
create_list.append(task)
create_list.append((status, task))
else:
task.id = module_dict[task.module]
update_list.append(task)
if create_list:
await TaskInfo.bulk_create(create_list, 10)
if block := [t.module for t in create_list if not t.default_status]:
_create_list = [t[1] for t in create_list]
await TaskInfo.bulk_create(_create_list, 10)
if block := [t[1].module for t in create_list if not t[0]]:
block_task = ",".join(block) + ","
if group_list := await GroupConsole.all():
for group in group_list: