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

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


服务器端代码(chat_server.py)

该脚本实现了以下功能:

  • 监听所有网络接口的指定端口(例如 12345)。
  • 接受客户端连接,并为每个连接启动一个线程进行处理。
  • 每当收到客户端发送的消息,就将该消息广播给其他所有已连接的客户端。
python 复制代码
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" 即可退出)。
python 复制代码
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. 服务器端

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

    bash 复制代码
    python3 chat_server.py

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

  2. 客户端

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

    bash 复制代码
    python3 chat_client.py

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

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

相关推荐
用户83562907805112 小时前
Python 实现 PDF 文件加密与解密方法
后端·python
用户83562907805112 小时前
使用 Python 冻结与拆分 Excel 窗格教程
后端·python
你好潘先生20 小时前
别再记命令了,用 yeero do 说句人话就能跑脚本,而且不烧 token
服务器·python·命令行
Agent_大师20 小时前
WebSocket 行情重连成功,K线缺口不会自动消失
python
荣码20 小时前
LLM结构化输出:让AI返回JSON而不是废话,我踩了4个坑
java·python
copyer_xyf21 小时前
FastAPI 如何连接 MySQL
后端·python
apocelipes1 天前
常用编程语言和库的正则表达式性能对比
c语言·c++·python·性能优化·golang·开发工具和环境
用户8356290780511 天前
使用 Python 在 PDF 中创建与管理书签
后端·python
Augustzero2 天前
Codex Desktop 新建会话无法发送消息:一次由旧版 CLI 路径引发的故障排查
chatgpt·agent
MeixianAgent2 天前
Python 回测数据入口怎么验?历史 K 线入库前先做 5 个检查
后端·python