【PythonWeb开发】Flask中间件钩子函数实现封IP

        在 Flask 框架中， 提供了几种类型的钩子（类似于Django的中间件），它们是在请求的不同阶段自动调用的函数。这些钩子让你能够对请求和响应的处理流程进行扩展，而无需修改核心代码。

Flask钩子的四种类型

---

• before_first_request：在处理第一个请求前调用，只执行一次。(Blueprint没有这个钩子)
• before_request：在每个请求之前调用。
• after_request：如果未触发异常，则在每个请求之后调用，用于处理响应。
• teardown_request：无论请求是否成功完成，都会在每个请求之后调用，常用于关闭资源，如数据库连接。

利用钩子和缓存实现封IP

---

        思路：要实现这样的功能，你可以使用Flask的before_request钩子函数，并结合一个简单的缓存机制来跟踪每个IP的请求频率。使用Flask-Caching扩展可以处理缓存需求，包括存储和清理IP地址的访问记录。下面是如何使用Flask-Caching来实现防止IP频繁访问的示例：

from flask import Blueprint, render_template, request, abort
from flask_caching import Cache


app = Flask(__name__)

# 配置缓存
app.config['CACHE_TYPE'] = 'simple'  # 使用简单的内存缓存
cache = Cache(app)


# 定义一个名为limit_remote_addr的函数，用于限制客户端IP的请求次数
@app.before_request
def limit_remote_addr():
    # 获取客户端的IP地址
    client_ip = request.remote_addr
    # 生成缓存键值，用于存储客户端IP的请求次数
    cache_key = 'ip_limit_' + client_ip

    # 从缓存中获取客户端IP的请求次数
    request_count = cache.get(cache_key)
    # 如果缓存中没有该客户端IP的请求次数，则初始化为0
    if request_count is None:
        request_count = 0
    print(request_count)
    # 如果客户端IP的请求次数大于等于5次，封禁IP
    if request_count >= 2:
        # 将客户端IP的请求次数重置为0，并设置缓存超时时间为10秒(封禁10秒)
        cache.set(cache_key, request_count + 1, timeout=10)
        abort(429)  # Too Many Requests HTTP状态码
    else:
        # 如果客户端IP的请求次数小于5次，将其请求次数加1，并设置缓存超时时间为1秒
        cache.set(cache_key, request_count + 1, timeout=1)


@app.route('/')
def hello_world():
    return 'Hello, World!'


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