使用simplehttpserver进行基本身份验证

from http.server import SimpleHTTPRequestHandler
import base64

# 自定义的简单HTTP请求处理类
class AuthHandler(SimpleHTTPRequestHandler):
    def do_AUTHHEAD(self):
        self.send_response(401)
        self.send_header('WWW-Authenticate', 'Basic realm="Test"')
        self.send_header('Content-type', 'text/html')
        self.end_headers()

    def do_GET(self):
        # 检查Authorization头部是否存在
        if self.headers.get('Authorization') is None:
            self.do_AUTHHEAD()
            self.wfile.write(b'No Authorization header received.')
            pass
        # 检查Authorization头部是否包含正确的用户名和密码
        elif self.headers.get('Authorization') == 'Basic ' + base64.b64encode(b'username:password').decode():
            SimpleHTTPRequestHandler.do_GET(self)
            pass
        else:
            self.do_AUTHHEAD()
            self.wfile.write(b'Authentication failed.')
            pass

# 启动HTTP服务器
if __name__ == '__main__':
    from http.server import HTTPServer
    server = HTTPServer(('localhost', 8000), AuthHandler)
    print('Starting server at http://localhost:8000')
    server.serve_forever()