You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

535 lines
23 KiB
Python

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

from utils.DBModels.ProtocolModel import (
ModbusTcpMasterVar, ModbusTcpSlaveVar, ModbusRtuMasterVar, ModbusRtuSlaveVar,
HartVar, TcRtdVar, AnalogVar, HartSimulateVar
)
from model.ProjectModel.GlobalConfigManager import GlobalConfigManager
from protocol.HartRtuSlaveManager import HartRtuSlaveManager
from protocol.TCP.TCPVarManage import *
from protocol.TCP.TemToMv import temToMv
from protocol.RPC.RpcClient import RpcClient
from protocol.RPC.RpcServer import RpcServer
from protocol.ModBus.ModbusManager import ModbusManager
from utils import Globals
import threading
import time
class ProtocolManage(object):
"""通讯变量查找类,用于根据变量名在数据库模型中检索变量信息"""
def __init__(self):
# 根据全局配置动态获取启用的模型类
self.MODEL_CLASSES = GlobalConfigManager.getEnabledModelClasses()
# 根据配置决定是否初始化TCPVarManager仅当启用IO或TCRTD模块时
if GlobalConfigManager.needsTcpVarManager():
# self.tcpVarManager = TCPVarManager('192.168.1.50', 5055)
self.tcpVarManager = TCPVarManager('127.0.0.1', 8000)
self.writeTC = [0] * 8
self.writeRTD = [0] * 8
else:
self.tcpVarManager = None
self.writeTC = None
self.writeRTD = None
self.RpcClient = None
self.RpcServer = None
self.varInfoCache = {} # 保持驼峰命名
self.historyDBManage = Globals.getValue('historyDBManage')
self.variableValueCache = {} # {varName: value}
# Modbus 管理器
self.modbusManager = ModbusManager()
# self.modbusManager.setVariableCache(self.variableValueCache, None, self.varInfoCache)
# HART模拟RTU从站管理器
self.hartRtuSlaveManager = None
if GlobalConfigManager.isModuleEnabled('hartSimulateModule'):
try:
self.hartRtuSlaveManager = HartRtuSlaveManager()
print("HART模拟RTU从站管理器已初始化")
except Exception as e:
print(f"初始化HART模拟RTU从站管理器失败: {e}")
self.hartRtuSlaveManager = None
self.refreshVarCache()
self.cacheLock = threading.Lock()
# 设置 Modbus 管理器的缓存锁
self.modbusManager.setVariableCache(self.variableValueCache, self.cacheLock, self.varInfoCache)
self.readThreadStop = threading.Event()
self.readThread = threading.Thread(target=self._backgroundReadAllVariables, daemon=True)
self.readThread.start()
def clearVarCache(self):
"""清空变量信息缓存"""
self.varInfoCache.clear()
def refreshVarCache(self):
"""重新加载所有变量信息到缓存(可选实现)"""
self.varInfoCache.clear()
# 重新获取最新的启用模型类列表
self.MODEL_CLASSES = GlobalConfigManager.getEnabledModelClasses()
for modelClass in self.MODEL_CLASSES:
try:
for varInstance in modelClass.select():
varName = getattr(varInstance, 'varName', None)
if varName:
varData = {}
for field in varInstance._meta.sorted_fields:
fieldName = field.name
varData[fieldName] = getattr(varInstance, fieldName)
self.varInfoCache[varName] = {
'modelType': modelClass.__name__,
'variableData': varData
}
except Exception as e:
print(f"刷新缓存时出错: {modelClass.__name__}: {e}")
def lookupVariable(self, variableName):
"""
根据变量名检索变量信息(优先查缓存)
:param variableName: 要查询的变量名
:return: 包含变量信息和模型类型的字典如果未找到返回None
"""
if variableName in self.varInfoCache:
return self.varInfoCache[variableName]
for modelClass in self.MODEL_CLASSES:
varInstance = modelClass.getByName(variableName)
if varInstance:
varData = {}
for field in varInstance._meta.sorted_fields:
fieldName = field.name
varData[fieldName] = getattr(varInstance, fieldName)
result = {
'modelType': modelClass.__name__,
'variableData': varData
}
self.varInfoCache[variableName] = result # 写入缓存
return result
return None
def setClentMode(self, clentName, rabbitmqHost='localhost'):
if self.RpcClient:
self.RpcClient.close()
self.RpcClient = RpcClient(clentName, rabbitmqHost, self)
# 使用非阻塞方式启动RPC客户端
self.RpcClient.startNonBlocking()
def closeClient(self):
if self.RpcClient:
self.RpcClient.close()
self.RpcClient = None
def setServerMode(self, rabbitmqHost='localhost'):
if self.RpcServer:
self.RpcServer.close()
self.RpcServer = RpcServer(rabbitmqHost)
def addClient(self, clientName):
if self.RpcServer:
self.RpcServer.addClient(clientName=clientName)
else:
return
def closeServer(self):
if self.RpcServer:
self.RpcServer.close()
self.RpcServer = None
def writeVariableValue(self, variableName, value, trigger=None, timeoutMS=2000):
"""
根据变量名写入变量值,根据变量类型进行不同处理(不保存到数据库)
:param variableName: 变量名
:param value: 要写入的值
:return: 写入成功返回True否则返回False
"""
varInfo = self.lookupVariable(variableName)
if not varInfo:
if self.RpcServer:
try:
existsVar, clientNames = self.RpcServer.existsVar(variableName)
if existsVar:
result = self.RpcServer.writeVar(variableName, value)
return result.get("result") == "success" if result else False
else:
return False
except Exception as e:
print(f"RPC写入变量 {variableName} 失败: {e}")
return False
return False
modelType = varInfo['modelType']
info = varInfo['variableData']
try:
# 拆分为四个独立的Modbus协议条件判断
if modelType == 'ModbusTcpMasterVar':
res = self.modbusManager.writeModbusTcpMasterValue(info, value)
# print(res)
if res == 'error':
return
elif modelType == 'ModbusTcpSlaveVar':
res = self.modbusManager.writeModbusTcpSlaveValue(info, value)
if res == 'error':
return
elif modelType == 'ModbusRtuMasterVar':
res = self.modbusManager.writeModbusRtuMasterValue(info, value)
if res == 'error':
return
elif modelType == 'ModbusRtuSlaveVar':
res = self.modbusManager.writeModbusRtuSlaveValue(info, value)
if res == 'error':
return
# HART协议变量处理
elif modelType == 'HartVar':
pass
# 温度/RTD变量处理
elif modelType == 'TcRtdVar':
if not self.tcpVarManager:
print("TCPVarManager未初始化无法处理TcRtdVar变量")
return False
channel = int(info['channelNumber']) - 1
varType = info['varType']
compensationVar = float(info['compensationVar'])
varModel = info['varModel']
model = self.getModelType(varModel) if self.getModelType(varModel) else localModel
if model == localModel:
if varType == 'PT100':
self.writeRTD[channel] = value
else:
self.writeTC[channel] = value
if varType in ['R', 'S', 'B', 'J', 'T', 'E', 'K', 'N', 'C', 'A', 'PT100'] and model != SimModel:
value = temToMv(varType, value + compensationVar)
if trigger and varType in ['R', 'S', 'B', 'J', 'T', 'E', 'K', 'N', 'C', 'A']:
trigger = TCTrigger
if trigger and varType in ['PT100']:
trigger = RTDTrigger
self.tcpVarManager.writeValue(varType, channel, value, trigger=trigger, model=model, timeoutMS=timeoutMS)
# 模拟量变量处理
elif modelType == 'AnalogVar':
if not self.tcpVarManager:
print("TCPVarManager未初始化无法处理AnalogVar变量")
return False
channel = int(info['channelNumber']) - 1
varType = info['varType']
varModel = info['varModel']
model = self.getModelType(varModel) if self.getModelType(varModel) else localModel
if info['varType'] in ['AI', 'AO']:
value = self.getRealAO(value, info['max'], info['min'])
trigger = FPGATrigger if trigger else trigger
self.tcpVarManager.writeValue(varType, channel, value, trigger=trigger, model=model, timeoutMS=timeoutMS)
# HART模拟变量处理
elif modelType == 'HartSimulateVar':
if not self.hartRtuSlaveManager:
print("HART RTU从站管理器未初始化无法处理HartSimulateVar变量")
return False
# 根据变量名确定是主变量还是动态变量
varName = variableName.lower()
if 'primary' in varName or '主变量' in variableName:
success = self.hartRtuSlaveManager.writeVariable('primaryVariable', float(value))
elif 'dynamic1' in varName or '动态变量1' in variableName:
success = self.hartRtuSlaveManager.writeVariable('dynamicVariable1', float(value))
elif 'dynamic2' in varName or '动态变量2' in variableName:
success = self.hartRtuSlaveManager.writeVariable('dynamicVariable2', float(value))
elif 'dynamic3' in varName or '动态变量3' in variableName:
success = self.hartRtuSlaveManager.writeVariable('dynamicVariable3', float(value))
else:
# 默认写入主变量
success = self.hartRtuSlaveManager.writeVariable('primaryVariable', float(value))
if not success:
print(f"HART模拟变量 {variableName} 写入失败")
return False
if self.RpcClient:
self.RpcClient.setVarContent(variableName, value, info['min'], info['max'], info['varType'])
return True
except Exception as e:
print(f"写入变量值失败: {str(e)}")
return False
def _backgroundReadAllVariables(self, interval=1.0):
"""后台读取所有变量增加间隔减少RPC调用频率"""
while not self.readThreadStop.is_set():
try:
allVarNames = list(self.getAllVariableNames())
for varName in allVarNames:
if self.readThreadStop.is_set():
break
try:
value = self._readVariableValueOriginal(varName)
with self.cacheLock:
self.variableValueCache[varName] = value
except Exception as e:
# 单个变量读取失败不影响其他变量
print(f"读取变量 {varName} 失败: {e}")
continue
# 增加间隔减少RPC调用频率
time.sleep(interval)
except Exception as e:
print(f"后台读取线程异常: {e}")
time.sleep(5) # 异常时等待更长时间
def getAllVariableNames(self):
return list(self.varInfoCache.keys())
def _readVariableValueOriginal(self, variableName):
varInfo = self.lookupVariable(variableName)
value = None
if not varInfo:
if self.RpcServer:
try:
# 使用safeRpcCall减少阻塞
existsVar, clientNames = self.safeRpcCall(
self.RpcServer.existsVar, variableName
) or (False, None)
if existsVar and clientNames:
varData = self.safeRpcCall(
self.RpcServer.getVarValue, clientNames[0], variableName
)
if varData and 'value' in varData:
# 安全地转换值
try:
value = float(varData['value'])
except (ValueError, TypeError):
print(f"无法转换变量 {variableName} 的值: {varData['value']}")
return None
else:
return None
else:
return None
except Exception as e:
print(f"RPC读取变量 {variableName} 失败: {e}")
return None
return None
modelType = varInfo['modelType']
info = varInfo['variableData']
try:
# 拆分为独立的协议条件判断
# if modelType == 'ModbusTcpMasterVar':
# value = self.modbusManager.readModbusTcpMasterValue(info)
if modelType == 'ModbusTcpSlaveVar':
value = self.modbusManager.readModbusTcpSlaveValue(info)
# elif modelType == 'ModbusRtuMasterVar':
# value = self.modbusManager.readModbusRtuMasterValue(info)
elif modelType == 'ModbusRtuSlaveVar':
value = self.modbusManager.readModbusRtuSlaveValue(info)
elif modelType == 'HartVar':
pass
elif modelType == 'TcRtdVar':
if not self.tcpVarManager:
return None
channel = int(info['channelNumber']) - 1
varType = info['varType']
varModel = info['varModel']
model = self.getModelType(varModel)
if model == SimModel:
if varType == 'PT100':
value = self.tcpVarManager.simRTDData[channel]
elif varType in ['R', 'S', 'B', 'J', 'T', 'E', 'K', 'N', 'C', 'A']:
value = self.tcpVarManager.simTCData[channel]
else:
if varType == 'PT100':
value = self.writeRTD[channel] if self.writeRTD else 0
elif varType in ['R', 'S', 'B', 'J', 'T', 'E', 'K', 'N', 'C', 'A']:
value = self.writeTC[channel] if self.writeTC else 0
elif modelType == 'AnalogVar':
if not self.tcpVarManager:
return None
channel = int(info['channelNumber']) - 1
varType = info['varType']
varModel = info['varModel']
model = self.getModelType(varModel)
value = self.tcpVarManager.readValue(varType, channel, model=model)
if varType in ['AI', 'AO']:
value = self.getRealAI(value, info['max'], info['min'])
elif modelType == 'HartSimulateVar':
if not self.hartRtuSlaveManager:
return None
# 根据变量名确定是主变量还是动态变量
varName = variableName.lower()
if 'primary' in varName or '主变量' in variableName:
value = self.hartRtuSlaveManager.readVariable('primaryVariable')
elif 'dynamic1' in varName or '动态变量1' in variableName:
value = self.hartRtuSlaveManager.readVariable('dynamicVariable1')
elif 'dynamic2' in varName or '动态变量2' in variableName:
value = self.hartRtuSlaveManager.readVariable('dynamicVariable2')
elif 'dynamic3' in varName or '动态变量3' in variableName:
value = self.hartRtuSlaveManager.readVariable('dynamicVariable3')
else:
# 默认读取主变量
value = self.hartRtuSlaveManager.readVariable('primaryVariable')
if value is not None and value != 'error':
if self.RpcClient:
self.RpcClient.setVarContent(variableName, value, info['min'], info['max'], info['varType'])
self.historyDBManage.writeVarValue(variableName, value)
return value
else:
return None
except Exception as e:
print(f"读取变量值失败: {str(e)}")
return None
def readVariableValue(self, variableName):
with self.cacheLock:
if variableName in self.variableValueCache:
return self.variableValueCache[variableName]
return self._readVariableValueOriginal(variableName)
def recvDeltaT(self):
if self.tcpVarManager:
return self.tcpVarManager.recvDeltaT()
return None
def shutdown(self):
if self.tcpVarManager:
self.tcpVarManager.shutdown()
# 关闭所有Modbus通讯
self.modbusManager.stopAllModbus()
# 关闭HART RTU从站
if self.hartRtuSlaveManager:
self.hartRtuSlaveManager.stopSlave()
self.closeClient()
self.closeServer()
# 关闭后台读取线程
if hasattr(self, 'readThreadStop') and hasattr(self, 'readThread'):
self.readThreadStop.set()
self.readThread.join(timeout=1)
def getRealAO(self, value, highValue, lowValue):
if highValue:
lowValue = float(lowValue)
highValue = float(highValue)
return (16 * (value - lowValue) + 4 * (highValue - lowValue)) / (1000 * (highValue - lowValue))
else:
return value / 1000
def getRealAI(self, mA, highValue, lowValue):
"""将毫安值转换为实际工程值getRealAO的反向计算"""
try:
if highValue:
lowValue = float(lowValue)
highValue = float(highValue)
return (1000 * mA * (highValue - lowValue) - 4 * (highValue - lowValue)) / 16.0 + lowValue
else:
return mA * 1000
except Exception as e:
print(f"工程值转换失败: {str(e)}")
return 0.0
def getModelType(self, varModel):
if varModel == '本地值':
return localModel
elif varModel == '远程值':
return NetModel
elif varModel == '模拟值':
return SimModel
def disconnectClient(self, clientName):
if self.RpcServer:
self.RpcServer.removeClient(clientName)
def safeRpcCall(self, method, *args, **kwargs):
"""安全的RPC调用带超时和异常处理"""
if not self.RpcServer:
return None
try:
# 直接调用,但使用较短的超时时间
result = method(*args, **kwargs)
return result
except Exception as e:
# 静默处理异常,避免日志过多
return None
# ==================== Modbus 通讯管理方法(委托给 ModbusManager ====================
def startModbusTcpMaster(self):
"""启动 Modbus TCP 主站"""
return self.modbusManager.startModbusTcpMaster()
def stopModbusTcpMaster(self):
"""停止 Modbus TCP 主站"""
return self.modbusManager.stopModbusTcpMaster()
def startModbusRtuMaster(self):
"""启动 Modbus RTU 主站"""
return self.modbusManager.startModbusRtuMaster()
def stopModbusRtuMaster(self):
"""停止 Modbus RTU 主站"""
return self.modbusManager.stopModbusRtuMaster()
def startModbusTcpSlave(self):
"""启动 Modbus TCP 从站"""
return self.modbusManager.startModbusTcpSlave()
def stopModbusTcpSlave(self):
"""停止 Modbus TCP 从站"""
return self.modbusManager.stopModbusTcpSlave()
def startModbusRtuSlave(self):
"""启动 Modbus RTU 从站"""
return self.modbusManager.startModbusRtuSlave()
def stopModbusRtuSlave(self):
"""停止 Modbus RTU 从站"""
return self.modbusManager.stopModbusRtuSlave()
def stopAllModbus(self):
"""停止所有 Modbus 通讯"""
return self.modbusManager.stopAllModbus()
def getModbusStatus(self):
"""获取所有 Modbus 通讯状态"""
return self.modbusManager.getModbusStatus()
def getModbusMessages(self):
"""获取 Modbus 报文信息"""
return self.modbusManager.getMessages()
def clearModbusMessages(self):
"""清空 Modbus 报文记录"""
self.modbusManager.clearMessages()
# ==================== HART模拟RTU从站管理方法 ====================
def startHartRtuSlave(self):
"""启动HART RTU从站"""
if not self.hartRtuSlaveManager:
return False
return self.hartRtuSlaveManager.startSlave()
def stopHartRtuSlave(self):
"""停止HART RTU从站"""
if self.hartRtuSlaveManager:
self.hartRtuSlaveManager.stopSlave()
def writeHartVariable(self, variableName: str, value: float) -> bool:
"""写入HART变量值"""
if not self.hartRtuSlaveManager:
return False
return self.hartRtuSlaveManager.writeVariable(variableName, value)
def readHartVariable(self, variableName: str):
"""读取HART变量值"""
if not self.hartRtuSlaveManager:
return None
return self.hartRtuSlaveManager.readVariable(variableName)