101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
#!/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)
|