# -*- codeing = utf-8 -*- # Time : 2022/9/17 15:05 # @Auther : zhouchao # @File: QT_test.py # @Software:PyCharm import logging import socket import numpy as np import cv2 def rec_socket(recv_sock: socket.socket, cmd_type: str, ack: bool) -> bool: if ack: cmd = 'A' + cmd_type else: cmd = 'D' + cmd_type while True: try: temp = recv_sock.recv(1) except ConnectionError as e: logging.error(f'连接出错, 错误代码:\n{e}') return False except TimeoutError as e: logging.error(f'超时了，错误代码: \n{e}') return False except Exception as e: logging.error(f'遇见未知错误，错误代码: \n{e}') return False if temp == b'\xaa': break # 获取报文长度 temp = b'' while len(temp) < 4: try: temp += recv_sock.recv(1) except Exception as e: logging.error(f'接收报文长度失败, 错误代码: \n{e}') return False try: data_len = int.from_bytes(temp, byteorder='big') except Exception as e: logging.error(f'转换失败,错误代码 \n{e}, \n报文内容\n{temp}') return False # 读取报文内容 temp = b'' while len(temp) < data_len: try: temp += recv_sock.recv(data_len) except Exception as e: logging.error(f'接收报文内容失败, 错误代码: \n{e}，\n报文内容\n{temp}') return False data = temp if cmd.strip().upper() != data[:4].decode('ascii').strip().upper(): logging.error(f'客户端接收指令错误,\n指令内容\n{data}') return False else: if cmd == 'DIM': print(data) # 进行数据校验 temp = b'' while len(temp) < 3: try: temp += recv_sock.recv(1) except Exception as e: logging.error(f'接收报文校验失败, 错误代码: \n{e}') return False if temp == b'\xff\xff\xbb': return True else: logging.error(f"接收了一个完美的只错了校验位的报文，\n data: {data}") return False def main(): socket_receive = socket.socket(socket.AF_INET, socket.SOCK_STREAM) socket_receive.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) socket_receive.bind(('127.0.0.1', 21123)) socket_receive.listen(5) socket_send = socket.socket(socket.AF_INET, socket.SOCK_STREAM) socket_send.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) socket_send.bind(('127.0.0.1', 21122)) socket_send.listen(5) print('等待连接') socket_send_1, receive_addr_1 = socket_send.accept() print("连接成功：", receive_addr_1) # socket_send_2 = socket_send_1 socket_send_2, receive_addr_2 = socket_receive.accept() print("连接成功：", receive_addr_2) while True: cmd = input().strip().upper() if cmd == 'IM': img = cv2.imread(r"/Users/zhouchao/Library/CloudStorage/OneDrive-macrosolid/PycharmProjects/wood_color/data/data20220919/dark/rgb60.png") # img = cv2.imread(r"C:\Users\FEIJINTI\PycharmProjects\wood_color\data\data20220919\dark\rgb60.png") img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) img = np.asarray(img, dtype=np.float32) width = img.shape[0] height = img.shape[1] img_bytes = img.tobytes() length = len(img_bytes) + 8 length = length.to_bytes(4, byteorder='big') width = width.to_bytes(2, byteorder='big') height = height.to_bytes(2, byteorder='big') send_message = b'\xaa' + length + (' ' + cmd).upper().encode('ascii') + width + height + img_bytes + b'\xff\xff\xbb' socket_send_1.send(send_message) print('发送成功') result = socket_send_2.recv(1) print(result) # if rec_socket(socket_send_2, cmd_type=cmd, ack=True): # print('接收指令成功') # else: # print('接收指令失败') # if rec_socket(socket_send_2, cmd_type=cmd, ack=False): # print('指令执行完毕') # else: # print('指令执行失败') elif cmd == 'TR': model = "/Users/zhouchao/Library/CloudStorage/OneDrive-macrosolid/PycharmProjects/wood_color/data/data20220919" # model = "C:/Users/FEIJINTI/PycharmProjects/wood_color/data/data20220919" model = model.encode('ascii') length = len(model) + 4 length = length.to_bytes(4, byteorder='big') send_message = b'\xaa' + length + (' ' + cmd).upper().encode('ascii') + model + b'\xff\xff\xbb' socket_send_1.send(send_message) print('发送成功') result = socket_send_2.recv(1) print(result) # if rec_socket(socket_send_2, cmd_type=cmd, ack=True): # print('接收指令成功') # else: # print('接收指令失败') # if rec_socket(socket_send_2, cmd_type=cmd, ack=False): # print('指令执行完毕') # else: # print('指令执行失败') elif cmd == 'MD': model = "/Users/zhouchao/Library/CloudStorage/OneDrive-macrosolid/PycharmProjects/wood_color/models/model_2020-11-08_20-49.p" # model = "C:/Users/FEIJINTI/PycharmProjects/wood_color/models/model_2020-11-08_20-49.p" model = model.encode('ascii') length = len(model) + 4 length = length.to_bytes(4, byteorder='big') send_message = b'\xaa' + length + (' ' + cmd).upper().encode('ascii') + model + b'\xff\xff\xbb' socket_send_1.send(send_message) print('发送成功') result = socket_send_2.recv(1) print(result) # if rec_socket(socket_send_2, cmd_type=cmd, ack=True): # print('接收指令成功') # else: # print('接收指令失败') # if rec_socket(socket_send_2, cmd_type=cmd, ack=False): # print('指令执行完毕') # else: # print('指令执行失败') elif cmd == 'KM': model = "E:/wood_color/data/2023318" # model = "C:/Users/FEIJINTI/PycharmProjects/wood_color/models/model_2020-11-08_20-49.p" model = model.encode('ascii') length = len(model) + 4 length = length.to_bytes(4, byteorder='big') send_message = b'\xaa' + length + (' ' + cmd).upper().encode('ascii') + model + b'\xff\xff\xbb' socket_send_1.send(send_message) print('发送成功') result = socket_send_2.recv(5) print(result) new_leng = int.from_bytes(result[1:], byteorder='big') result = socket_send_2.recv(new_leng) print(result) # if rec_socket(socket_send_2, cmd_type=cmd, ack=True): # print('接收指令成功') # else: # print('接收指令失败') # if rec_socket(socket_send_2, cmd_type=cmd, ack=False): # print('指令执行完毕') # else: # print('指令执行失败') else: print('指令错误') if __name__ == '__main__': main()