接前面的工作 。这里主要在做一个简单的简单的http服务器。其实这个http是网上抄别人的。
主要文件有三个。
1,server.py 很简单的处理socket,还有处理httphead的线程。是一个单线程模式。
2.HttpHead.py 处理 http协议的一些 文本拼接,分板工作。
3,huya-server.py 处理huya视频直播地址的取点,并生成http返回页面。
直接代码。
1,server.py
# server.py
# -*- coding=utf-8 -*-
import socket
import threading
from HttpHead import HttpRequest
# 处理每一个请求
def tcp_link(sock, addr):
#print('Accept new connection from %s:%s...' % addr)
request = sock.recv(1024)
#存在空请求
if request!='':
http_req = HttpRequest()
http_req.pass_request(request)
# 发送数据
sock.send(http_req.get_response())
sock.close()
# 开启服务器
def start_server():
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 监听端口:
s.bind(('127.0.0.1', 9999))
s.listen(5)
# print('Waiting for connection...')
while True:
# 接受一个新连接:
sock, addr = s.accept()
# 创建新线程来处理TCP连接:
t = threading.Thread(target=tcp_link, args=(sock, addr))
t.start()
if __name__ == '__main__':
start_server()
pass
2.HttpHead.py 处理 http协议的一些 文本拼接,分板工作。
# -*- coding:utf-8 -*-
import os
# 返回码
class ErrorCode(object):
OK = "HTTP/1.1 200 OK\r\n"
NOT_FOUND = "HTTP/1.1 404 Not Found\r\n"
# Content类型
class ContentType(object):
HTML = 'Content-Type: text/html\r\n'
PNG = 'Content-Type: img/png\r\n'
class HttpRequest(object):
RootDir = 'C:\inetpub\wwwroot\python1' # 根目录
NotFoundHtml = RootDir+'/'+'404.html' # 404页面
def __init__(self):
self.method = None
self.url = None
self.protocol = None
self.host = None
self.request_data = None
self.response_line = ErrorCode.OK # 响应码
self.response_head = ContentType.HTML # 响应头部
self.response_body = '' # 响应主题
# 解析请求,得到请求的信息
def pass_request(self, request):
#print 'request:', request
request_line, body = request.split('\r\n', 1)
header_list = request_line.split(' ')
self.method = header_list[0].upper()
self.url = header_list[1]
#print self.url
self.protocol = header_list[2]
# 获得请求参数
if self.method == 'POST':
self.request_data = {}
request_body = body.split('\r\n\r\n', 1)[1]
parameters = request_body.split('\n') # 每一行是一个字段
for i in parameters:
key, val = i.split('=')
self.request_data[key] = val
self.handle_file_request(HttpRequest.RootDir + self.url)
if self.method == 'GET':
file_name = ''