liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增 消息定时推送功能

Browse Source
This commit is contained in:
liuwei
2025-06-10 11:37:08 +08:00
parent ac29a826ee
commit 21130c943d
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
plugins/message_push_task/main.py
View File

@@ -5,6 +5,7 @@ from loguru import logger
from base.plugin_common.message_plugin_interface import MessagePluginInterface from base.plugin_common.message_plugin_interface import MessagePluginInterface
from base.plugin_common.plugin_interface import PluginStatus from base.plugin_common.plugin_interface import PluginStatus
from db.connection import DBConnectionManager
from db.task_db import TaskDBOperator from db.task_db import TaskDBOperator
from utils.decorator.async_job import async_job from utils.decorator.async_job import async_job
from utils.decorator.plugin_decorators import plugin_stats_decorator from utils.decorator.plugin_decorators import plugin_stats_decorator
@@ -55,17 +56,17 @@ class MessagePushTask(MessagePluginInterface):
# 保存上下文对象 # 保存上下文对象
self.event_system = context.get("event_system") self.event_system = context.get("event_system")
self.db_manager = context.get("db_manager")
self.db_manager = DBConnectionManager.get_instance()
# 初始化组件 # 初始化组件
self.db = TaskDBOperator(self.db_manager) self.db = TaskDBOperator(self.db_manager)
async_job.every_seconds(5)(self.process_scheduled_tasks)
# 初始化数据库表 # 初始化数据库表
if not self.db.init_tables(): if not self.db.init_tables():
self.LOG.error("初始化数据库表失败") self.LOG.error("初始化数据库表失败")
return False return False
async_job.every_seconds(5)(self.process_scheduled_tasks)
# 加载配置 # 加载配置
self._commands = self._config.get("MessagePush", {}).get("command", ["推送", "消息推送"]) self._commands = self._config.get("MessagePush", {}).get("command", ["推送", "消息推送"])
self.command_format = self._config.get("MessagePush", {}).get("command-format", "推送 消息内容") self.command_format = self._config.get("MessagePush", {}).get("command-format", "推送 消息内容")
@@ -242,7 +243,7 @@ class MessagePushTask(MessagePluginInterface):
self.LOG.error(f"处理定时任务出错: {e}") self.LOG.error(f"处理定时任务出错: {e}")
def _calculate_next_schedule_time(self, current_time: datetime, interval: str, end_time: datetime = None) -> \ def _calculate_next_schedule_time(self, current_time: datetime, interval: str, end_time: datetime = None) -> \
Optional[datetime]: Optional[datetime]:
"""计算下次执行时间 """计算下次执行时间
Args: Args: