面试题：Python构建自定义HTTP请求中间件及处理复杂响应

设计思路

请求处理：
- 使用Python的wsgiref模块（简单的WSGI服务器实现）或者Flask、Django等Web框架来处理HTTP请求。这里以Flask为例，因为它相对轻量且易于理解。
- 对于不同的请求URL，根据预设规则动态修改请求头。可以通过一个字典来存储不同URL对应的请求头修改规则。
响应处理：
- 处理分块传输编码的响应，在Flask中，响应对象会自动处理分块传输编码的情况。如果是手动处理，可以根据Transfer - Encoding: chunked头，按块读取响应内容。
- 对于响应头中的特殊字段处理，同样可以通过在响应处理阶段检查响应头字段，并根据需要进行处理。

核心代码实现（基于Flask）

from flask import Flask, request, Response
app = Flask(__name__)

# 存储不同URL对应的请求头修改规则
url_header_rules = {
    '/specific_url': {'new_header': 'new_value'}
}

@app.before_request
def modify_request_header():
    for url, headers in url_header_rules.items():
        if request.url_rule and request.url_rule.rule == url:
            for key, value in headers.items():
                request.headers[key] = value


@app.after_request
def handle_response(response):
    # 处理分块传输编码响应（Flask自动处理，这里为说明完整性）
    # 如果手动处理分块传输编码，可以如下：
    # if 'Transfer - Encoding' in response.headers and response.headers['Transfer - Encoding'] == 'chunked':
    #     # 按块读取响应内容并处理
    # 处理响应头中的特殊字段
    if 'Special - Header' in response.headers:
        # 这里可以根据特殊字段的值进行相应处理
        pass
    return response


if __name__ == '__main__':
    app.run(debug=True)

上述代码中：

before_request装饰器的函数modify_request_header在每个请求处理前，根据请求的URL修改请求头。
after_request装饰器的函数handle_response在每个请求处理后，对响应进行处理，包括检查和处理分块传输编码以及特殊的响应头字段。

面试题：Python构建自定义HTTP请求中间件及处理复杂响应

知识考点

面试题答案

设计思路

核心代码实现（基于Flask）