自定义请求体

from http.server import BaseHTTPRequestHandler, HTTPServer
from urllib.parse import urlparse, parse_qs
import json

PORT_NUMBER = 8022


def handle_header(header, response):
    replace_headers = {}  # 需要替换或者新增的header
    remove_headers = []  # 需要删除的header

    header_parameters = header  # 客户端原始请求的header内容
    if header_parameters is None or not isinstance(header_parameters, dict):
        print("Invalid event.headerParameters")
        return

    # 示例：替换或新增 header1
    replace_headers["header1"] = "header1"

    # 示例：删除 header2
    if header_parameters.get("header2") is not None:
        remove_headers.append("header2")

    # replace_headers 和 remove_headers 至少有一个非空
    if len(replace_headers) < 1 and len(remove_headers) < 1:
        print("Invalid custom request headers")
        return

    response["replace_headers"] = replace_headers
    response["remove_headers"] = remove_headers


def handle_body(body, response):
    replace_body = ""  # 替换之后的body内容
    # 示例：替换body
    if body is not None:
        replace_body = "hello world"

    # replace_body 必须为字符串类型
    if not isinstance(replace_body, str):
        print("Invalid custom request body")
        return

    response["replace_body"] = replace_body


def handle_query(query, response):
    replace_query = {}  # 需要替换或者新增的query
    remove_query = []  # 需要删除的query

    query_parameters = query
    if query_parameters is None or not isinstance(query_parameters, dict):
        print("Invalid event.queryStringParameters")
        return

    # 示例：替换或新增 query1
    replace_query["query1"] = "query1"

    # 示例：删除原始请求中的 query2
    if query_parameters.get("query2") is not None:
        remove_query.append("query2")

    # replace_query 和 remove_query 至少有一个非空
    if len(replace_query) < 1 and len(remove_query) < 1:
        print("Invalid custom request query")
        return

    response["replace_query"] = replace_query
    response["remove_query"] = remove_query


class MyHandler(BaseHTTPRequestHandler):

    def _set_headers(self):
        self.send_response(200)
        self.send_header('Content-type', 'application/json')
        self.end_headers()

    def do_POST(self):
        response = {}
        try:
            request_body = self.rfile.read(int(self.headers.getheader("Content-length")))
        except:
            request_body = "no body."

        # 重写header
        handle_header(self.headers.__dict__, response)

        # 重写query
        parsed_url = urlparse(self.path)
        request_query = parse_qs(parsed_url.query)
        handle_query(request_query, response)

        # 重写body
        handle_body(request_body, response)

        self._set_headers()
        self.wfile.write(json.dumps(response).encode('utf-8'))
        return


try:
    server = HTTPServer(('', PORT_NUMBER), MyHandler)
    print(f'Started HTTP server on port {PORT_NUMBER}')

    # Wait forever for incoming http requests
    server.serve_forever()

except KeyboardInterrupt:
    print('^C received, shutting down the web server')
    server.socket.close()