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的处理流程：

1. HTTPServer绑定对应的应答类（BaseHTTPRequestHandler ）
   http_server = HTTPServer(('', int(port)), ServerHTTP)
2. 监听端口：
   ** http_server.serve_forever()**
   serve_forever()方法使用select.select()循环监听请求，当接收到请求后调用
   当监听到请求时，取出请求对象
3. 应答：
   创建新线程以连接对象(开始理解成请求对象)为参数实例化应答类: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’ )之后提供的数据。
数据将被分解为一个主值和一个次要参数的字典、并以元组的形式返回。