在本文中，我们将探讨如何使用Flask和Flask-JWT-Extended库来保护您的API免受跨站请求攻击（CSRF）。我们将首先简要介绍CSRF攻击的概念，然后详细说明如何使用Flask-JWT-Extended库来保护您的API。
什么是跨站请求攻击（CSRF）？
跨站请求攻击（CSRF）是一种网络攻击手段，攻击者通过在合法用户的浏览器中植入恶意代码，诱使用户在不知情的情况下执行非预期的操作。这些操作可能包括更改密码、删除账户或执行其他敏感操作。为了防止这种攻击，我们需要确保只有经过身份验证的用户才能执行特定操作。
使用Flask-JWT-Extended保护API
Flask-JWT-Extended是一个用于Flask应用的扩展，它提供了JSON Web Tokens（JWT）的支持。JWT是一种用于身份验证和授权的令牌，可以在客户端和服务器之间安全地传输信息。接下来，我们将介绍如何使用Flask-JWT-Extended库来保护您的API免受CSRF攻击。
安装Flask-JWT-Extended
首先，您需要安装Flask-JWT-Extended库：
pip install Flask-JWT-Extended
初始化Flask-JWT-Extended
接下来，在您的Flask应用中导入并初始化Flask-JWT-Extended：
from flask_jwt_extended import JWTManager
app = Flask(__name__)
app.config['JWT_SECRET_KEY'] = 'your-secret-key'
jwt = JWTManager(app)
创建登录视图函数
现在，我们需要创建一个登录视图函数，用于验证用户凭据并生成JWT：
from flask import request, jsonify
from flask_jwt_extended import create_access_token
@app.route('/login', methods=['POST'])
def login():
username = request.json.get('username', None)
password = request.json.get('password', None)
# 验证用户凭据
if username and password:
# 假设用户已经验证
access_token = create_access_token(identity=username)
return jsonify(access_token=access_token), 200
else:
return jsonify(message="Invalid credentials"), 401
保护API端点
接下来，我们需要在API视图函数中使用@jwt_required()装饰器来保护API端点：
from flask import request, jsonify
from flask_jwt_extended import jwt_required, get_jwt_identity
@app.route('/protected', methods=['GET'])
@jwt_required()
def protected():
current_user = get_jwt_identity()
return jsonify(logged_in_as=current_user), 200
客户端请求
最后，客户端在发起请求时需要在请求头中包含JWT：
Authorization: Bearer <your-access-token>
通过以上步骤，您的Flask API将使用JWT来保护API端点免受跨站请求攻击。请注意，这里的示例仅用于演示目的，实际应用中您需要根据自己的需求进行相应的调整。
总结
在本文中，我们介绍了如何使用Flask和Flask-JWT-Extended库来保护您的API免受跨站请求攻击。通过使用JWT，您可以确保只有经过身份验证的用户才能访问受保护的API端点，从而提高您的应用的安全性。
在 Flask 应用中，flask_wtf 和 flask_jwt_extended 是两个不同的库，它们分别用于处理不同的功能
- Flask-WTF：Flask-WTF 是一个用于处理表单和验证的库。它基于 WTForms 库，提供了一些额外的功能，如 CSRF 保护、表单验证和表单渲染。Flask-WTF 主要用于处理用户输入，确保数据的有效性和安全性。
- Flask-JWT-Extended：Flask-JWT-Extended 是一个用于处理 JSON Web Tokens (JWT) 的库。JWT 是一种用于身份验证和授权的轻量级安全令牌。Flask-JWT-Extended 提供了一组用于生成、验证和保护 JWT 的功能。它可以与 Flask 应用程序无缝集成，以实现基于 JWT 的身份验证和授权。
总之，Flask-WTF 和 Flask-JWT-Extended 是两个不同的库，分别用于处理表单和 JWT 身份验证。它们可以在同一个 Flask 应用程序中一起使用，以实现更强大的功能和更高的安全性。
