聊天室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()
运行说明
-
服务器端
在一台电脑上运行服务器脚本:python3 chat_server.py
服务器将监听本机所有网络接口的 12345 端口。
-
客户端
在其他电脑上运行客户端脚本:python3 chat_client.py
按提示输入服务器的 IP 地址(服务器所在电脑的局域网 IP 或公网 IP),连接成功后即可相互聊天。
这种简单实现仅适用于基础演示,实际使用时可能需要增加错误处理、安全认证等机制。