每日更新;

source/dev_EnergyRouter.py

@@ -4,21 +4,114 @@ from pathlib import Path
 from utl import trans_list_to_str
 from func_frame import make_frame_modbus, check_frame_modbus
 
+modbus_map = {
+}
 
 class EnergyRouter:
     def __init__(self, ip="192.168.100.10", port=7, adddr_modbus=0x01):
         self._ip = ip
         self._port = port
-        self._adddr_modbus =adddr_modbus
+        self._addr_modbus =adddr_modbus
 
         self.tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         self.tcp_socket.connect((self._ip, self._port))
 
-    def frame_update(fp):
-        pass
+    def frame_read(self, daddr=0x00, dlen=0x10):
+        block_modbus = {
+            'addr_dev' : self._addr_modbus,
+            'data_addr' : daddr,
+            'data_len' : dlen,
+            'type' : 'read',
+        }
+        frame = make_frame_modbus(block_modbus)
+
+        if self.tcp_socket is None:
+            print(trans_list_to_str(frame))
+            return
+        
+        # self.tcp_socket.recv(8)
+        self.tcp_socket.send(bytearray(frame))
+        time.sleep(0.5)
+        frame_recv = self.tcp_socket.recv(128)
+        output_text = check_frame_modbus(frame_recv, block_modbus)
+        print(output_text)
+
+
+    def frame_update(self, path_bin):
+        """ 程序升级
+
+        """
+        block_modbus = {
+            'addr_dev' : self.addr_modbus,
+            'type' : 'update',
+            'step' : 'start',
+            'index' : 0,
+            'file' : Path(path_bin).read_bytes(),
+            'header_offset': 128,
+        }
+        # 启动帧
+        frame_master = bytearray(make_frame_modbus(block_modbus))
+
+        # 等待擦除完成返回
+        try_times = 3000
+        self.tcp_socket.recv()
+        while try_times:
+            time.sleep(0.4)
+            self.tcp_socket.send(frame_master)
+            frame_slave = self.tcp_socket.recv()
+            if not frame_slave:
+                try_times -= 1
+                continue
+            
+            block_modbus['file_block_size'] = check_frame_modbus(frame_slave, block_modbus)
+            break
+
+        if block_modbus['file_block_size'] == 0:
+            raise("Error slave response.")
+
+        # 避免接收到延迟返回报文
+        time.sleep(0.4)
+
+        # 文件传输
+        block_modbus['step'] = 'trans'
+        data_remain = len(block_modbus['file']) - block_modbus['header_offset']
+        while data_remain > 0:
+            frame_master = bytearray(make_frame_modbus(block_modbus))
+
+            self.tcp_socket.recv()
+            self.tcp_socket.send(frame_master)
+            time.sleep(0.2)
+            frame_slave = None
+            while not frame_slave:
+                frame_slave = self.tcp_socket.recv()
+            check_frame_modbus(frame_slave, block_modbus)
+            
+            block_modbus['index'] += 1
+            data_remain -= block_modbus['file_block_size']
+
+        # 结束升级
+        block_modbus['step'] = 'end'
+        frame_master = bytearray(make_frame_modbus(block_modbus))
+
+        self.tcp_socket.recv()
+        self.tcp_socket.send(frame_master)
+        time.sleep(0.1)
+        frame_slave = None
+        while not frame_slave:
+            frame_slave = self.tcp_socket.recv()
+        check_frame_modbus(frame_slave[:18], block_modbus)
 
 
 if __name__ == "__main__":
+    """ 升级测试 """
+    path_file = Path("F:\\Work\\FPGA\\Test\\Vivado\\test_update\\test_update.vitis\\upgrade_system\\Debug\\sd_card\\BOOT.dat")
+
     dev_ep = EnergyRouter()
 
+    dev_ep.frame_read()
+
+    dev_ep.frame_update(path_file)
+
+
+