326 lines
13 KiB
Python
326 lines
13 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
VWED.device 模块 - 设备处理器注册并运行
|
||
"""
|
||
|
||
import asyncio
|
||
from typing import Dict, Any, List, Union, Callable
|
||
from ..device_handler_service import get_device_service, DeviceType, DeviceBrand, get_protocol_registry
|
||
from ..script_registry_service import get_global_registry
|
||
from utils.logger import get_logger
|
||
|
||
|
||
class VWEDDeviceModule:
|
||
"""VWED.device 模块 - 设备处理器注册并运行"""
|
||
|
||
def __init__(self, script_id: str):
|
||
self.script_id = script_id
|
||
self.device_service = get_device_service()
|
||
self.registry = get_global_registry()
|
||
self.logger = get_logger(f"script.{script_id}.device")
|
||
|
||
def register_and_run(self, device_ids: List[str], device_type: Union[str, DeviceType] = DeviceType.VEHICLE,
|
||
brand_name: str = "huarui", command_type: str = None,
|
||
handler: Callable = None, description: str = "",
|
||
device_brand: str = None, protocol_key: str = None,
|
||
auto_encode: bool = True, **kwargs):
|
||
"""批量注册并运行设备处理器（新接口）
|
||
|
||
系统会自动：
|
||
1. 根据设备品牌和类型生成MQTT topics
|
||
2. 订阅相应的topics
|
||
3. 开始监听消息
|
||
4. 消息到达时自动调用处理函数
|
||
5. 自动转发处理结果
|
||
|
||
使用方式：
|
||
# 批量注册华瑞小车，处理order指令
|
||
VWED.device.register_and_run(
|
||
device_ids=["AGV001", "AGV002", "AGV003"],
|
||
device_type="vehicle",
|
||
brand_name="huarui",
|
||
command_type="order",
|
||
handler=agv_order_processor,
|
||
description="华瑞小车order指令处理器"
|
||
)
|
||
|
||
# 批量注册仙工小车，处理state指令
|
||
VWED.device.register_and_run(
|
||
device_ids=["SEER001", "SEER002"],
|
||
device_type="vehicle",
|
||
brand_name="seer",
|
||
command_type="state",
|
||
handler=seer_state_processor,
|
||
description="仙工小车state指令处理器"
|
||
)
|
||
|
||
# 批量注册门设备，处理instantActions指令
|
||
VWED.device.register_and_run(
|
||
device_ids=["DOOR001", "DOOR002"],
|
||
device_type="door",
|
||
brand_name="huarui",
|
||
command_type="instantActions",
|
||
handler=door_processor,
|
||
description="门设备instantActions指令处理器"
|
||
)
|
||
|
||
Args:
|
||
device_ids: 设备ID列表（必需）- 支持批量注册
|
||
device_type: 设备类型（默认vehicle）
|
||
brand_name: 设备品牌（huarui或seer，默认huarui）
|
||
command_type: 指令类型（必需）- 只能处理一种指令类型，如"order"、"state"、"factsheet"、"instantActions"
|
||
handler: 消息处理函数（必需）
|
||
description: 设备描述信息
|
||
device_brand: 设备品牌（兼容参数，建议使用brand_name）
|
||
protocol_key: 自定义协议标识（已弃用）
|
||
auto_encode: 是否自动编码（已弃用）
|
||
**kwargs: 其他配置参数
|
||
"""
|
||
if handler is None:
|
||
raise ValueError("handler参数不能为空")
|
||
|
||
if not device_ids or len(device_ids) == 0:
|
||
raise ValueError("device_ids参数不能为空")
|
||
|
||
# 确保device_type是DeviceType枚举
|
||
if isinstance(device_type, str):
|
||
try:
|
||
device_type = DeviceType(device_type)
|
||
except ValueError:
|
||
valid_types = [dt.value for dt in DeviceType]
|
||
raise ValueError(f"device_type参数错误：'{device_type}' 不是有效的设备类型。支持的类型有：{', '.join(valid_types)}")
|
||
|
||
# 兼容性处理：如果提供了device_brand，使用它覆盖brand_name
|
||
if device_brand is not None:
|
||
brand_name = device_brand
|
||
|
||
# 过滤已弃用的参数
|
||
filtered_kwargs = {k: v for k, v in kwargs.items()
|
||
if k not in ['listen_topics', 'forward_topics', 'protocol_key', 'auto_encode']}
|
||
|
||
# 调用新的设备服务批量注册接口
|
||
registered_device_ids = self.device_service.register_and_run(
|
||
device_ids=device_ids,
|
||
device_type=device_type,
|
||
brand_name=brand_name,
|
||
command_type=command_type,
|
||
handler=handler,
|
||
script_id=self.script_id,
|
||
description=description,
|
||
**filtered_kwargs
|
||
)
|
||
|
||
brand_info = f" (品牌: {brand_name})" if brand_name else ""
|
||
self.logger.info(f"设备处理器批量注册成功: {len(registered_device_ids)}个 {device_type.value} 设备{brand_info}")
|
||
return registered_device_ids
|
||
|
||
def register_single_device(self, device_id: str, device_type: Union[str, DeviceType],
|
||
listen_topics: List[str], forward_topics: List[str] = None,
|
||
handler: Callable = None, description: str = "",
|
||
device_brand: str = None, protocol_key: str = None,
|
||
auto_encode: bool = True, **kwargs):
|
||
"""兼容旧接口：注册单个设备处理器
|
||
|
||
这是为了兼容旧版本代码而保留的接口，建议使用新的register_and_run方法
|
||
"""
|
||
if handler is None:
|
||
raise ValueError("handler参数不能为空")
|
||
|
||
if not listen_topics:
|
||
raise ValueError("listen_topics参数不能为空")
|
||
|
||
# 这里可以实现旧版本的直接topic注册逻辑
|
||
# 为了简化，我们可以发出警告并建议用户升级到新接口
|
||
self.logger.warning(f"使用了已弃用的register_single_device接口，建议升级到新的register_and_run接口")
|
||
|
||
# 简单实现：调用新接口
|
||
return self.register_and_run(
|
||
device_ids=[device_id],
|
||
device_type=device_type,
|
||
brand_name=device_brand or "huarui",
|
||
handler=handler,
|
||
description=description,
|
||
**kwargs
|
||
)
|
||
|
||
def stop_handler(self, device_id: str):
|
||
"""停止指定设备处理器的运行"""
|
||
self.device_service.stop_handler(device_id)
|
||
self.logger.info(f"设备处理器已停止: {device_id}")
|
||
|
||
def restart_handler(self, device_id: str):
|
||
"""重启设备处理器"""
|
||
# 获取设备配置
|
||
handler_info = self.registry.get_device_handler(device_id)
|
||
if not handler_info:
|
||
raise ValueError(f"设备处理器不存在: {device_id}")
|
||
|
||
# 先停止
|
||
self.stop_handler(device_id)
|
||
|
||
# 重新注册并运行
|
||
self.register_and_run(
|
||
device_id=handler_info["device_id"],
|
||
device_type=handler_info["device_type"],
|
||
listen_topics=handler_info["listen_topics"],
|
||
forward_topics=handler_info["forward_topics"],
|
||
handler=handler_info["handler"],
|
||
description=handler_info["description"]
|
||
)
|
||
|
||
self.logger.info(f"设备处理器已重启: {device_id}")
|
||
|
||
def get_running_handlers(self) -> Dict[str, Any]:
|
||
"""获取所有正在运行的设备处理器"""
|
||
return self.device_service.get_running_handlers()
|
||
|
||
def get_handler_status(self, device_id: str) -> Dict[str, Any]:
|
||
"""获取设备处理器状态"""
|
||
return self.device_service.get_handler_status(device_id)
|
||
|
||
async def publish_message(self, topic: str, payload: Any):
|
||
"""主动发布MQTT消息
|
||
|
||
使用方式：
|
||
await VWED.device.publish_message(
|
||
topic="factory/agv/command",
|
||
payload={"action": "move", "x": 100, "y": 200}
|
||
)
|
||
"""
|
||
await self.device_service.publish_message(topic, payload)
|
||
self.logger.info(f"MQTT消息已发布: {topic}")
|
||
|
||
def sync_publish_message(self, topic: str, payload: Any):
|
||
"""同步方式发布MQTT消息"""
|
||
asyncio.create_task(self.publish_message(topic, payload))
|
||
|
||
def subscribe_topic(self, topic: str, handler: Callable):
|
||
"""订阅额外的MQTT topic
|
||
|
||
注意：这是临时订阅，建议使用register_and_run进行完整的设备注册
|
||
"""
|
||
# 创建临时设备ID
|
||
temp_device_id = f"temp_{self.script_id}_{topic.replace('/', '_')}"
|
||
|
||
self.register_and_run(
|
||
device_id=temp_device_id,
|
||
device_type="custom",
|
||
listen_topics=[topic],
|
||
forward_topics=[],
|
||
handler=handler,
|
||
description=f"临时订阅topic: {topic}"
|
||
)
|
||
|
||
return temp_device_id
|
||
|
||
def unsubscribe_topic(self, device_id: str):
|
||
"""取消订阅topic（通过停止设备处理器）"""
|
||
self.stop_handler(device_id)
|
||
|
||
def get_device_types(self) -> List[str]:
|
||
"""获取支持的设备类型列表"""
|
||
return [dt.value for dt in DeviceType]
|
||
|
||
def get_device_brands(self) -> List[str]:
|
||
"""获取支持的设备品牌列表"""
|
||
return [db.value for db in DeviceBrand]
|
||
|
||
def get_protocols(self) -> Dict[str, Dict[str, Any]]:
|
||
"""获取所有可用的协议"""
|
||
protocol_registry = get_protocol_registry()
|
||
return protocol_registry.list_protocols()
|
||
|
||
def register_custom_protocol(self, protocol_key: str, brand: str, device_type: str,
|
||
encode_func: Callable, decode_func: Callable,
|
||
supported_commands: List[str] = None):
|
||
"""注册自定义设备协议
|
||
|
||
使用方式：
|
||
def my_encode(command):
|
||
# 自定义编码逻辑
|
||
return {"custom_format": command}
|
||
|
||
def my_decode(response):
|
||
# 自定义解码逻辑
|
||
return {"decoded": response}
|
||
|
||
VWED.device.register_custom_protocol(
|
||
protocol_key="my_brand_vehicle",
|
||
brand="my_brand",
|
||
device_type="vehicle",
|
||
encode_func=my_encode,
|
||
decode_func=my_decode,
|
||
supported_commands=["move", "stop"]
|
||
)
|
||
"""
|
||
protocol_registry = get_protocol_registry()
|
||
protocol_registry.register_custom_protocol(
|
||
protocol_key, brand, device_type,
|
||
encode_func, decode_func, supported_commands
|
||
)
|
||
|
||
self.logger.info(f"自定义协议已注册: {protocol_key} ({brand}_{device_type})")
|
||
|
||
def test_protocol_encoding(self, protocol_key: str, test_command: Dict[str, Any]) -> Dict[str, Any]:
|
||
"""测试协议编码功能
|
||
|
||
使用方式：
|
||
result = VWED.device.test_protocol_encoding(
|
||
protocol_key="huarui_vehicle",
|
||
test_command={"action": "move", "x": 100, "y": 200}
|
||
)
|
||
"""
|
||
protocol_registry = get_protocol_registry()
|
||
protocol = protocol_registry.get_protocol(protocol_key)
|
||
|
||
if not protocol:
|
||
return {
|
||
"success": False,
|
||
"error": f"协议不存在: {protocol_key}"
|
||
}
|
||
|
||
try:
|
||
encoded = protocol.encode_command(test_command)
|
||
return {
|
||
"success": True,
|
||
"original": test_command,
|
||
"encoded": encoded,
|
||
"protocol": protocol_key
|
||
}
|
||
except Exception as e:
|
||
return {
|
||
"success": False,
|
||
"error": f"编码失败: {str(e)}",
|
||
"protocol": protocol_key
|
||
}
|
||
|
||
def create_message_filter(self, conditions: Dict[str, Any]) -> Callable:
|
||
"""创建消息过滤器
|
||
|
||
使用方式：
|
||
filter_func = VWED.device.create_message_filter({
|
||
"device_type": "vehicle",
|
||
"action": "move"
|
||
})
|
||
|
||
def my_handler(message):
|
||
if filter_func(message):
|
||
# 处理符合条件的消息
|
||
pass
|
||
"""
|
||
def filter_func(message):
|
||
payload = message.payload if hasattr(message, 'payload') else message
|
||
|
||
for key, expected_value in conditions.items():
|
||
if key == "device_type":
|
||
actual_value = message.device_type.value if hasattr(message, 'device_type') else None
|
||
else:
|
||
actual_value = payload.get(key)
|
||
|
||
if actual_value != expected_value:
|
||
return False
|
||
return True
|
||
|
||
return filter_func |