118 lines
3.3 KiB
Python
118 lines
3.3 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
块处理器基类模块
|
||
提供所有块处理器的基类和注册功能
|
||
"""
|
||
|
||
from abc import ABC, abstractmethod
|
||
from typing import Dict, List, Any, Optional, Type, Callable
|
||
import json
|
||
import logging
|
||
|
||
# 获取日志记录器
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# 块处理器注册表
|
||
_block_handlers = {}
|
||
|
||
class BlockHandler(ABC):
|
||
"""
|
||
块处理器抽象基类
|
||
所有具体的块处理器都应该继承这个类
|
||
"""
|
||
|
||
@abstractmethod
|
||
async def execute(
|
||
self,
|
||
block: Dict[str, Any],
|
||
input_params: Dict[str, Any],
|
||
context: Any # TaskContext类型，避免循环导入
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
执行块
|
||
|
||
Args:
|
||
block: 块定义
|
||
input_params: 解析后的输入参数
|
||
context: 任务上下文
|
||
|
||
Returns:
|
||
Dict[str, Any]: 执行结果，必须包含success字段
|
||
"""
|
||
pass
|
||
|
||
async def _record_task_log(
|
||
self,
|
||
block: Dict[str, Any],
|
||
result: Dict[str, Any],
|
||
context: Any # TaskContext类型，避免循环导入
|
||
) -> None:
|
||
"""
|
||
记录任务日志
|
||
|
||
Args:
|
||
block: 块定义
|
||
result: 执行结果
|
||
context: 任务上下文
|
||
"""
|
||
from sqlalchemy import insert
|
||
from data.models.tasklog import VWEDTaskLog
|
||
from data.session import get_async_session
|
||
import uuid
|
||
from datetime import datetime
|
||
|
||
try:
|
||
# 创建任务日志记录
|
||
task_log_id = str(uuid.uuid4())
|
||
|
||
# 自定义JSON序列化器，处理datetime对象和bytes对象
|
||
def json_serializer(obj):
|
||
if isinstance(obj, datetime):
|
||
return obj.isoformat()
|
||
if isinstance(obj, bytes):
|
||
return obj.decode('utf-8', errors='backslashreplace')
|
||
raise TypeError(f"Object of type {type(obj)} is not JSON serializable")
|
||
|
||
async with get_async_session() as session:
|
||
stmt = insert(VWEDTaskLog).values(
|
||
id=task_log_id,
|
||
level=1 if result.get("success", False) else 3, # 1: 信息, 3: 错误
|
||
message=json.dumps(result, ensure_ascii=False, default=json_serializer),
|
||
task_block_id=block.get("id", "unknown"),
|
||
task_id=context.task_def_id,
|
||
task_record_id=context.task_record_id
|
||
)
|
||
await session.execute(stmt)
|
||
await session.commit()
|
||
except Exception as e:
|
||
logger.error(f"记录任务日志失败: {str(e)}")
|
||
|
||
|
||
# 注册装饰器
|
||
def register_handler(block_type: str):
|
||
"""
|
||
注册块处理器的装饰器
|
||
|
||
Args:
|
||
block_type: 块类型
|
||
"""
|
||
def decorator(cls):
|
||
_block_handlers[block_type] = cls()
|
||
return cls
|
||
return decorator
|
||
|
||
|
||
# 获取块处理器
|
||
def get_block_handler(block_type: str) -> Optional[BlockHandler]:
|
||
"""
|
||
获取块处理器
|
||
|
||
Args:
|
||
block_type: 块类型
|
||
|
||
Returns:
|
||
Optional[BlockHandler]: 对应的块处理器，如果不存在则返回None
|
||
"""
|
||
return _block_handlers.get(block_type) |