文档中心

			

				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()