241 lines
9.2 KiB
Python
241 lines
9.2 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
Modbus TCP 模拟服务器
|
||
用于测试在线脚本中的Modbus通信功能
|
||
"""
|
||
|
||
import time
|
||
import threading
|
||
from pymodbus.server.startstop import StartTcpServer
|
||
from pymodbus.device import ModbusDeviceIdentification
|
||
from pymodbus.datastore import ModbusSequentialDataBlock, ModbusSlaveContext, ModbusServerContext
|
||
from pymodbus.datastore.context import ModbusServerContext
|
||
import logging
|
||
|
||
# 配置日志
|
||
logging.basicConfig(level=logging.INFO)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class ModbusTestServer:
|
||
"""Modbus TCP 测试服务器"""
|
||
|
||
def __init__(self, host='localhost', port=5020):
|
||
self.host = host
|
||
self.port = port
|
||
self.server_thread = None
|
||
self.running = False
|
||
|
||
# 创建数据存储
|
||
# 参数：(起始地址, 初始值列表)
|
||
# 线圈 (0x区域) - 可读写的离散值
|
||
coils = ModbusSequentialDataBlock(0, [False] * 100)
|
||
|
||
# 离散输入 (1x区域) - 只读的离散值
|
||
discrete_inputs = ModbusSequentialDataBlock(0, [True, False, True, False] * 25)
|
||
|
||
# 输入寄存器 (3x区域) - 只读的16位值
|
||
input_registers = ModbusSequentialDataBlock(0, [100, 200, 300, 400] * 25)
|
||
|
||
# 保持寄存器 (4x区域) - 可读写的16位值
|
||
holding_registers = ModbusSequentialDataBlock(0, [1000, 2000, 3000, 4000] * 25)
|
||
|
||
# 创建从站上下文
|
||
slave_context = ModbusSlaveContext(
|
||
di=discrete_inputs, # 离散输入
|
||
co=coils, # 线圈
|
||
hr=holding_registers, # 保持寄存器
|
||
ir=input_registers # 输入寄存器
|
||
)
|
||
|
||
# 创建服务器上下文（支持从站ID 1-10）
|
||
self.context = ModbusServerContext(slaves={
|
||
1: slave_context, # 从站ID 1
|
||
2: slave_context, # 从站ID 2 (共享同一个上下文，简化测试)
|
||
}, single=False)
|
||
|
||
# 设备信息
|
||
self.identity = ModbusDeviceIdentification()
|
||
self.identity.VendorName = 'VWED Test'
|
||
self.identity.ProductCode = 'VWED-MODBUS-SIM'
|
||
self.identity.VendorUrl = 'https://github.com/vwed'
|
||
self.identity.ProductName = 'VWED Modbus TCP Simulator'
|
||
self.identity.ModelName = 'Test Server v1.0'
|
||
self.identity.MajorMinorRevision = '1.0.0'
|
||
|
||
# 启动数据更新线程
|
||
self.update_thread = None
|
||
|
||
def start_server(self):
|
||
"""启动Modbus服务器"""
|
||
if self.running:
|
||
logger.warning("服务器已经在运行")
|
||
return
|
||
|
||
logger.info(f"启动Modbus TCP服务器: {self.host}:{self.port}")
|
||
|
||
self.running = True
|
||
|
||
# 启动数据更新线程
|
||
self.update_thread = threading.Thread(target=self._update_data_loop, daemon=True)
|
||
self.update_thread.start()
|
||
|
||
# 启动服务器（在单独线程中）
|
||
self.server_thread = threading.Thread(
|
||
target=self._run_server,
|
||
daemon=True
|
||
)
|
||
self.server_thread.start()
|
||
|
||
logger.info("Modbus TCP服务器启动成功")
|
||
|
||
def _run_server(self):
|
||
"""运行服务器"""
|
||
try:
|
||
StartTcpServer(
|
||
context=self.context,
|
||
identity=self.identity,
|
||
address=(self.host, self.port)
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"服务器运行错误: {e}")
|
||
self.running = False
|
||
|
||
def _update_data_loop(self):
|
||
"""更新测试数据的循环"""
|
||
counter = 0
|
||
while self.running:
|
||
try:
|
||
# 更新输入寄存器数据（模拟传感器读数）
|
||
for slave_id in [1, 2]:
|
||
slave_context = self.context[slave_id]
|
||
|
||
# 更新输入寄存器（模拟变化的传感器数据）
|
||
new_values = [
|
||
100 + counter % 100, # 地址0: 100-199循环
|
||
200 + (counter * 2) % 200, # 地址1: 200-399循环
|
||
300 + (counter * 3) % 300, # 地址2: 300-599循环
|
||
400 + (counter * 4) % 400, # 地址3: 400-799循环
|
||
]
|
||
|
||
for i, value in enumerate(new_values):
|
||
slave_context.setValues(3, i, [value]) # 3 = 输入寄存器
|
||
|
||
# 更新离散输入（模拟开关状态）
|
||
discrete_values = [
|
||
counter % 2 == 0, # 地址0: 每秒切换
|
||
counter % 4 < 2, # 地址1: 每2秒切换
|
||
counter % 6 < 3, # 地址2: 每3秒切换
|
||
counter % 8 < 4, # 地址3: 每4秒切换
|
||
]
|
||
|
||
for i, value in enumerate(discrete_values):
|
||
slave_context.setValues(2, i, [value]) # 2 = 离散输入
|
||
|
||
counter += 1
|
||
time.sleep(1) # 每秒更新一次
|
||
|
||
except Exception as e:
|
||
logger.error(f"数据更新错误: {e}")
|
||
time.sleep(1)
|
||
|
||
def stop_server(self):
|
||
"""停止服务器"""
|
||
logger.info("正在停止Modbus TCP服务器...")
|
||
self.running = False
|
||
|
||
def get_server_info(self):
|
||
"""获取服务器信息"""
|
||
return {
|
||
'host': self.host,
|
||
'port': self.port,
|
||
'running': self.running,
|
||
'slaves': list(self.context.slaves()) if hasattr(self.context, 'slaves') and callable(self.context.slaves) else [1, 2],
|
||
}
|
||
|
||
def set_coil_value(self, slave_id, address, value):
|
||
"""设置线圈值（用于测试）"""
|
||
try:
|
||
slave_context = self.context[slave_id]
|
||
slave_context.setValues(1, address, [bool(value)]) # 1 = 线圈
|
||
logger.info(f"设置线圈 slave={slave_id}, addr={address}, value={value}")
|
||
except Exception as e:
|
||
logger.error(f"设置线圈失败: {e}")
|
||
|
||
def set_holding_register_value(self, slave_id, address, value):
|
||
"""设置保持寄存器值（用于测试）"""
|
||
try:
|
||
slave_context = self.context[slave_id]
|
||
slave_context.setValues(4, address, [int(value)]) # 4 = 保持寄存器
|
||
logger.info(f"设置保持寄存器 slave={slave_id}, addr={address}, value={value}")
|
||
except Exception as e:
|
||
logger.error(f"设置保持寄存器失败: {e}")
|
||
|
||
def get_all_values(self, slave_id=1):
|
||
"""获取所有值（用于调试）"""
|
||
try:
|
||
slave_context = self.context[slave_id]
|
||
|
||
# 读取各种类型的数据
|
||
coils = slave_context.getValues(1, 0, 10) # 线圈
|
||
discrete = slave_context.getValues(2, 0, 10) # 离散输入
|
||
input_regs = slave_context.getValues(3, 0, 10) # 输入寄存器
|
||
holding_regs = slave_context.getValues(4, 0, 10) # 保持寄存器
|
||
|
||
return {
|
||
'coils': coils,
|
||
'discrete_inputs': discrete,
|
||
'input_registers': input_regs,
|
||
'holding_registers': holding_regs
|
||
}
|
||
except Exception as e:
|
||
logger.error(f"获取数据失败: {e}")
|
||
return None
|
||
|
||
|
||
def main():
|
||
"""主函数 - 启动测试服务器"""
|
||
print("=" * 60)
|
||
print("VWED Modbus TCP 测试服务器")
|
||
print("=" * 60)
|
||
|
||
# 创建并启动服务器
|
||
server = ModbusTestServer(host='localhost', port=5020)
|
||
server.start_server()
|
||
|
||
print(f"服务器信息: {server.get_server_info()}")
|
||
print("\n测试数据布局:")
|
||
print("- 线圈 (0x区域): 地址0-99, 初始值为False")
|
||
print("- 离散输入 (1x区域): 地址0-99, 动态变化的True/False")
|
||
print("- 输入寄存器 (3x区域): 地址0-99, 动态变化的数值")
|
||
print("- 保持寄存器 (4x区域): 地址0-99, 初始值1000,2000,3000,4000...")
|
||
print("\n支持的从站ID: 1, 2")
|
||
print(f"\n连接地址: {server.host}:{server.port}")
|
||
print("\n测试示例:")
|
||
print("python -c \"")
|
||
print("from services.online_script.built_in_modules.modbus_module import VWEDModbusModule")
|
||
print("m = VWEDModbusModule('test')")
|
||
print("print(m.read_holding_register('localhost', 5020, 1, 0, 2))")
|
||
print("\"")
|
||
|
||
try:
|
||
while True:
|
||
time.sleep(5)
|
||
# 显示当前数据状态
|
||
values = server.get_all_values(1)
|
||
if values:
|
||
print(f"\n[{time.strftime('%H:%M:%S')}] 从站1当前数据:")
|
||
print(f" 输入寄存器[0-3]: {values['input_registers'][:4]}")
|
||
print(f" 离散输入[0-3]: {values['discrete_inputs'][:4]}")
|
||
print(f" 保持寄存器[0-3]: {values['holding_registers'][:4]}")
|
||
|
||
except KeyboardInterrupt:
|
||
print("\n\n正在关闭服务器...")
|
||
server.stop_server()
|
||
print("服务器已关闭")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main() |