聊天室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),连接成功后即可相互聊天。

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

相关推荐
IVEN_12 小时前
只会Python皮毛?深入理解这几点,轻松进阶全栈开发
python·全栈
Ray Liang14 小时前
用六边形架构与整洁架构对比是伪命题?
java·python·c#·架构设计
AI攻城狮14 小时前
如何给 AI Agent 做"断舍离":OpenClaw Session 自动清理实践
python
千寻girling14 小时前
一份不可多得的 《 Python 》语言教程
人工智能·后端·python
AI攻城狮17 小时前
用 Playwright 实现博客一键发布到稀土掘金
python·自动化运维
曲幽17 小时前
FastAPI分布式系统实战:拆解分布式系统中常见问题及解决方案
redis·python·fastapi·web·httpx·lock·asyncio
孟健1 天前
Karpathy 用 200 行纯 Python 从零实现 GPT:代码逐行解析
python
码路飞1 天前
写了个 AI 聊天页面,被 5 种流式格式折腾了一整天 😭
javascript·python
曲幽2 天前
FastAPI压力测试实战:Locust模拟真实用户并发及优化建议
python·fastapi·web·locust·asyncio·test·uvicorn·workers