From 15a6f3c8711c6e8238e8e1542a1b0b69b14d41b8 Mon Sep 17 00:00:00 2001 From: XiaoZhao45 <130202157+xiaozhao45@users.noreply.github.com> Date: Sun, 10 Mar 2024 17:16:22 +0800 Subject: [PATCH] Client programs There are no some features of the server program --- main-user.py | 247 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 247 insertions(+) create mode 100644 main-user.py diff --git a/main-user.py b/main-user.py new file mode 100644 index 0000000..b1b3c2d --- /dev/null +++ b/main-user.py @@ -0,0 +1,247 @@ +import sys +import socket +import os + +# PyQt5相关库导入 +from PyQt5.QtWidgets import QApplication, QMainWindow, QWidget, QLabel, QLineEdit, QTextEdit, QPushButton, QHBoxLayout, QVBoxLayout, QPlainTextEdit, QMenuBar, QMenu, QAction, QMessageBox +from PyQt5.QtGui import QTextCursor, QIcon, QColor, QKeySequence +from PyQt5.QtNetwork import QTcpSocket, QHostAddress, QAbstractSocket, QTcpServer +from PyQt5.QtCore import pyqtSignal, pyqtSlot + +local_ip = socket.gethostbyname(socket.gethostname()) + +if local_ip == "127.0.0.1": + noInternet_tip = QMessageBox.warning( "警告", "当前网络连接不可用，请检查网络连接！") + + +class ChatClient(QWidget): + def __init__(self): + super().__init__() + self.socket = None + self.username = None + + # 创建界面元素 + self.ip_label = QLabel("IP:") + self.ip_edit = QLineEdit() + self.port_label = QLabel("Port:") + self.port_edit = QLineEdit("8888") + self.connect_btn = QPushButton("连接") + self.username_label = QLabel("用户名:") + self.username_edit = QLineEdit() + self.confirm_username_btn = QPushButton("确定") + self.messages_display = QPlainTextEdit() + self.input_edit = QTextEdit() + self.send_btn = QPushButton("发送") + self.send_btn.setEnabled(False) + + # 绑定事件处理函数 + self.connect_btn.clicked.connect(self.connect_to_server) + self.input_edit.textChanged.connect(self.check_input) + self.send_btn.clicked.connect(self.send_message) + self.confirm_username_btn.clicked.connect(self.confirm_username) + + # 布局 + ip_layout = QHBoxLayout() + ip_layout.addWidget(self.ip_label) + ip_layout.addWidget(self.ip_edit) + ip_layout.addWidget(self.port_label) + ip_layout.addWidget(self.port_edit) + ip_layout.addWidget(self.connect_btn) + + username_layout = QHBoxLayout() + username_layout.addWidget(self.username_label) + username_layout.addWidget(self.username_edit) + username_layout.addWidget(self.confirm_username_btn) + + input_layout = QHBoxLayout() + input_layout.addWidget(self.input_edit) + input_layout.addWidget(self.send_btn) + + main_layout = QVBoxLayout() + main_layout.addLayout(ip_layout) + main_layout.addLayout(username_layout) + main_layout.addWidget(self.messages_display) + main_layout.addLayout(input_layout) + + self.setLayout(main_layout) + + def disconnect_from_server(self): + if self.socket is not None: + self.socket.disconnectFromHost() + self.handle_disconnected() # 直接调用断开连接的处理逻辑 + + def connect_to_server(self): + try: + ip = QHostAddress(self.ip_edit.text()) + port = int(self.port_edit.text()) + self.socket = QTcpSocket(self) + self.socket.readyRead.connect(self.receive_message) + self.socket.disconnected.connect(self.handle_disconnected) + self.socket.error.connect(self.handle_socket_error) + self.socket.connectToHost(ip, port) + self.send_btn.setEnabled(True) + + if self.socket.waitForConnected(1000): + self.update_messages("已连接到服务器\n欲要发送消息请确定用户名\n\n") + self.username_edit.setEnabled(True) + self.confirm_username_btn.setEnabled(True) + self.connect_btn.setEnabled(False) + self.connect_btn.setText("断开连接") # 修改连接按钮文本为“断开连接” + self.connect_btn.clicked.disconnect() # 断开原有点击事件连接 + self.connect_btn.clicked.connect(self.disconnect_from_server) # 连接新的点击事件处理函数 + self.connect_btn.setEnabled(True) # 在这里启用“断开连接”按钮 + + else: + self.handle_socket_error(self.socket.error()) + + + except ValueError as e: + self.update_messages(f"输入错误: {str(e)}\n\n") + + @pyqtSlot(QAbstractSocket.SocketError) + def handle_socket_error(self, error): + self.update_messages(f"Socket错误: {self.socket.errorString()}\n\n") + self.socket.close() + self.socket = None + + def receive_message(self): + try: + message = self.socket.readAll().data().decode('utf-8') + self.update_messages(message) + except UnicodeDecodeError: + self.update_messages("接收消息时出现编码错误\n\n") + + def send_message(self): + try: + message = self.input_edit.toPlainText() + if not message: + return + formatted_message = f"{self.username}：{message}\n\n" + self.socket.write(formatted_message.encode('utf-8')) + + self.update_messages(formatted_message) + self.input_edit.clear() + self.input_edit.setFocus() # 将焦点重新设置到编辑框 + self.check_input() # 检查输入框状态 + except Exception as e: + print(f"发送消息时发生错误: {str(e)}") + + def handle_disconnected(self): + self.update_messages("已与服务器断开连接\n\n") + self.username_edit.setEnabled(False) + self.confirm_username_btn.setEnabled(True) + self.connect_btn.setText("连接") # 更改按钮文本为“连接” + self.connect_btn.clicked.disconnect() # 断开当前点击事件处理函数 + self.connect_btn.clicked.connect(self.connect_to_server) # 连接原有的点击事件处理函数 + self.socket = None + + msg_box = QMessageBox(QMessageBox.Warning, "双重提示", "您已断开连接！") + msg_box.exec_() + + def update_messages(self, message): + cursor = self.messages_display.textCursor() + cursor.movePosition(QTextCursor.End) + cursor.insertText(message) + + # 设置文本颜色 + username_start = message.find(":") + 1 + username_end = message.find("：") + if username_start > 0 and username_end > username_start: + cursor.setPosition(cursor.position() - len(message) + username_start, QTextCursor.MoveAnchor) + cursor.setPosition(cursor.position() + username_end - username_start, QTextCursor.KeepAnchor) + color = QColor("blue") + format_ = cursor.charFormat() + format_.setForeground(color) + cursor.setCharFormat(format_) + + # 滚动到底部 + self.messages_display.verticalScrollBar().setValue(self.messages_display.verticalScrollBar().maximum()) + + def confirm_username(self): + username = self.username_edit.text() + "[Users]" + if username: + self.username = username # 在这里设置username属性 + self.username_edit.setReadOnly(True) + self.confirm_username_btn.setEnabled(False) + self.check_input() # 调用check_input时，username已经初始化 + self.input_edit.setReadOnly(False) + + def check_input(self): + if self.input_edit.toPlainText() and self.username: # 现在这个条件判断就不会出错了 + self.send_btn.setEnabled(True) + else: + self.send_btn.setEnabled(False) + +class MainWindow(QMainWindow): + def __init__(self): + super().__init__() + self.setWindowTitle("聊天客户端") + self.resize(600, 400) + self.setWindowIcon(QIcon('ico02.ico')) + + self.chat_client = ChatClient() + self.setCentralWidget(self.chat_client) + # 创建顶部菜单栏 + self.menu_bar = QMenuBar(self) + self.setMenuBar(self.menu_bar) + + # 创建“文件”菜单及其动作 + file_menu = QMenu('文件', self) + self.menu_bar.addMenu(file_menu) + exit_action = QAction('退出', self, shortcut=QKeySequence.Quit) + exit_action.triggered.connect(self.close_application) + save_chat_action = QAction('保存聊天记录', self) + save_chat_action.triggered.connect(self.save_chat_to_file) + file_menu.addAction(exit_action) + file_menu.addAction(save_chat_action) + + # 创建“编辑”菜单及其动作 + edit_menu = QMenu('编辑', self) + self.menu_bar.addMenu(edit_menu) + cut_action = QAction('剪切', self, shortcut=QKeySequence.Cut) + copy_action = QAction('复制', self, shortcut=QKeySequence.Copy) + paste_action = QAction('粘贴', self, shortcut=QKeySequence.Paste) + cut_action.setEnabled(False) # 确保在没有选中内容时，剪切不可用 + copy_action.setEnabled(False) # 同理，复制也不可用 + self.chat_client.input_edit.copyAvailable.connect(lambda c: cut_action.setEnabled(c)) + self.chat_client.input_edit.copyAvailable.connect(lambda c: copy_action.setEnabled(c)) + cut_action.triggered.connect(lambda: self.chat_client.input_edit.cut()) + copy_action.triggered.connect(lambda: self.chat_client.input_edit.copy()) + paste_action.triggered.connect(lambda: self.chat_client.input_edit.paste()) + edit_menu.addAction(cut_action) + edit_menu.addAction(copy_action) + edit_menu.addAction(paste_action) + + # 创建“关于”菜单及其动作 + about_menu = QMenu('关于', self) + self.menu_bar.addMenu(about_menu) + about_action = QAction('软件信息', self) + about_action.triggered.connect(self.show_about_dialog) + about_menu.addAction(about_action) + + def close_application(self): + self.close() + + def update_messages(self, message: str): + self.chat_room.update_clients(message) + + def save_chat_to_file(self): + current_dir = os.getcwd() + filename = "chat_log.txt" + full_path = os.path.join(current_dir, filename) + with open(full_path, 'w', encoding='utf-8') as f: + chat_history = self.chat_client.messages_display.toPlainText() + f.write(chat_history) + # 更新消息到聊天客户端的显示区域而不是chat_room + self.chat_client.update_messages(f"聊天记录已保存到当前目录：{filename}\n") + + def show_about_dialog(self): + message = f"软件名称：聊天室客户端\n软件作者：XiaoZhao\nGithub开源地址：http://github.com/xiaozhao45" + QMessageBox.about(self, "关于本软件", message) + + +if __name__ == '__main__': + app = QApplication(sys.argv) + main_window = MainWindow() + main_window.show() + sys.exit(app.exec_())