wsgiref.util模块中的常用工具函数介绍

wsgiref.util模块是Python标准库的一个部分，提供了一些在编写WSGI（Web Server Gateway Interface）应用程序时常用的工具函数。下面将介绍几个常用的工具函数并给出使用示例。

1. wsgiref.util.shift_path_info(environ)

该函数用于获取HTTP请求中的URL路径，并将路径中的 段移除。返回值为移除 段后的路径（不含斜杠），如果路径为空，则返回空字符串。

示例：

from wsgiref.util import shift_path_info

def application(environ, start_response):
    path = environ.get('PATH_INFO', '')
    first_path = shift_path_info(environ)
    
    start_response('200 OK', [('Content-type', 'text/plain')])
    return [f"Path: {path}
First Path: {first_path}".encode()]

# 测试：
# 请求URL：http://localhost:8080/hello/world
# 输出：
# Path: /hello/world
# First Path: hello

2. wsgiref.util.request_uri(environ, include_query=True)

该函数返回完整的请求URL，包括scheme、host、port、path以及可选的查询字符串。

示例：

from wsgiref.util import request_uri

def application(environ, start_response):
    uri = request_uri(environ, include_query=False)
    
    start_response('200 OK', [('Content-type', 'text/plain')])
    return [f"Request URI: {uri}".encode()]

# 测试：
# 请求URL：http://localhost:8080/hello/world?name=John
# 输出：
# Request URI: http://localhost:8080/hello/world

3. wsgiref.util.setup_testing_defaults(environ)

该函数可以设置environ字典的默认值，以模拟一个测试用的WSGI环境。

示例：

from wsgiref.util import setup_testing_defaults

def application(environ, start_response):
    setup_testing_defaults(environ)

    start_response('200 OK', [('Content-type', 'text/plain')])
    return [f"Method: {environ['REQUEST_METHOD']} 
"
            f"Path: {environ['PATH_INFO']} 
"
            f"Query String: {environ['QUERY_STRING']} 
".encode()]

# 测试：
# 输出：
# Method: GET
# Path: /
# Query String: 

# 调用该应用程序后，可以通过environ字典获取HTTP请求的方法、路径和查询字符串。

4. wsgiref.util.shift_query_string(environ)

该函数用于从environ字典中移除查询字符串并返回。移除后的environ字典不再包含QUERY_STRING字段。

示例：

from wsgiref.util import shift_query_string

def application(environ, start_response):
    query_string = shift_query_string(environ)
    
    start_response('200 OK', [('Content-type', 'text/plain')])
    return [f"Query String: {query_string}".encode()]

# 测试：
# 请求URL：http://localhost:8080/hello/world?name=John&age=25
# 输出：
# Query String: name=John&age=25
# 调用该函数后，environ字典中将不再包含QUERY_STRING字段。

上述是wsgiref.util模块中的一些常用工具函数介绍及使用示例。这些函数可以帮助开发者更方便地处理HTTP请求和构建WSGI应用程序。