208 lines
7.7 KiB
Python
208 lines
7.7 KiB
Python
import sys
|
||
import json
|
||
import socket
|
||
import threading
|
||
import logging
|
||
from datetime import datetime
|
||
from queue import Queue
|
||
import tkinter as tk
|
||
from tkinter import ttk, scrolledtext
|
||
|
||
# 配置日志记录
|
||
logging.basicConfig(
|
||
level=logging.INFO,
|
||
format='%(asctime)s - %(levelname)s - %(message)s',
|
||
handlers=[
|
||
logging.FileHandler("client.log", encoding='utf-8'),
|
||
logging.StreamHandler()
|
||
]
|
||
)
|
||
|
||
class TcpClientGUI:
|
||
def __init__(self, master):
|
||
self.master = master
|
||
self.client_socket = None
|
||
self.receive_thread = None
|
||
self.running = False
|
||
self.receive_buffer = bytearray() # 接收缓冲区
|
||
self.is_receiving_json = False # JSON接收状态标志
|
||
|
||
# 创建界面组件
|
||
self.create_widgets()
|
||
|
||
def create_widgets(self):
|
||
# 连接参数区
|
||
param_frame = ttk.Frame(self.master, padding=10)
|
||
param_frame.grid(row=0, column=0, sticky="ew")
|
||
|
||
ttk.Label(param_frame, text="服务器IP:").grid(row=0, column=0, sticky="w")
|
||
self.txt_ip = ttk.Entry(param_frame, width=20)
|
||
self.txt_ip.grid(row=0, column=1, padx=5)
|
||
self.txt_ip.insert(0, "127.0.0.1")
|
||
|
||
ttk.Label(param_frame, text="端口:").grid(row=0, column=2, sticky="w")
|
||
self.txt_port = ttk.Entry(param_frame, width=10)
|
||
self.txt_port.grid(row=0, column=3, padx=5)
|
||
self.txt_port.insert(0, "8080")
|
||
|
||
# 按钮区
|
||
btn_frame = ttk.Frame(self.master, padding=10)
|
||
btn_frame.grid(row=1, column=0, sticky="ew")
|
||
|
||
self.btn_connect = ttk.Button(btn_frame, text="连接", command=self.connect)
|
||
self.btn_connect.pack(side="left", padx=5)
|
||
|
||
self.btn_disconnect = ttk.Button(btn_frame, text="断开", command=self.disconnect, state=tk.DISABLED)
|
||
self.btn_disconnect.pack(side="left", padx=5)
|
||
|
||
# 消息发送区
|
||
send_frame = ttk.Frame(self.master, padding=10)
|
||
send_frame.grid(row=2, column=0, sticky="ew")
|
||
|
||
ttk.Label(send_frame, text="发送消息:").pack(anchor="w")
|
||
self.txt_send = ttk.Entry(send_frame, width=50)
|
||
self.txt_send.pack(fill="x", pady=5)
|
||
self.txt_send.bind("<Return>", self.send_message)
|
||
|
||
# 接收显示区
|
||
receive_frame = ttk.Frame(self.master, padding=10)
|
||
receive_frame.grid(row=3, column=0, sticky="nsew")
|
||
|
||
self.txt_received = scrolledtext.ScrolledText(receive_frame, wrap=tk.WORD, width=60, height=20)
|
||
self.txt_received.pack(fill="both", expand=True)
|
||
|
||
# 状态栏
|
||
self.status_var = tk.StringVar(value="未连接")
|
||
status_bar = ttk.Label(self.master, textvariable=self.status_var, relief=tk.SUNKEN)
|
||
status_bar.grid(row=4, column=0, sticky="ew")
|
||
|
||
# 配置网格布局权重
|
||
self.master.rowconfigure(3, weight=1)
|
||
self.master.columnconfigure(0, weight=1)
|
||
|
||
def connect(self):
|
||
"""建立TCP连接"""
|
||
try:
|
||
self.client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
self.client_socket.connect(
|
||
(self.txt_ip.get(), int(self.txt_port.get()))
|
||
)
|
||
self.running = True
|
||
self.receive_thread = threading.Thread(target=self.receive_messages, daemon=True)
|
||
self.receive_thread.start()
|
||
self.update_status("已连接", "green")
|
||
self.btn_connect.config(state=tk.DISABLED)
|
||
self.btn_disconnect.config(state=tk.NORMAL)
|
||
except Exception as e:
|
||
self.log_error(f"连接失败: {str(e)}")
|
||
|
||
def disconnect(self):
|
||
"""断开连接"""
|
||
self.running = False
|
||
if self.client_socket:
|
||
self.client_socket.close()
|
||
self.update_status("未连接", "red")
|
||
self.btn_connect.config(state=tk.NORMAL)
|
||
self.btn_disconnect.config(state=tk.DISABLED)
|
||
|
||
def send_message(self, event=None):
|
||
"""发送消息"""
|
||
try:
|
||
message = self.txt_send.get()
|
||
if message:
|
||
# 添加换行符并编码为UTF-8
|
||
data = message.encode('utf-8') + b'\n'
|
||
self.client_socket.sendall(data)
|
||
self.append_text(f"[{datetime.now().strftime('%H:%M:%S')}] 发送: {message}\n")
|
||
self.txt_send.delete(0, tk.END)
|
||
except Exception as e:
|
||
self.log_error(f"发送失败: {str(e)}")
|
||
|
||
def receive_messages(self):
|
||
"""接收消息主循环"""
|
||
try:
|
||
while self.running:
|
||
data = self.client_socket.recv(1024)
|
||
if not data:
|
||
break
|
||
|
||
# 将接收到的数据添加到缓冲区
|
||
self.receive_buffer.extend(data)
|
||
|
||
# 处理缓冲区数据
|
||
while len(self.receive_buffer) > 0:
|
||
if self.is_receiving_json:
|
||
self.process_json_data()
|
||
else:
|
||
self.process_normal_data()
|
||
|
||
except ConnectionAbortedError:
|
||
pass # 正常断开连接
|
||
except Exception as e:
|
||
self.log_error(f"接收错误: {str(e)}")
|
||
finally:
|
||
self.disconnect()
|
||
|
||
def process_normal_data(self):
|
||
"""处理普通数据"""
|
||
# 查找OK标识（2字节）
|
||
if len(self.receive_buffer) >= 2:
|
||
if self.receive_buffer[:2] == b'OK':
|
||
self.is_receiving_json = True
|
||
del self.receive_buffer[:2] # 移除标识
|
||
return
|
||
else:
|
||
# 处理普通消息（查找换行符）
|
||
newline_pos = self.receive_buffer.find(b'\n')
|
||
if newline_pos != -1:
|
||
message = self.receive_buffer[:newline_pos].decode('utf-8', errors='replace')
|
||
self.append_text(f"[{datetime.now().strftime('%H:%M:%S')}] 接收: {message}\n")
|
||
del self.receive_buffer[:newline_pos+1]
|
||
|
||
def process_json_data(self):
|
||
"""处理JSON数据"""
|
||
try:
|
||
# 尝试解析JSON数组
|
||
data_str = self.receive_buffer.decode('utf-8', errors='replace')
|
||
data = json.loads(data_str)
|
||
|
||
if isinstance(data, list):
|
||
self.save_json_file(data)
|
||
self.append_text(f"[{datetime.now().strftime('%H:%M:%S')}] 收到JSON数组，已保存文件\n")
|
||
self.receive_buffer.clear()
|
||
self.is_receiving_json = False
|
||
except json.JSONDecodeError:
|
||
# 数据不完整，等待更多数据
|
||
pass
|
||
|
||
def save_json_file(self, data):
|
||
"""保存JSON文件"""
|
||
try:
|
||
filename = f"log_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
|
||
with open(filename, 'w', encoding='utf-8') as f:
|
||
json.dump(data, f, ensure_ascii=False, indent=2)
|
||
logging.info(f"文件已保存: {filename}")
|
||
except Exception as e:
|
||
self.log_error(f"保存文件失败: {str(e)}")
|
||
|
||
def append_text(self, text):
|
||
"""线程安全更新接收文本框"""
|
||
self.txt_received.insert(tk.END, text)
|
||
self.txt_received.see(tk.END)
|
||
|
||
def update_status(self, message, color="black"):
|
||
"""更新状态栏"""
|
||
self.status_var.set(message)
|
||
# 正确写法：直接调用主窗口的update方法
|
||
self.master.update()
|
||
|
||
def log_error(self, message):
|
||
"""记录错误日志"""
|
||
logging.error(message)
|
||
self.txt_received.insert(tk.END, f"错误: {message}\n")
|
||
|
||
if __name__ == "__main__":
|
||
root = tk.Tk()
|
||
root.title("TCP客户端")
|
||
app = TcpClientGUI(root)
|
||
root.mainloop() |