Python 创建一个简单的在线聊天系统

LIY若依

已于 2024-07-26 13:00:01 修改

阅读量4.2k

点赞数 24

文章标签：服务器网络 linux

于 2024-07-26 12:12:35 首次发布

本文链接：https://blog.csdn.net/m0_74972192/article/details/140712270

版权

在这篇博客中，我们将创建一个简单的在线聊天系统，使用 Python 的 socket 库和多线程技术来实现服务器和客户端之间的通信。通过这个项目，你将学到如何使用 socket 创建网络连接，以及如何处理多个客户端的并发连接。

1. 准备工作

在开始之前，请确保你的环境中已经安装了 socket。如果尚未安装，请运行以下命令：

pip install socket

2. 服务器端代码

服务器端代码负责监听客户端的连接请求，并在接收到消息时将其广播给所有已连接的客户端。以下是服务器端的完整代码：

import socket
import threading

class ChatServer:
    def __init__(self, host, port):
        self.host = host
        self.port = port
        self.clients = []
        self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.server_socket.bind((self.host, self.port))
        self.server_socket.listen(5)  # Allow up to 5 client connections

    def broadcast(self, message, sender):
        for client in self.clients:
            if client != sender:
                try:
                    client.send(message.encode())
                except:
                    client.close()
                    if client in self.clients:
                        self.clients.remove(client)

    def handle_client(self, client_socket):
        try:
            while True:
                message = client_socket.recv(1024).decode()
                if message:
                    self.broadcast(message, client_socket)
                else:
                    break
        except Exception as e:
            print(f"Error handling client: {e}")
        finally:
            client_socket.close()
            if client_socket in self.clients:
                self.clients.remove(client_socket)

    def start(self):
        print(f"Server is listening on {self.host}:{self.port}")
        while True:
            client_socket, client_address = self.server_socket.accept()
            print(f"Accepted connection from {client_address}")
            self.clients.append(client_socket)
            client_handler = threading.Thread(target=self.handle_client, args=(client_socket,))
            client_handler.start()

if __name__ == "__main__":
    HOST = '127.0.0.1'  # localhost
    PORT = 12345
    server = ChatServer(HOST, PORT)
    server.start()

代码解析

导入模块：导入 socket 和 threading 模块。
初始化服务器：创建 ChatServer 类，初始化服务器地址和端口，创建服务器套接字并绑定地址和端口。
广播消息：定义 broadcast 方法，将消息发送给所有已连接的客户端。
处理客户端：定义 handle_client 方法，处理客户端的消息接收和转发。
启动服务器：定义 start 方法，监听客户端连接请求并启动新线程处理每个客户端。

3. 客户端代码

客户端代码负责连接到服务器，并发送和接收消息。以下是客户端的完整代码：

import socket
import threading
import sys

class ChatClient:
    def __init__(self, host, port):
        self.host = host
        self.port = port
        self.client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    def connect_to_server(self):
        try:
            self.client_socket.connect((self.host, self.port))
            print(f"Connected to server at {self.host}:{self.port}")
            receive_thread = threading.Thread(target=self.receive_messages)
            receive_thread.start()
            self.send_messages()
        except Exception as e:
            print(f"Error connecting to server: {e}")
            sys.exit()

    def send_messages(self):
        while True:
            message = input()
            self.client_socket.send(message.encode())

    def receive_messages(self):
        while True:
            try:
                message = self.client_socket.recv(1024).decode()
                if message:
                    print(message)
            except Exception as e:
                print(f"Error receiving message: {e}")
                break

if __name__ == "__main__":
    HOST = '127.0.0.1'  # localhost
    PORT = 12345
    client = ChatClient(HOST, PORT)
    client.connect_to_server()

代码解析

导入模块：导入 socket、threading 和 sys 模块。
初始化客户端：创建 ChatClient 类，初始化服务器地址和端口，创建客户端套接字。
连接服务器：定义 connect_to_server 方法，连接到服务器并启动接收消息的线程。
发送消息：定义 send_messages 方法，从用户输入获取消息并发送到服务器。
接收消息：定义 receive_messages 方法，接收服务器转发的消息并打印。

4. 运行结果

客户端 1

Connected to server at 127.0.0.1:12345
Hello, this is client 1.

客户端 2

Connected to server at 127.0.0.1:12345
Hello, this is client 2.

5.说明

服务器端：服务器端代码负责监听客户端的连接请求，并在接收到消息时将其广播给所有已连接的客户端。服务器使用 socket 库创建网络连接，并使用多线程技术处理多个客户端的并发连接。
客户端：客户端代码负责连接到服务器，并发送和接收消息。客户端使用 socket 库连接到服务器，并使用多线程技术同时处理消息的发送和接收。
多线程：为了处理多个客户端的并发连接，服务器端和客户端都使用了多线程技术。服务器端为每个连接的客户端创建一个新线程，客户端为接收消息创建一个新线程。
网络连接：使用 socket 库创建 TCP 连接，实现服务器和客户端之间的通信。服务器监听特定的端口，客户端连接到该端口进行通信。