stage_one

hardware/transmitter.py

@@ -3,12 +3,19 @@ from pymodbus.exceptions import ModbusException
 import socket
 from config.ini_manager import ini_manager
 from config.settings import app_set_config
+import time
 
 
 
 class TransmitterController:
     def __init__(self, relay_controller):
         self.relay_controller = relay_controller
+        self.test_upper_weight=5043
+        self.test_lower_weight=1256
+        self.is_start_upper=False
+        self.is_start_lower=False
+        self.start_time_upper=None
+        self.start_time_lower=None
         # 变送器配置
         self.config = {
             1: {  # 上料斗
@@ -73,7 +80,8 @@ class TransmitterController:
             self.relay_controller.modbus_client.close()
 
     # 直接读取 变送器返回的数据并解析
-    def read_data(self, transmitter_id):
+    def read_data_normal(self, transmitter_id):
+        
         """
             Args: transmitter_id 为1 表示上料斗, 为2 表示下料斗
             return: 读取成功返回重量 weight: int, 失败返回 None
@@ -82,9 +90,7 @@ class TransmitterController:
         BUFFER_SIZE= 1024
         IP = None
         PORT = None
-        weight = 0
-
-
+        weight = 0  
         if transmitter_id == 1:
             # 上料斗变送器的信息：
             IP = ini_manager.upper_transmitter_ip
@@ -97,8 +103,86 @@ class TransmitterController:
         if not IP or not PORT:
             print(f"未配置变送器 {transmitter_id} 的IP或PORT")
             return 0
-        if app_set_config.debug_mode:
-            print(f"调试模式，未读数据({transmitter_id},IP: {IP}:{PORT})")
+        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+            try:
+                s.settimeout(TIMEOUT)
+                s.connect((IP, PORT))
+                # print(f"连接上料斗变送器 {IP}:{PORT} 成功")
+                
+                # 接收数据（变送器主动推送，recv即可获取数据）
+                data = s.recv(BUFFER_SIZE)
+                if data:
+                    # print(f"收到原始数据：{data}")
+
+                    # 提取出完整的一个数据包 (\r\n结尾)
+                    packet = self.get_latest_valid_packet(data)
+                    if not packet:
+                        print("未获取到有效数据包!!")
+                        return None
+                    # 解析重量
+                    weight = self.parse_weight(packet)
+                else:
+                    print("未收到设备数据")
+            
+            except ConnectionRefusedError:
+                print(f"变送器连接失败：{IP}:{PORT} 拒绝连接（设备离线/端口错误）")
+            except socket.timeout:
+                print(f"读取变送器数据超时：{TIMEOUT}秒内未收到数据")
+            except Exception as e:
+                print(f"读取异常：{e}")
+        
+        # 成功返回重量（int），失败返回None
+        return weight 
+
+    def read_data(self, transmitter_id):
+        
+        """
+        测试用：模拟读取变送器数据mock
+            Args: transmitter_id 为1 表示上料斗, 为2 表示下料斗
+            return: 读取成功返回重量 weight: int, 失败返回 None
+        """
+        TIMEOUT = 2 # 超时时间为 2秒
+        BUFFER_SIZE= 1024
+        IP = None
+        PORT = None
+        weight = 0
+        
+
+        if transmitter_id == 1:
+            # 上料斗变送器的信息：
+            IP = ini_manager.upper_transmitter_ip
+            PORT = ini_manager.upper_transmitter_port
+
+            if self.is_start_upper:
+                if self.start_time_upper is None:
+                    self.start_time_upper=time.time()
+                weight=self.test_upper_weight-50*(time.time()-self.start_time_upper)
+                if weight<0:
+                    weight=0
+            else:
+                weight=self.test_upper_weight
+                self.start_time_upper=None
+
+            return weight
+
+        elif transmitter_id == 2:
+            # 下料斗变送器的信息：
+            IP = ini_manager.lower_transmitter_ip
+            PORT = ini_manager.lower_transmitter_port
+
+            if self.is_start_lower:
+                if self.start_time_lower is None:
+                    self.start_time_lower=time.time()
+                weight=self.test_lower_weight-50*(time.time()-self.start_time_lower)
+            else:
+                weight=self.test_lower_weight
+                self.start_time_lower=None
+
+            return weight
+            
+        if not IP or not PORT:
+            print(f"未配置变送器 {transmitter_id} 的IP或PORT")
+            return 0
         with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
             try:
                 s.settimeout(TIMEOUT)