|
| 1 | +# coding=utf-8 |
| 2 | +""" |
| 3 | + @project: MaxKB |
| 4 | + @Author:虎虎 |
| 5 | + @file: simple_task.py |
| 6 | + @date:2026/1/14 19:18 |
| 7 | + @desc: |
| 8 | +""" |
| 9 | +from trigger.handler.impl.task.application_task import ApplicationTask |
| 10 | +from trigger.handler.impl.trigger.scheduled_trigger import ScheduledTrigger |
| 11 | + |
| 12 | +simple_task_handlers = [ApplicationTask] |
| 13 | + |
| 14 | +simple_trigger_handlers = [ScheduledTrigger] |
| 15 | + |
| 16 | + |
| 17 | +def execute(trigger_task, **kwargs): |
| 18 | + """ |
| 19 | + 执行触发器任务 |
| 20 | + @param trigger_task: 触发器任务数据 |
| 21 | + @param kwargs: 额外数据 |
| 22 | + @return: |
| 23 | + """ |
| 24 | + for simple_task_handler in simple_task_handlers: |
| 25 | + if simple_task_handler.support(trigger_task, **kwargs): |
| 26 | + return simple_task_handler.execute(trigger_task, **kwargs) |
| 27 | + raise Exception("不支持的处理器类型") |
| 28 | + |
| 29 | + |
| 30 | +def deploy(trigger, **kwargs): |
| 31 | + """ |
| 32 | + 部署触发器 |
| 33 | + @param trigger: 触发器字典数据 |
| 34 | + @param kwargs: 额外数据 |
| 35 | + @return: |
| 36 | + """ |
| 37 | + for simple_trigger_handler in simple_trigger_handlers: |
| 38 | + if simple_trigger_handler.support(trigger, **kwargs): |
| 39 | + return simple_trigger_handler.deploy(trigger, **kwargs) |
| 40 | + raise Exception("不支持的触发器类型") |
| 41 | + |
| 42 | + |
| 43 | +def undeploy(trigger, **kwargs): |
| 44 | + """ |
| 45 | + 取消部署触发器 |
| 46 | + @param trigger: 触发器字典数据 |
| 47 | + @param kwargs: 额外数据 |
| 48 | + @return: |
| 49 | + """ |
| 50 | + for simple_trigger_handler in simple_trigger_handlers: |
| 51 | + if simple_trigger_handler.support(trigger, **kwargs): |
| 52 | + return simple_trigger_handler.undeploy(trigger, **kwargs) |
| 53 | + raise Exception("不支持的触发器类型") |
0 commit comments