当前位置：首页 > news >正文

聊天室Python脚本——ChatGPT，好用

news 来源：原创 2025/6/17 11:04:11

下面提供两个 Python 脚本，一个作为服务器端（chat_server.py），一个作为客户端（chat_client.py）。你可以在一台电脑上运行服务器脚本，然后在不同电脑上运行客户端脚本（连接时指定服务器的 IP 地址），实现多台电脑之间互相发送消息聊天。

服务器端代码（chat_server.py）

该脚本实现了以下功能：

监听所有网络接口的指定端口（例如 12345）。
接受客户端连接，并为每个连接启动一个线程进行处理。
每当收到客户端发送的消息，就将该消息广播给其他所有已连接的客户端。

import socket
import threading

# 保存所有客户端连接的列表，使用锁保证线程安全
clients = []
lock = threading.Lock()

def broadcast(message, sender_socket):
    with lock:
        for client in clients:
            if client != sender_socket:
                try:
                    client.send(message)
                except Exception as e:
                    print("发送消息错误：", e)

def handle_client(client_socket, addr):
    print(f"新连接：{addr}")
    while True:
        try:
            message = client_socket.recv(1024)
            if not message:
                print(f"{addr} 断开连接")
                break
            print(f"收到来自 {addr} 的消息：{message.decode()}")
            broadcast(message, client_socket)
        except Exception as e:
            print(f"处理 {addr} 时发生错误：{e}")
            break
    with lock:
        if client_socket in clients:
            clients.remove(client_socket)
    client_socket.close()

def main():
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    host = "0.0.0.0"  # 绑定所有可用的网络接口
    port = 12345
    server_socket.bind((host, port))
    server_socket.listen(5)
    print(f"服务器启动，监听端口 {port}")

    while True:
        client_socket, addr = server_socket.accept()
        with lock:
            clients.append(client_socket)
        thread = threading.Thread(target=handle_client, args=(client_socket, addr))
        thread.start()

if __name__ == '__main__':
    main()

客户端代码（chat_client.py）

该脚本实现了以下功能：

连接到指定服务器（需要输入服务器的 IP 地址）和端口（与服务器一致）。
启动一个线程不断接收服务器转发的消息，并打印在屏幕上。
主线程读取用户输入，并将消息发送到服务器（输入 “exit” 即可退出）。

import socket
import threading
import sys

def receive_messages(client_socket):
    while True:
        try:
            message = client_socket.recv(1024)
            if not message:
                print("服务器关闭了连接。")
                break
            print("收到消息：", message.decode())
        except Exception as e:
            print("接收消息错误：", e)
            break

def main():
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_ip = input("请输入服务器 IP 地址：")
    port = 12345
    try:
        client_socket.connect((server_ip, port))
    except Exception as e:
        print("连接服务器失败：", e)
        sys.exit()

    print("成功连接到聊天服务器。你可以开始发送消息了（输入 'exit' 退出）。")
    
    # 启动接收消息的线程
    thread = threading.Thread(target=receive_messages, args=(client_socket,))
    thread.daemon = True
    thread.start()

    while True:
        message = input("")
        if message.lower() == "exit":
            break
        try:
            client_socket.send(message.encode())
        except Exception as e:
            print("发送消息错误：", e)
            break

    client_socket.close()

if __name__ == '__main__':
    main()

运行说明

服务器端
在一台电脑上运行服务器脚本：
```
python3 chat_server.py
```
服务器将监听本机所有网络接口的 12345 端口。
客户端
在其他电脑上运行客户端脚本：
```
python3 chat_client.py
```
按提示输入服务器的 IP 地址（服务器所在电脑的局域网 IP 或公网 IP），连接成功后即可相互聊天。

这种简单实现仅适用于基础演示，实际使用时可能需要增加错误处理、安全认证等机制。