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

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

相关推荐
十年Java程序媛几秒前
Java 多线程基础与锁机制详解|进程线程、锁升级、synchronized 与 Lock 对比
java·开发语言·spring boot
弯_弯1 分钟前
FDE 的工程能力画像:数据接入、PoC 与可复用资产的落地口径
java·服务器·开发语言
余槐i7 分钟前
transformers 升级后加载旧权重报错:三处不兼容如何逐一验证
python·大模型·transformers·模型量化·模型推理
雾沉川7 分钟前
Visual C++运行库合集 v105.0 安装教程
开发语言·c++·dll
CCblandy43 分钟前
向量检索 ANN 参数实测:换一种数据分布,recall 掉了 51 个百分点
人工智能·python·算法·自然语言处理·性能优化
niucloud-admin1 小时前
NIUCLOUD LITE AI开发文档—模块开发:支付管理
开发语言
云和数据.ChenGuang1 小时前
print(f“a+b=“,result) print(f“a+b={result}“)区别
java·前端·人工智能·python·深度学习
2601_962885721 小时前
如何用 Python 算最大回撤和回撤恢复期?
java·数据库·python
dear_bi_MyOnly2 小时前
栈和队列:从原理到应用全解析
c语言·开发语言·c++·学习·分类
Want5952 小时前
Python大雪纷飞①
开发语言·python·数学建模