Python中如何正确使用ContentType()函数进行内容类型判断

发布时间：2023-12-23 19:21:39

在Python中，可以使用ContentType()函数来进行内容类型的判断。这个函数位于http.server模块中，可以用于解析HTTP请求头中的内容类型字段。

ContentType()函数的使用方式如下：

from http.server import HTTPServer
from http.server import BaseHTTPRequestHandler

class MyRequestHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        content_type = self.headers.get('Content-Type', '')

        if content_type == 'text/html':
            # 处理HTML类型的请求
            self.send_response(200)
            self.send_header('Content-Type', 'text/html')
            self.end_headers()
            self.wfile.write(b'<html><body><h1>Hello, World!</h1></body></html>')

        elif content_type == 'application/json':
            # 处理JSON类型的请求
            self.send_response(200)
            self.send_header('Content-Type', 'application/json')
            self.end_headers()
            self.wfile.write(b'{"message": "Hello, World!"}')

        else:
            # 处理其他类型的请求
            self.send_response(400)
            self.send_header('Content-Type', 'text/plain')
            self.end_headers()
            self.wfile.write(b'Unsupported Content-Type')

httpd = HTTPServer(('localhost', 8000), MyRequestHandler)
httpd.serve_forever()

上面的例子展示了如何使用ContentType()函数来根据HTTP请求头中的内容类型字段来处理不同类型的请求。在do_GET方法中，首先通过self.headers.get('Content-Type', '')来获取请求头中的Content-Type字段的值。然后使用条件语句判断内容类型，根据不同的内容类型来发送不同的响应。

例如，当接收到的请求头中的Content-Type字段为text/html时，表示客户端希望接收HTML类型的响应。此时，服务器向客户端发送一个包含<html><body><h1>Hello, World!</h1></body></html>内容的HTML响应。

当接收到的请求头中的Content-Type字段为application/json时，表示客户端希望接收JSON类型的响应。此时，服务器向客户端发送一个包含{"message": "Hello, World!"}内容的JSON响应。

如果接收到的请求头中的内容类型字段不是预期的类型，那么服务器会向客户端发送一个400状态码的响应，并返回一个Unsupported Content-Type的提示信息。

使用ContentType()函数可以方便地对内容类型进行判断，并根据不同的内容类型来发送不同的响应。这对于构建Web服务和API非常有用。