117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
import time
|
|
import socket
|
|
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._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_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)
|
|
|
|
|
|
|
|
|