087. 编写一个函数，实现简单的Web框架

在 Python 中，可以使用内置的 http.server 模块来实现一个简单的 Web 框架。虽然这个模块主要用于开发和测试，但它可以帮助我们理解 Web 框架的基本原理。以下是一个简单的 Web 框架实现，支持基本的路由和请求处理。

示例代码

以下代码实现了一个简单的 Web 框架，支持基本的路由和请求处理。

import http.server
import socketserver

class SimpleWebFramework(http.server.SimpleHTTPRequestHandler):
    routes = {
        '/': 'index.html',
        '/about': 'about.html',
        '/contact': 'contact.html'
    }

    def do_GET(self):
        if self.path in self.routes:
            self.path = self.routes[self.path]
            return super().do_GET()
        else:
            self.send_response(404)
            self.end_headers()
            self.wfile.write(b"404 Not Found")

def start_server(port=8000):
    with socketserver.TCPServer(("", port), SimpleWebFramework) as httpd:
        print(f"Web 服务器启动，监听端口 {port}")
        try:
            httpd.serve_forever()
        except KeyboardInterrupt:
            print("Web 服务器关闭")
        finally:
            httpd.server_close()

# 主函数
if __name__ == "__main__":
    start_server()

代码说明

继承 http.server.SimpleHTTPRequestHandler：创建一个自定义的请求处理类 SimpleWebFramework，继承自 http.server.SimpleHTTPRequestHandler。

定义路由：使用一个字典 routes 定义简单的路由规则，将 URL 路径映射到静态文件。

处理 GET 请求：

  • 重写 do_GET 方法来处理 GET 请求。

  • 如果请求的路径在路由表中，将路径映射到对应的静态文件，并调用父类的 do_GET 方法。

  • 如果路径不在路由表中，返回 404 错误。

启动服务器：

  • 使用 socketserver.TCPServer 创建一个 HTTP 服务器，指定监听端口和请求处理类。

  • 使用 serve_forever 方法启动服务器。

异常处理：使用 try-except 捕获 KeyboardInterrupt，以便在用户按下 Ctrl+C 时优雅地关闭服务器。

示例输出

假设运行上述代码，Web 服务器将启动并监听指定的端口。当客户端访问不同的路由时，服务器将返回对应的静态文件内容。

注意事项

安全性：

  • 这个简单的 Web 框架没有实现身份验证或加密，因此不适用于生产环境。

  • 在实际应用中，需要实现更复杂的安全机制。

功能限制：

  • 这个实现仅支持静态文件的路由和请求处理，不支持动态内容生成。

  • 在实际应用中，可以扩展实现动态内容生成和数据库支持。

错误处理：在实际应用中，建议添加更详细的异常处理机制，以处理网络错误、文件读取错误等情况。

扩展功能

如果你需要更复杂的 Web 框架功能，可以考虑以下内容：

  1. 支持动态内容生成：使用模板引擎（如 Jinja2）生成动态内容。
  2. 支持中间件：实现中间件机制，用于处理请求和响应的预处理和后处理。
  3. 支持数据库：集成数据库支持，实现数据的持久化存储。
  4. 支持 RESTful API：实现 RESTful API 的路由和处理。
  5. 支持会话管理：实现会话管理机制，用于跟踪用户状态。

视频讲解

BiliBili： 视睿网络-哔哩哔哩视频 (bilibili.com)