443 lines
14 KiB
Python
443 lines
14 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
SUW Client - Python翻译版本
|
||
原文件: SUWClient.rb
|
||
用途: TCP客户端，与服务器通信
|
||
"""
|
||
|
||
import socket
|
||
import json
|
||
import struct
|
||
import threading
|
||
import time
|
||
from typing import List, Dict, Any, Optional
|
||
|
||
# 常量定义
|
||
TCP_SERVER_PORT = 7999
|
||
OP_CMD_REQ_GETCMDS = 0x01
|
||
OP_CMD_REQ_SETCMD = 0x03
|
||
OP_CMD_RES_GETCMDS = 0x02
|
||
OP_CMD_RES_SETCMD = 0x04
|
||
|
||
|
||
class SUWClient:
|
||
"""SUWood 客户端类"""
|
||
|
||
def __init__(self, host="127.0.0.1", port=TCP_SERVER_PORT):
|
||
self.host = host
|
||
self.port = port
|
||
self.sock = None
|
||
self.seqno = 0
|
||
self.connect()
|
||
|
||
def connect(self):
|
||
"""连接到服务器"""
|
||
try:
|
||
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
self.sock.connect((self.host, self.port))
|
||
print(f"✅ 连接到服务器 {self.host}:{self.port}")
|
||
except Exception as e:
|
||
print(f"❌ 连接失败: {e}")
|
||
self.sock = None
|
||
|
||
def reconnect(self):
|
||
"""重新连接"""
|
||
if self.sock:
|
||
try:
|
||
self.sock.close()
|
||
except:
|
||
pass
|
||
|
||
self.connect()
|
||
|
||
def send_msg(self, cmd: int, msg: str):
|
||
"""发送消息"""
|
||
if not self.sock:
|
||
print("❌ 未连接到服务器")
|
||
return False
|
||
|
||
try:
|
||
opcode = (cmd & 0xffff) | 0x01010000
|
||
self.seqno += 1
|
||
|
||
# 打包消息：[消息长度, 操作码, 序列号, 保留字段]
|
||
msg_bytes = msg.encode('utf-8')
|
||
header = struct.pack('iiii', len(msg_bytes), opcode, self.seqno, 0)
|
||
|
||
full_msg = header + msg_bytes
|
||
self.sock.send(full_msg)
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"❌ 发送消息失败: {e}")
|
||
return False
|
||
|
||
def recv_msg(self) -> Optional[str]:
|
||
"""接收消息 - 修复编码问题"""
|
||
if not self.sock:
|
||
print("❌ 未连接到服务器")
|
||
return None
|
||
|
||
try:
|
||
# 接收头部（16字节）
|
||
header = self.sock.recv(16)
|
||
if len(header) < 16:
|
||
return None
|
||
|
||
# 解包获取消息长度
|
||
msg_len = struct.unpack('iiii', header)[0]
|
||
|
||
# 接收消息内容
|
||
msg = b""
|
||
to_recv_len = msg_len
|
||
|
||
while to_recv_len > 0:
|
||
chunk = self.sock.recv(to_recv_len)
|
||
if not chunk:
|
||
break
|
||
msg += chunk
|
||
to_recv_len = msg_len - len(msg)
|
||
|
||
# 【修复】改进编码处理
|
||
if not msg:
|
||
return None
|
||
|
||
# 首先尝试UTF-8
|
||
try:
|
||
return msg.decode('utf-8')
|
||
except UnicodeDecodeError:
|
||
# 尝试其他编码
|
||
encodings = ['latin1', 'gbk', 'gb2312', 'cp1252', 'iso-8859-1']
|
||
for encoding in encodings:
|
||
try:
|
||
decoded = msg.decode(encoding)
|
||
if encoding != 'utf-8':
|
||
print(f"⚠️ 使用 {encoding} 编码解码成功")
|
||
return decoded
|
||
except UnicodeDecodeError:
|
||
continue
|
||
|
||
# 如果所有编码都失败，使用错误处理模式
|
||
print("⚠️ 所有编码都失败，使用错误处理模式")
|
||
return msg.decode('utf-8', errors='ignore')
|
||
|
||
except Exception as e:
|
||
print(f"❌ 接收消息失败: {e}")
|
||
return None
|
||
|
||
|
||
# 全局客户端实例
|
||
_client_instance = None
|
||
|
||
|
||
def get_client():
|
||
"""获取客户端实例"""
|
||
global _client_instance
|
||
if _client_instance is None:
|
||
_client_instance = SUWClient()
|
||
return _client_instance
|
||
|
||
|
||
def get_cmds() -> List[Dict[str, Any]]:
|
||
"""获取命令列表 - 修复错误处理"""
|
||
msg = json.dumps({
|
||
"cmd": "get_cmds",
|
||
"params": {"from": "su"}
|
||
})
|
||
|
||
client = get_client()
|
||
cmds = []
|
||
|
||
try:
|
||
if client.send_msg(OP_CMD_REQ_GETCMDS, msg):
|
||
res = client.recv_msg()
|
||
if res:
|
||
try:
|
||
# 尝试清理响应数据，移除可能的截断部分
|
||
cleaned_res = res.strip()
|
||
|
||
# 如果响应以 } 结尾，尝试找到完整的JSON
|
||
if cleaned_res.endswith('}'):
|
||
# 尝试从后往前找到匹配的 { 和 }
|
||
brace_count = 0
|
||
end_pos = len(cleaned_res) - 1
|
||
|
||
for i in range(end_pos, -1, -1):
|
||
if cleaned_res[i] == '}':
|
||
brace_count += 1
|
||
elif cleaned_res[i] == '{':
|
||
brace_count -= 1
|
||
if brace_count == 0:
|
||
# 找到完整的JSON
|
||
cleaned_res = cleaned_res[:i+1]
|
||
break
|
||
|
||
res_hash = json.loads(cleaned_res)
|
||
if res_hash.get('ret') == 1:
|
||
cmds = res_hash.get('data', {}).get('cmds', [])
|
||
# 只在有命令时输出日志
|
||
if cmds:
|
||
print(f"✅ 成功获取 {len(cmds)} 个命令")
|
||
else:
|
||
print(f"⚠️ 服务器返回错误: {res_hash.get('msg', '未知错误')}")
|
||
|
||
except json.JSONDecodeError as e:
|
||
# 只在调试模式下打印详细错误信息
|
||
if len(res) > 200: # 只对长响应打印详细信息
|
||
print(f"⚠️ JSON解析失败: {e}")
|
||
print(f"原始响应: {res[:200]}...")
|
||
|
||
# 尝试更激进的清理
|
||
try:
|
||
# 查找可能的JSON开始位置
|
||
start_pos = res.find('{')
|
||
if start_pos != -1:
|
||
# 尝试找到匹配的结束位置
|
||
brace_count = 0
|
||
for i in range(start_pos, len(res)):
|
||
if res[i] == '{':
|
||
brace_count += 1
|
||
elif res[i] == '}':
|
||
brace_count -= 1
|
||
if brace_count == 0:
|
||
cleaned_res = res[start_pos:i+1]
|
||
res_hash = json.loads(cleaned_res)
|
||
if res_hash.get('ret') == 1:
|
||
cmds = res_hash.get(
|
||
'data', {}).get('cmds', [])
|
||
# 只在有命令时输出日志
|
||
if cmds:
|
||
print(
|
||
f"✅ 修复后成功获取 {len(cmds)} 个命令")
|
||
# 【关键修复】确保修复后的命令被返回
|
||
return cmds
|
||
break
|
||
except:
|
||
print("❌ 无法修复JSON格式")
|
||
|
||
return []
|
||
|
||
except Exception as e:
|
||
print("========= get_cmds err is: =========")
|
||
print(e)
|
||
print("========= get_cmds res is: =========")
|
||
if 'res' in locals():
|
||
print(f"响应内容: {res[:200]}...")
|
||
else:
|
||
print("No response")
|
||
|
||
# 尝试重新连接
|
||
try:
|
||
client.reconnect()
|
||
except:
|
||
pass
|
||
|
||
return cmds
|
||
|
||
|
||
def set_cmd(cmd: str, params: Dict[str, Any]):
|
||
"""设置命令"""
|
||
cmds = {
|
||
"cmd": "set_cmd",
|
||
"params": params.copy()
|
||
}
|
||
|
||
cmds["params"]["from"] = "su"
|
||
cmds["params"]["cmd"] = cmd
|
||
|
||
msg = json.dumps(cmds)
|
||
client = get_client()
|
||
|
||
try:
|
||
if client.send_msg(OP_CMD_REQ_SETCMD, msg):
|
||
client.recv_msg() # 接收响应但不处理
|
||
|
||
except Exception as e:
|
||
print(f"❌ set_cmd 错误: {e}")
|
||
client.reconnect()
|
||
|
||
|
||
class CommandProcessor:
|
||
"""命令处理器"""
|
||
|
||
def __init__(self):
|
||
self.cmds_queue = []
|
||
self.pause = 0
|
||
self.running = False
|
||
self.timer_thread = None
|
||
|
||
def start(self):
|
||
"""启动命令处理器"""
|
||
if self.running:
|
||
return
|
||
|
||
self.running = True
|
||
self.timer_thread = threading.Thread(
|
||
target=self._timer_loop, daemon=True)
|
||
self.timer_thread.start()
|
||
print("✅ 命令处理器已启动")
|
||
|
||
def stop(self):
|
||
"""停止命令处理器"""
|
||
self.running = False
|
||
if self.timer_thread:
|
||
self.timer_thread.join(timeout=2)
|
||
print("⛔ 命令处理器已停止")
|
||
|
||
def _timer_loop(self):
|
||
"""定时器循环"""
|
||
while self.running:
|
||
try:
|
||
if self.pause > 0:
|
||
self.pause -= 1
|
||
else:
|
||
self._process_commands()
|
||
|
||
time.sleep(1) # 1秒间隔
|
||
|
||
except Exception as e:
|
||
print(f"❌ 命令处理循环错误: {e}")
|
||
time.sleep(1)
|
||
|
||
def _process_commands(self):
|
||
"""处理命令"""
|
||
try:
|
||
# 获取新命令
|
||
swcmds0 = get_cmds()
|
||
swcmds = self.cmds_queue + swcmds0
|
||
self.cmds_queue.clear()
|
||
|
||
# 处理每个命令
|
||
for swcmd in swcmds:
|
||
self._execute_command(swcmd)
|
||
|
||
except Exception as e:
|
||
print(f"❌ 处理命令时出错: {e}")
|
||
|
||
def _execute_command(self, swcmd: Dict[str, Any]):
|
||
"""执行单个命令"""
|
||
try:
|
||
data = swcmd.get("data")
|
||
|
||
if isinstance(data, str):
|
||
# 直接执行字符串命令（注意安全性）
|
||
print(f"执行字符串命令: {data}")
|
||
# 在实际应用中，这里应该更安全地执行命令
|
||
|
||
elif isinstance(data, dict) and "cmd" in data:
|
||
cmd = data.get("cmd")
|
||
print(f"执行命令: {cmd}, 数据: {data}")
|
||
|
||
if self.pause > 0:
|
||
self.cmds_queue.append(swcmd)
|
||
elif cmd.startswith("pause_"):
|
||
self.pause = data.get("value", 1)
|
||
else:
|
||
pre_pause_time = data.get("pre_pause", 0)
|
||
if pre_pause_time > 0:
|
||
data_copy = data.copy()
|
||
del data_copy["pre_pause"]
|
||
swcmd_copy = swcmd.copy()
|
||
swcmd_copy["data"] = data_copy
|
||
self.pause = pre_pause_time
|
||
self.cmds_queue.append(swcmd_copy)
|
||
else:
|
||
# 执行命令
|
||
self._call_suwood_method(cmd, data)
|
||
|
||
after_pause_time = data.get("after_pause", 0)
|
||
if after_pause_time > 0:
|
||
self.pause = after_pause_time
|
||
|
||
except Exception as e:
|
||
print(f"❌ 执行命令时出错: {e}")
|
||
|
||
def _call_suwood_method(self, cmd: str, data: Dict[str, Any]):
|
||
"""调用SUWood方法"""
|
||
try:
|
||
# 这里需要导入suw_core并调用相应方法
|
||
from .suw_core import get_selection_manager, get_machining_manager, get_deletion_manager
|
||
|
||
# 根据命令类型调用相应的管理器
|
||
if cmd.startswith("sel_"):
|
||
# 选择相关命令
|
||
selection_manager = get_selection_manager()
|
||
if hasattr(selection_manager, cmd):
|
||
method = getattr(selection_manager, cmd)
|
||
method(data)
|
||
else:
|
||
print(f"⚠️ 选择管理器方法不存在: {cmd}")
|
||
elif cmd.startswith("mach_"):
|
||
# 加工相关命令
|
||
machining_manager = get_machining_manager()
|
||
if hasattr(machining_manager, cmd):
|
||
method = getattr(machining_manager, cmd)
|
||
method(data)
|
||
else:
|
||
print(f"⚠️ 加工管理器方法不存在: {cmd}")
|
||
elif cmd.startswith("del_"):
|
||
# 删除相关命令
|
||
deletion_manager = get_deletion_manager()
|
||
if hasattr(deletion_manager, cmd):
|
||
method = getattr(deletion_manager, cmd)
|
||
method(data)
|
||
else:
|
||
print(f"⚠️ 删除管理器方法不存在: {cmd}")
|
||
else:
|
||
print(f"⚠️ 未知命令类型: {cmd}")
|
||
|
||
except ImportError:
|
||
print("⚠️ suw_core 模块未找到")
|
||
except Exception as e:
|
||
print(f"❌ 调用SUWood方法时出错: {e}")
|
||
|
||
|
||
# 全局命令处理器实例
|
||
_processor_instance = None
|
||
|
||
|
||
def get_processor():
|
||
"""获取命令处理器实例"""
|
||
global _processor_instance
|
||
if _processor_instance is None:
|
||
_processor_instance = CommandProcessor()
|
||
return _processor_instance
|
||
|
||
|
||
def start_command_processor():
|
||
"""启动命令处理器"""
|
||
processor = get_processor()
|
||
processor.start()
|
||
|
||
|
||
def stop_command_processor():
|
||
"""停止命令处理器"""
|
||
processor = get_processor()
|
||
processor.stop()
|
||
|
||
|
||
# 自动启动命令处理器（可选）
|
||
if __name__ == "__main__":
|
||
print("🚀 SUW客户端测试")
|
||
|
||
# 测试连接
|
||
client = get_client()
|
||
if client.sock:
|
||
print("连接成功，测试获取命令...")
|
||
cmds = get_cmds()
|
||
print(f"获取到 {len(cmds)} 个命令")
|
||
|
||
# 启动命令处理器
|
||
start_command_processor()
|
||
|
||
try:
|
||
# 保持运行
|
||
while True:
|
||
time.sleep(10)
|
||
except KeyboardInterrupt:
|
||
print("\n停止客户端...")
|
||
stop_command_processor()
|
||
else:
|
||
print("连接失败")
|