在Flask框架中，装饰器是实现功能复用的强大工具

在Flask框架中，装饰器是实现功能复用的强大工具。它们可以用于各种场景，比如路由处理、权限校验、日志记录等。以下是一些Flask装饰器的技巧：

自定义装饰器： 创建一个自定义装饰器，可以让你轻松地为视图函数添加额外的功能，例如记录请求信息。

from functools import wraps

from flask import Flask, g

app = Flask(__name__)

def record_request(func):

@wraps(func)

def wrapper(*args, **kwargs):

g.request_start_time = time.time()

response = func(*args, **kwargs)

# 记录请求结束时间和日志等

return response

return wrapper

@app.route('/')

@record_request

def index():

pass

组合装饰器： Flask中的@login_required就是一个组合装饰器的例子。它可以结合多个装饰器，例如先认证再记录请求。

def authenticate(func):

@wraps(func)

def wrapper(*args, **kwargs):

if not g.user:

return redirect(url_for('login'))

return func(*args, **kwargs)

def record_request(func):

# ...

@app.route('/secure')

@authenticate

@record_request

def secure_view():

pass

使用上下文vars： 通过g对象，可以在装饰器内部存储和检索跨函数的数据，这在处理数据库连接或会话时非常有用。

def get_db_connection(func):

@wraps(func)

def wrapper(*args, **kwargs):

if 'db' not in g:

g.db = connect_to_database()

return func(*args, db=g.db, **kwargs)

@app.route('/data')

@get_db_connection

def data_view():

# 使用g.db进行数据库操作

pass

传递额外参数： 装饰器可以接受额外的参数。

def require_admin(role=None):

def decorator(func):

@wraps(func)

def wrapper(*args, **kwargs):

if g.user and not user.has_role(role):

return abort(403)

return func(*args, **kwargs)

return wrapper

return decorator

@app.route('/admin')

@require_admin(role='admin')

def admin_view():

pass

使用闭包： 装饰器可以用作闭包，允许你封装私有变量或仅在装饰器内部使用的函数。

def counter():

count = [0]

def decorator(func):

@wraps(func)

def wrapper(*args, **kwargs):

count[0] += 1

return func(*args, **kwargs)

return wrapper

return decorator

@app.route('/visit', methods=['GET'])

@counter()

def visit():

return f"This is visit number {g.visit_count}", 200

使用这些技巧，你可以编写出更加灵活和强大的Flask应用程序。