目录

聊天室Python脚本ChatGPT,好用

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

下面提供两个 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()

运行说明

  1. 服务器端

    在一台电脑上运行服务器脚本:

    python3 chat_server.py

    服务器将监听本机所有网络接口的 12345 端口。

  2. 客户端

    在其他电脑上运行客户端脚本:

    python3 chat_client.py

    按提示输入服务器的 IP 地址(服务器所在电脑的局域网 IP 或公网 IP),连接成功后即可相互聊天。

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