import socket
import select
import re
#模拟浏览器接受的请求和应答
def recv(new_tcp,recv_data):
ret= re.match(r"[^/]*/([^ ]*)\sHTTP",recv_data).group(1)
if ret =="":
ret = "index.html"
try:
file = open ("./html/"+ret,"rb")
except:
con = "<h1>页面去火星了<h1>".encode("utf-8")
b = "HTTP/1.1 404 NOFOUND\r\n"
b += "Content-Type: text/html; charset=utf-8\r\n"
b += "Content-Length:%d\r\n"%len(con)
b+="\r\n"
new_tcp.send(b.encode("utf-8"))
new_tcp.send(con)
else:
content = file.read()
file.close()
a = "HTTP/1.1 200 200\r\n"
a += "Content-Type: text/html; charset=utf-8\r\n"
a += "Content-Length:%d\r\n"%len(content)
a +="\r\n"
new_tcp.send(a.encode("utf-8"))
new_tcp.send(content)
def main():
socket_tcp = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
# 取消端口占用
socket_tcp.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
socket_tcp.bind(('',8080))
socket_tcp.listen(128)
#设置非堵塞
socket_tcp.setblocking(False)
#创建一个epoll对象
epl =select.epoll()
epl.register(socket_tcp.fileno(),select.EPOLLIN)
#创建字典
tcp_dict=dict()
while True:
tcp_list = epl.poll()
for fd,event in tcp_list:
if fd == socket_tcp.fileno():
new_tcp,new_port = socket_tcp.accept()
epl.register(new_tcp.fileno(),select.EPOLLIN)
tcp_dict[new_tcp.fileno()] = new_tcp
else:
recv_data = tcp_dict[fd].recv(1024).decode("utf-8")
if recv_data:
recv(tcp_dict[fd],recv_data)
else:
tcp_dict[fd].close()
epl.unregister[fd]
del tcp_dict[fd]
if __name__ == "__main__":
main()