Python中BaseHTTPRequestHandler实现简单的API接口

#! /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()

http_server = HTTPServer(('', int(port)), ServerHTTP)

http_server.serve_forever()

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文本流对象，可读取请求信息