再给我写一个 用python写一个聊天室 能实现一对多 入场填写用户名 入场通知 私聊 查询聊天室有哪些人 需要服务端代码和客户端代码
时间: 2024-03-03 09:49:36 浏览: 92
好的,以下是一个简单的 Python 聊天室实现,包含服务端和客户端代码。
服务端代码:
```python
import socket
import threading
# 服务器地址和端口
HOST = '127.0.0.1'
PORT = 8888
# 存储连接的客户端和对应的用户名
clients = {}
# 处理客户端发送的消息
def handle_client(conn, addr):
# 获取客户端的用户名
conn.send('请输入您的用户名:'.encode())
username = conn.recv(1024).decode()
clients[conn] = username
# 入场通知
broadcast(f'{username} 进入聊天室')
while True:
try:
# 接收客户端发送的消息
msg = conn.recv(1024).decode()
if msg:
# 私聊
if msg.startswith('@'):
name, msg = msg.split(' ', 1)
send_private_msg(name[1:], f'{username}: {msg}')
else:
# 广播消息
broadcast(f'{username}: {msg}')
else:
# 客户端断开连接
del clients[conn]
broadcast(f'{username} 离开聊天室')
conn.close()
break
except:
# 发生异常,客户端断开连接
del clients[conn]
broadcast(f'{username} 离开聊天室')
conn.close()
break
# 广播消息
def broadcast(msg):
for client_conn in clients:
client_conn.send(msg.encode())
# 发送私聊消息
def send_private_msg(name, msg):
for client_conn, username in clients.items():
if username == name:
client_conn.send(f'[私聊]{msg}'.encode())
break
# 启动服务端
def start_server():
# 创建套接字
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.bind((HOST, PORT))
server_socket.listen(5)
print(f'服务器已启动,地址:{HOST}:{PORT}')
# 接收客户端连接,创建线程处理
while True:
conn, addr = server_socket.accept()
print(f'客户端 {addr[0]}:{addr[1]} 已连接')
threading.Thread(target=handle_client, args=(conn, addr)).start()
if __name__ == '__main__':
start_server()
```
客户端代码:
```python
import socket
import threading
# 服务器地址和端口
HOST = '127.0.0.1'
PORT = 8888
# 连接服务器
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client_socket.connect((HOST, PORT))
# 输入用户名
username = input('请输入您的用户名:')
client_socket.send(username.encode())
# 接收消息
def receive_msg():
while True:
try:
msg = client_socket.recv(1024).decode()
print(msg)
except:
# 发生异常,断开连接
client_socket.close()
break
# 发送消息
def send_msg():
while True:
msg = input('')
if msg == 'quit':
# 断开连接
client_socket.send(msg.encode())
client_socket.close()
break
else:
# 发送消息
client_socket.send(msg.encode())
# 启动接收消息和发送消息的线程
threading.Thread(target=receive_msg).start()
threading.Thread(target=send_msg).start()
```
使用方法:
1. 运行服务端代码 `python server.py`,启动服务器;
2. 运行客户端代码 `python client.py`,连接到服务器;
3. 输入用户名;
4. 可以发送消息,格式为 `@用户名 消息` 进行私聊,输入 `quit` 断开连接。
注意事项:
1. 本聊天室只是一个简单实现,存在很多安全问题,如无法验证客户端发送的消息;
2. 为了简单起见,服务端只能同时处理一个客户端的消息,需要改进为多线程处理;
3. 客户端和服务端的地址和端口号需要保持一致。
阅读全文