单进程但链接非阻塞web静态服务器
"""
@author: Bre Athy
@contact: https://www.zhihu.com/people/you-yi-shi-de-hu-xi
@productware: PyCharm
@file: 单进程但链接非阻塞web静态服务器.py
@time: 2019/11/9 5:33
"""
import socket,re
def service_client(new_socket,recv_data):
path = r"E:\__init__\Downloads\Compressed\mikutap (2)"
ret = re.match(r"GET (.*?) H", recv_data)
if not ret:
print("客户端连接已断开")
else:
if ret.group(1) == r"/":
path += r"\index.html"
else:
path += ret.group(1).replace("/", "\\")
# 准备应答内容
try:
response_headers = bytes("HTTP/1.1 200 OK\r\n", encoding='gbk')
with open(path, "rb")as fp:
response_body = fp.read()
except:
response_body = bytes("<h1>页面不在了哟<h1>", encoding='gbk')
response_headers = bytes("HTTP/1.1 404 NotFound\r\n", encoding='gbk')
response_headers+=(("Content-Length:%d\r\n"%len(response_body)+"\r\n").encode("gbk"))
new_socket.send(response_headers+response_body)
def main():
# 创建套接字并绑定
tcp_server_socket=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
tcp_server_socket.bind(("",7788))
# 准备监听
tcp_server_socket.listen(128)
# 套接字非堵塞,如果没有连接就会报异常
tcp_server_socket.setblocking(False)
# 存储客户端连接套接字的列表
client_socket_list=[]
while True:
try:
# 等待客户端连接
new_socket,addr=tcp_server_socket.accept()
except Exception as ret:
pass
else:
# 收到的客户端套接字也要设置非堵塞
new_socket.setblocking(False)
# 存储到列表
client_socket_list.append(new_socket)
# 遍历列表
for client_socket in client_socket_list:
# 数据没收到会报出异常
try:
recv_data=client_socket.recv(1024)
except Exception as ret:
pass
else:
# 有返回值,但返回值为空,这是关闭连接的信号
if recv_data:
recv_data=recv_data.decode("gbk")
service_client(client_socket,recv_data)
else:
client_socket.close()
client_socket_list.remove(client_socket)
# 为客户端服务
if __name__ == "__main__":
main()