feat: restore source parity and harden agent runtime

2026-06-22 11:17:26 +08:00
parent e33f08277b
commit 0793eb82d6
596 changed files with 168879 additions and 290 deletions
@@ -0,0 +1,100 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+通知节点处理器
+处理流程中的通知发送
+"""
+import logging
+from typing import Dict, TYPE_CHECKING
+
+from online_dev.workflow.engine.handlers.base_handler import BaseNodeHandler
+
+if TYPE_CHECKING:
+    from online_dev.workflow.engine.base import ExecutionContext
+
+logger = logging.getLogger(__name__)
+
+
+class NotifyHandler(BaseNodeHandler):
+    """
+    通知节点处理器
+    
+    支持多种通知渠道:
+    - site: 站内信
+    - email: 邮件
+    - sms: 短信
+    - wechat: 微信
+    - dingtalk: 钉钉
+    - feishu: 飞书
+    """
+
+    async def execute(self, context: 'ExecutionContext', node: Dict) -> None:
+        """
+        执行通知节点：发送通知
+        """
+        from online_dev.workflow.engine.assignee_resolver import assignee_resolver
+
+        node_id = self.get_node_id(node)
+        node_name = self.get_node_name(node)
+        node_config = self.get_node_config(node)
+
+        # 解析通知对象
+        recipient_ids = await assignee_resolver.resolve(
+            context.db,
+            {
+                'assigneeType': node_config.get('recipientType', 'user'),
+                'assignees': node_config.get('recipients', []),
+                'assigneeLevel': node_config.get('recipientLevel'),
+                'assigneeField': node_config.get('recipientField'),
+                'assigneeFields': node_config.get('recipientFields'),
+                'assigneeFieldMappings': node_config.get('recipientFieldMappings'),
+            },
+            context.instance,
+            context.form_data,
+        )
+
+        channels = node_config.get('channels', ['site'])
+        title = node_config.get('title', '')
+        content = node_config.get('content', '')
+
+        logger.info(f"通知节点 - 发送给 {len(recipient_ids)} 人，渠道: {channels}")
+
+        # 构建模板变量上下文
+        notify_context = {
+            'initiator': '',
+            'title': context.instance.title,
+            'instance_no': context.instance.instance_no,
+            'node_name': node_name,
+            'form': context.form_data,
+        }
+
+        # 调用通知服务发送
+        if recipient_ids:
+            from core.message.service import NotifyService
+            await NotifyService.send(
+                db=context.db,
+                recipient_ids=recipient_ids,
+                title=title,
+                content=content,
+                channels=channels,
+                msg_type="workflow",
+                context=notify_context,
+                sender_id=context.instance.initiator_id,
+            )
+
+        # 记录通知日志
+        await self.create_log(
+            context, node, 'notify',
+            comment=f'发送通知: {title}',
+            extra_data={
+                'recipients': recipient_ids,
+                'channels': channels,
+                'title': title,
+                'content': content,
+            },
+        )
+
+        logger.info(f"通知发送完成")
+
+        # 通知发送后继续推进
+        await self.advance_to_next(context, node)