pip install websockets

import asyncio
import websockets
from datetime import datetime

# 保存所有活躍的客戶端連接
connected_clients = set()

async def handle_client(websocket, path):
    """
    處理單個客戶端連接
    """
    # 將新客戶端添加到集合
    connected_clients.add(websocket)
    try:
        async for message in websocket:
            # 廣播消息給所有客戶端
            timestamp = datetime.now().strftime("%H:%M:%S")
            broadcast_message = f"[{timestamp}] {message}"
            await broadcast(broadcast_message)
    except websockets.exceptions.ConnectionClosed:
        print("客戶端斷開連接")
    finally:
        # 客戶端斷開后移除
        connected_clients.remove(websocket)

async def broadcast(message):
    """
    向所有連接的客戶端廣播消息
    """
    if connected_clients:
        await asyncio.gather(
            *[client.send(message) for client in connected_clients]
        )

async def start_server():
    """
    啟動 WebSocket 服務器
    """
    async with websockets.serve(handle_client, "localhost", 8765):
        print("服務器已啟動，監(jiān)聽端口 8765...")
        await asyncio.Future()  # 永久運行

if __name__ == "__main__":
    asyncio.run(start_server())

import asyncio
import websockets

async def client():
    """
    WebSocket 客戶端實現(xiàn)
    """
    async with websockets.connect("ws://localhost:8765") as websocket:
        # 創(chuàng)建兩個任務：一個接收消息，一個發(fā)送消息
        receive_task = asyncio.create_task(receive_messages(websocket))
        send_task = asyncio.create_task(send_messages(websocket))
        await asyncio.gather(receive_task, send_task)

async def receive_messages(websocket):
    """
    接收服務器消息
    """
    async for message in websocket:
        print(f"\n收到消息: {message}")

async def send_messages(websocket):
    """
    發(fā)送用戶輸入的消息
    """
    while True:
        message = await asyncio.get_event_loop().run_in_executor(None, input, "輸入消息 (輸入 'exit' 退出): ")
        if message.strip().lower() == 'exit':
            break
        await websocket.send(message)

if __name__ == "__main__":
    asyncio.run(client())

python server.py

python client.py