使用Python开发一个简单的聊天应用。应用将允许多个用户之间进行实时文字聊天。
首先导入一些构建该应用所需的Python库:
import socket
import select
import errno
import sys
socket
库用于网络通信select
库用于实现异步I/Oerrno
库将获取错误号sys
库将提供一些系统功能下一步是创建一个Socket。这将允许应用程序通过网络发送和接收数据:
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
这里创建了一个TCP Socket:
socket.AF_INET
指定使用IPv4地址socket.SOCK_STREAM
指定这是一个TCP Socket为了接收连接,需要将Socket绑定到地址和端口。这里我们将绑定到localhost
的端口8000
:
host = "localhost"
port = 8000
server_socket.bind((host, port))
这样就在指定主机和端口上进行了监听。
下一步是调用listen()
来将Socket置于侦听模式:
server_socket.listen(5)
5
是可排队连接的最大数量。
在侦听后,可以使用accept()
开始接收连接:
client_socket, address = server_socket.accept()
这将接收到一个客户端Socket和客户端的地址。
可以使用循环来不断接收新的连接:
while True:
client_socket, address = server_socket.accept()
# 处理连接
有了连接后,就可以通过Socket发送和接收数据。
例如,接收1024字节的数据:
data = client_socket.recv(1024)
发送一些数据:
msg = "Hello World"
client_socket.sendall(msg.encode("utf-8"))
send()
而不是sendall()
数据可能不会完全发送完成后需要正确关闭Socket:
client_socket.shutdown(socket.SHUT_RDWR)
client_socket.close()
这将关闭Socket的发送和接收通道,然后关闭Socket自身。
下面是一个支持多用户聊天的完整示例:
import select
import socket
import sys
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
host = "localhost"
port = 8000
server_socket.bind((host, port))
server_socket.listen(100)
all_sockets = [server_socket]
clients = {}
while True:
read_ready, _, _ = select.select(all_sockets, [], [])
for sock in read_ready:
if sock == server_socket:
client_socket, address = server_socket.accept()
all_sockets.append(client_socket)
clients[client_socket] = address
else:
data = sock.recv(1024)
if not data:
print(f"Closed connection from {clients[sock]}")
all_sockets.remove(sock)
sock.close()
del clients[sock]
else:
user = clients[sock]
broadcast(f"User {user[0]}:{user[1]} said: {data.decode()}", sock)
def broadcast(msg, exclude_socket):
for sock in all_sockets:
if sock != server_socket and sock != exclude_socket:
sock.sendall(msg.encode())
主要步骤概述:
select
来处理读事件这样就实现了一个支持多个用户的简单聊天应用!