BaseHTTPRequestHandler实现API接口
BaseHTTPRequestHandler介绍
这是一个以TCPServer
为基础开发的模块,可以在请求外层添加http
协议报文,发送http
协议。
#! /usr/bin/env python3
# -*- coding:UTF-8 -*-
from http.server import HTTPServer, BaseHTTPRequestHandler
import json
import cgi
import datetime
host = ('',8003)
engine = create_engine('mysql+pymysql://pointgrab_user:pointgrabAaaa1111@rm-bp1d2s03ka9b803602o.mysql.rds.aliyuncs.com:3306/pointgrab_info',echo=False)
class TodoHandler(BaseHTTPRequestHandler):
def do_GET(self):
self.send_error(415, 'Only post is supported')
def do_POST(self):
ctype, pdict = cgi.parse_header(self.headers['content-type'])
# print(ctype, pdict)
token = self.headers['X-Auth-Token']
# print(token)
if token == 'token' and ctype == 'application/json':
path = str(self.path) # 获取请求的url
if path == '/api/counting/':
# print(path)
length = int(self.headers['content-length']) # 获取除头部后的请求参数的长度
datas = self.rfile.read(length) # 获取请求参数数据,请求数据为json字符串
# print(datas)
rjson = json.loads(datas.decode())
# print(rjson,type(rjson))
countingdf = counting(rjson)
data_into_sql(countingdf)
self.send_response(200)
self.send_header('Content-type', 'application/json')
self.end_headers()
self.wfile.write(json.dumps('Counting data is inserted').encode())
elif path == '/api/traffic/':
# print(path)
length = int(self.headers['content-length']) # 获取除头部后的请求参数的长度
datas = self.rfile.read(length) # 获取请求参数数据,请求数据为json字符串
# print(datas)
rjson = json.loads(datas.decode())
# print(rjson,type(rjson))
trafficdf = traffic(rjson)
data_into_sql(trafficdf)
self.send_response(200)
self.send_header('Content-type', 'application/json')
self.end_headers()
self.wfile.write(json.dumps('Traffic data is inserted').encode())
else:
self.send_error(404, "Not Found")
else:
self.send_error(415, "Only json data is supported.")
if __name__ == '__main__':
server = HTTPServer(host, TodoHandler)
print("Starting server, listen at: %s:%s" % host)
server.serve_forever()
请求代码
POST /api/counting/ HTTP/1.1
Host: 127.0.0.1:8003
Content-Type: application/json
X-Auth-Token: token
Accept: */*
Host: 127.0.0.1:8003
content-length: 126
Connection: keep-alive
cache-control: no-cache
{"areaId":"vQ_82PjcQFyt5LHA_751sg","devices":["R3QAztuSQZusLP5Fz9jVXg"],"count":2,"type":"COUNTING","timestamp":1526376098800}
BaseHTTPServer:
主要包含两个类HTTPServer和BaseHTTPRequestHandler
HTTPServer:
继承SocketServer.TCPServer,用于获取请求,并将请求分配给应答程序处理
BaseHTTPRequestHandler:
继承SocketServer.StreamRequestHandler,对http连接的请求作出应答(response)
基于BaseHTTPServer 的Http Server的处理流程:
- HTTPServer绑定对应的应答类(BaseHTTPRequestHandler )
http_server = HTTPServer(('', int(port)), ServerHTTP)- 监听端口:
** http_server.serve_forever()**
serve_forever()方法使用select.select()循环监听请求,当接收到请求后调用
当监听到请求时,取出请求对象- 应答:
创建新线程以连接对象(开始理解成请求对象)为参数实例化应答类:ServerHTTP()应答类根据请求方式调用ServerHTTP.do_XXX处理方法
常用方法/属性:
'''
BaseHTTPRequestHandler.path #包含的请求路径和GET请求的数据
BaseHTTPRequestHandler.command #请求类型GET、POST...
BaseHTTPRequestHandler.request_version #请求的协议类型HTTP/1.0、HTTP/1.1
BaseHTTPRequestHandler.headers #请求的头
BaseHTTPRequestHandler.responses #HTTP错误代码及对应错误信息的字典
BaseHTTPRequestHandler.handle() #用于处理某一连接对象的请求,调用handle_one_request方法处理
BaseHTTPRequestHandler.handle_one_request() #根据请求类型调用do_XXX()方法,XXX为请求类型
BaseHTTPRequestHandler.do_XXX() #处理请求
BaseHTTPRequestHandler.send_error() #发送并记录一个完整的错误回复到客户端,内部调用send_response()方法实现
BaseHTTPRequestHandler.send_response() #发送一个响应头并记录已接收的请求
BaseHTTPRequestHandler.send_header() #发送一个指定的HTTP头到输出流。 keyword 应该指定头关键字,value 指定它的值
BaseHTTPRequestHandler.end_headers() #发送一个空白行,标识发送HTTP头部结束
BaseHTTPRequestHandler.wfile #self.connection.makefile('rb', self.wbufsize) self.wbufsize = -1 应答的HTTP文本流对象,可写入应答信息
BaseHTTPRequestHandler.rfile #self.connection.makefile('wb', self.rbufsize) self.rbufsize = 0 请求的HTTP文本流对象,可读取请求信息
'''
cgi.parse_header(string)
解析HTTP报头字段(如'content-type’ )之后提供的数据。
数据将被分解为一个主值和一个次要参数的字典、并以元组的形式返回。